Code Server/Code Server.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "Code Server.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/PradyumnaKrishna/Colab-Hacks/blob/master/Code%20Server/Code%20Server.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "hWw875DoFeRr"
      },
      "source": [
        "# **CODE on Colab**\n",
        "\n",
        "`Welcome to a new notebook. As I showed you a previous notebook called Colab RDP, This notebook feature VSCode (If possible going to add more) integration on web browser using colab.`\n",
        "\n",
        "\n",
        "To use other similiar Notebooks browse my Repository [Colab Hacks](https://github.com/PradyumnaKrishna/Colab-Hacks)"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "cellView": "form",
        "id": "l1N_NosDIOaK"
      },
      "source": [
        "# @title Mount & Clone\n",
        "\n",
        "# Imports\n",
        "import requests\n",
        "\n",
        "# @markdown Mount GDrive\n",
        "Gdrive = False  # @param {type: 'boolean'}\n",
        "global path\n",
        "\n",
        "# @markdown Clone Repository (Github)\n",
        "repositoryUrl = \"\" # @param {type: 'string'}\n",
        "\n",
        "class MountClone:\n",
        "    def __init__(\n",
        "            self,\n",
        "            Gdrive=False,\n",
        "            repositoryUrl=\"\",\n",
        "            verbose=False):\n",
        "        \n",
        "        if Gdrive:\n",
        "            self.mount_drive()\n",
        "        if repositoryUrl:\n",
        "            self.clone_repo(repositoryUrl, verbose)\n",
        "\n",
        "    def __str__(self):\n",
        "        try:\n",
        "            return self.path\n",
        "        except NameError:\n",
        "            return ''\n",
        "        \n",
        "    def mount_drive(self):\n",
        "        from google.colab import drive\n",
        "        drive.mount('/content/drive')\n",
        "        self.path = \"/content/drive\"\n",
        "\n",
        "    def clone_repo(self, repositoryUrl, verbose=False):\n",
        "        response = requests.get(repositoryUrl)\n",
        "        if response.status_code == 200:\n",
        "            print(\"✔️ Public repository\")\n",
        "            ! cd /content\n",
        "            ! git clone $repositoryUrl\n",
        "            folder = repositoryUrl.split(\"/\")[-1]\n",
        "            self.path = f\"/content/{folder}\"\n",
        "        else:\n",
        "            print(\"❌ Not a public Repository\")\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    path = MountClone(Gdrive=Gdrive, repositoryUrl=repositoryUrl)"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "1-gADX-DD6_I",
        "cellView": "form"
      },
      "source": [
        "# @title VSCode\n",
        "\n",
        "# Imports\n",
        "import os\n",
        "import requests\n",
        "import re\n",
        "import shlex\n",
        "import time\n",
        "import signal\n",
        "\n",
        "from subprocess import Popen, PIPE, call\n",
        "\n",
        "# @markdown This code cell runs [code-server](https://github.com/cdr/code-server) with agro tunnel.\n",
        "\n",
        "# @markdown Password (Can be None)\n",
        "Password = \"\"  # @param {type: 'string'}\n",
        "\n",
        "# @markdown Sleep\n",
        "Sleep = False  # @param {type: 'boolean'}\n",
        "\n",
        "# @markdown Run this cell to start vscode\n",
        "\n",
        "class ColabCode:\n",
        "    def __init__(\n",
        "            self,\n",
        "            port=10000,\n",
        "            password='',\n",
        "            verbose=False,\n",
        "            sleep=False):\n",
        "\n",
        "        self.install_server(verbose)\n",
        "        self.start_code(port=port, password=password, verbose=verbose)\n",
        "\n",
        "        if sleep:\n",
        "            time.sleep(43200)\n",
        "\n",
        "    @staticmethod\n",
        "    def __get_agro():\n",
        "        hostname = None\n",
        "        output = requests.get(\"http://localhost:45678/metrics\").text\n",
        "        result = re.search('cloudflared_tunnel_user_hostnames_counts{userHostname=\"https://(.+?)\"}', output)\n",
        "        if result is None:\n",
        "            raise Exception(\"Cannot get any result from cloudflared metrics\")\n",
        "        hostname = result.group(1)\n",
        "        if hostname is None:\n",
        "            raise Exception(\"Cannot get the hostname from cloudflared, it looks like the connection has failed.\")\n",
        "        return hostname\n",
        "\n",
        "    def install_server(self, verbose=False):\n",
        "        os.system(\"kill -9 $(ps aux | grep 'cloudflared' | awk '{print $2}')\")\n",
        "\n",
        "        if not os.path.isfile(\"cloudflared\"):\n",
        "            ! wget -q -nc https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 -O cloudflared &> /dev/null\n",
        "            ! chmod +x cloudflared\n",
        "        else:\n",
        "            if verbose:\n",
        "                print(\"DEBUG: Skipping installation\")\n",
        "        \n",
        "        if not os.path.isdir(\"/root/.local/share/code-server/\"):\n",
        "            settings_path = '/root/.local/share/code-server/User'\n",
        "            ! curl -fsSL https://code-server.dev/install.sh | sh &> /dev/null\n",
        "            if not os.path.exists(settings_path):\n",
        "                os.makedirs(settings_path)\n",
        "            with open(f'{settings_path}/settings.json', 'w') as settings:\n",
        "                settings.write(\"\"\"{\n",
        "    \"extensions.autoUpdate\": false,\n",
        "    \"extensions.autoCheckUpdates\": false\n",
        "}\"\"\")\n",
        "\n",
        "    def start_code(self, port=10000, password=None, verbose=False):\n",
        "        if password:\n",
        "            code_cmd = f'PASSWORD={password} code-server --port {port} --disable-telemetry'\n",
        "        else:\n",
        "            code_cmd = f'code-server --port {port} --auth none --disable-telemetry'\n",
        "\n",
        "        agro_cmd = f'./cloudflared tunnel --url http://127.0.0.1:10000 --metrics localhost:45678'\n",
        "\n",
        "        code_proc = Popen([code_cmd], shell=True, stdout=PIPE)\n",
        "        agro_proc = Popen(shlex.split(agro_cmd), stdout=PIPE)\n",
        "\n",
        "        if verbose:\n",
        "            print(f\"DEBUG: Code Server process: {code_proc.pid}\")\n",
        "            print(f\"DEBUG: Cloudflared process: {agro_proc.pid}\")\n",
        "\n",
        "        time.sleep(3)\n",
        "        try:\n",
        "            info = self.__get_agro()\n",
        "            print('VSCode Server is ready, Visit https://{}/{}'.format(\n",
        "                info, \n",
        "                f'?folder={path}' if path else ''))\n",
        "        except NameError:\n",
        "            info = self.__get_agro()\n",
        "            print(f'VSCode Server is ready, Visit https://{info}')\n",
        "        except Exception:\n",
        "            os.kill(code_proc.pid, signal.SIGKILL)\n",
        "            os.kill(agro_proc.pid, signal.SIGKILL)\n",
        "            print(f\"FAILED: Killing Server... Retry again ...\")\n",
        "            sleep = False\n",
        "\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    ColabCode(password=Password, sleep=Sleep)"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "cellView": "form",
        "id": "KDhBSBj8Ncoz"
      },
      "source": [
        "# @title VSCode Extension Install\n",
        "# @markdown Visit [Open VSX](https://open-vsx.org) and copy the download link of extension and paste here\n",
        "# @markdown any downloadable url of `vsix` file supported\n",
        "\n",
        "downloadUrl = \"https://open-vsx.org/api/ms-python/python/2020.10.332292344/file/ms-python.python-2020.10.332292344.vsix\" # @param {type: 'string'}\n",
        "\n",
        "if downloadUrl:\n",
        "    ! wget $downloadUrl -O /tmp/extension.vsix &> /dev/null\n",
        "    ! code-server --install-extension /tmp/extension.vsix\n",
        "    ! rm /tmp/extension.vsix"
      ],
      "execution_count": null,
      "outputs": []
    }
  ]
}