VS Code(code-server) setup script on Colab via an ngrok tunnel

ngrok.py

import fire
from pyngrok import ngrok

def set_ngrok_auth_token(ngrok_auth_token):
  ngrok.set_auth_token(ngrok_auth_token)
  
if __name__ == "__main__":
  fire.Fire(set_ngrok_auth_token)

setup.sh

#!/bin/bash

# Download and install code-server if not already downloaded
if [ ! -f install.sh ]; then
    wget https://raw.githubusercontent.com/coder/code-server/main/install.sh
    chmod +x install.sh
fi
./install.sh

# Install pyngrok
pip install pyngrok fire

# Download ngrok.py from this gist if not already downloaded
if [ ! -f ngrok.py ]; then
    wget https://gist.githubusercontent.com/sachith-gunasekara/06ca96a0ff907415c3d94a09420c1e6c/raw/be236c014068a5cedcc3facda7b0f018c2682c16/ngrok.py
fi

# Set up ngrok with auth token
python ngrok.py --ngrok_auth_token $2

# Tunnel given port to ngrok server
ngrok http $1 & /usr/bin/code-server --port $1 & sleep 5s && cat /root/.config/code-server/config.yaml