{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/buzz-lightsnack-2007/Desktop-CoLab/blob/main/DesktopCoLab.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "-j036FZBO7_O"
      },
      "source": [
        "# Desktop@CoLab\n",
        "\n",
        "**A Linux virtual machine, all in Google's servers.**\n",
        "\n",
        "<small>_Uses [remocolab](https://urlzs.com/jBN9S) by [demotomohiro](https://urlzs.com/ds5GF)._</small>\n",
        "\n",
        "---\n",
        "\n",
        "\n",
        "_**Important**: To start a new instance using the same account, please re-set the runtime. Forgetting to do so will cause an error._ \n",
        "\n",
        "---"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "Uys7SO6pOod_"
      },
      "outputs": [],
      "source": [
        "# DESKTOP@COLAB\n",
        "# Have a Linux virtual machine hosted by Google. \n",
        "\n",
        "#@title ## Configuration\n",
        "class config: \n",
        "  connection = \"SSH only\" #@param [\"SSH only\", \"VNC\"]\n",
        "\n",
        "  class software: \n",
        "  #@markdown ##Software  \n",
        "    upgrade_all_binaries = 0 #@param {type:\"slider\", min:0, max:1, step:1}\n",
        "\n",
        "  class storage: \n",
        "    #@markdown ##Storage\n",
        "    #@markdown ### Google Drive Storage Integration\n",
        "    use = 0 #@param {type:\"slider\", min:0, max:1, step:1}\n",
        "    drive_root = 'My Drive' #@param ['My Drive', 'Shared drives'] \n",
        "    folder = \"\" #@param {type:\"string\"}\n",
        "    drive_folder = '/'.join((drive_root, folder)) # Actual Google Drive folder\n",
        "    mount_point = '' #@param {type:\"string\"}\n",
        "\n",
        "# actual runtime\n",
        "## Import modules. \n",
        "import os\n",
        "\n",
        "## Set formatting.\n",
        "class formatting:\n",
        "  clear = '\\033[0m'\n",
        "  class color: \n",
        "    purple = '\\033[95m'\n",
        "    cyan = '\\033[96m'\n",
        "    cyan_dark = '\\033[36m'\n",
        "    blue = '\\033[94m'\n",
        "    green = '\\033[92m'\n",
        "    yellow = '\\033[93m'\n",
        "    red = '\\033[91m'\n",
        "  class font:\n",
        "    bold = '\\033[1m'\n",
        "    underline = '\\033[4m'\n",
        "\n",
        "## Define screen functions. \n",
        "def clear(): \n",
        "  from IPython.display import clear_output; return clear_output()\n",
        "\n",
        "class info: # screen messages\n",
        "  def status(message):\n",
        "    print(formatting.font.bold + 'Status: \\t' + formatting.clear + message)\n",
        "  def err(message):\n",
        "    print(formatting.font.bold + formatting.color.red + 'Error: \\t' + formatting.clear + formatting.clear + message)\n",
        "  def warning(message):\n",
        "    print(formatting.font.bold + formatting.color.yellow + 'Warning: \\t' + formatting.clear + formatting.clear + message)\n",
        "  def success(message):\n",
        "    print(formatting.font.bold + formatting.color.green + 'Success: \\t' + formatting.clear + formatting.clear + message)\n",
        "\n",
        "\n",
        "#try:\n",
        "def mount_GDrive():\n",
        "  ## Configure storage. \n",
        "  info.status('Google Drive integration has been enabled! To properly integrate to your Google Drive, please accept the Google Account permission prompts.')\n",
        "  \n",
        "  try:\n",
        "    from google.colab import drive\n",
        "    info.status('Connecting to Google Drive…')\n",
        "    drive.mount('/content/drive')\n",
        "  \n",
        "  except:\n",
        "    config.storage.use = 0\n",
        "    info.error('Google Drive could not be connected.')\n",
        "    info.status('Google Drive integration has been disabled for this instance. ')\n",
        "\n",
        "  else: \n",
        "    info.success('Google Drive has been connected.')\n",
        "    clear()\n",
        "\n",
        "# Upgrade necessary components. \n",
        "def update_components():\n",
        "  os.system('add-apt-repository ppa:apt-fast/stable -y && apt update && apt install apt-fast -y && apt-fast clean && apt-fast autoremove -y')\n",
        "\n",
        "  if config.software.upgrade_all_binaries:\n",
        "    info.status('Upgrading all binaries.')\n",
        "    try:\n",
        "      if os.system('apt-fast update && apt-fast upgrade -y && apt-fast clean && apt-fast autoremove -y') > 0: raise\n",
        "    except:\n",
        "      info.err('An exception has been raised when trying to upgrade one or more of the binaries.')\n",
        "    else: \n",
        "      info.success('All binaries have been upgraded!')\n",
        "\n",
        "def install_remoColab():\n",
        "  # Install RemoColab. \n",
        "  info.status('Installing RemoColab…')\n",
        "  try:\n",
        "    os.system('pip install git+https://github.com/demotomohiro/remocolab.git')\n",
        "    import remocolab\n",
        "  except:\n",
        "    info.err('Could not install RemoColab.')\n",
        "    os.kill(os.getpid(), 1)\n",
        "  else:  \n",
        "    info.success('Successfully installed RemoColab.')\n",
        "\n",
        "def establish_connection():\n",
        "  import remocolab\n",
        "\n",
        "  def install_wm():\n",
        "    # install xfce. \n",
        "    info.status('Installing XFCE.')\n",
        "    try:\n",
        "      if os.system('apt-fast install xfce4 xfce4-terminal -y') > 0: \n",
        "        if os.system('apt install xfce4 xfce4-terminal -y') > 0: raise\n",
        "    except:\n",
        "      info.err('Could not install the XFCE desktop.')\n",
        "      config.connection = 'SSH only'; info.status('VNC connection will be disabled.')\n",
        "    else:\n",
        "      info.success('Successfully installed the XFCE desktop.')\n",
        "\n",
        "  if config.connection == 'VNC': install_wm()\n",
        "\n",
        "  info.status('Setting up ' + config.connection + ' connection. ')\n",
        "  if config.connection == 'VNC':\n",
        "    if config.storage.use == 1:\n",
        "      if config.storage.mount_point != None:\n",
        "        remocolab.setupVNC(tunnel = \"argotunnel\", mount_gdrive_from = config.storage.drive_folder, mount_gdrive_to = 'Google Drive')\n",
        "      else:  \n",
        "        remocolab.setupVNC(tunnel = \"argotunnel\", mount_gdrive_from = config.storage.drive_folder, mount_gdrive_to = 'mount_point')\n",
        "    else:\n",
        "      remocolab.setupVNC(tunnel = \"argotunnel\")\n",
        "    info.success('The VNC connection is available at ' + formatting.font.bold + 'vnc://localhost:5901' + formatting.clear + ' unless stated otherwise.')\n",
        "  else:\n",
        "    if config.storage.use == 1:\n",
        "      try:\n",
        "        if config.storage.mount_point != None:\n",
        "          remocolab.setupSSHD(tunnel = \"argotunnel\", mount_gdrive_from = config.storage.drive_folder, mount_gdrive_to = 'Google Drive')\n",
        "        else:  \n",
        "          remocolab.setupSSHD(tunnel = \"argotunnel\", mount_gdrive_from = config.storage.drive_folder, mount_gdrive_to = 'mount_point')\n",
        "      except:\n",
        "        remocolab.setupSSHD(tunnel = \"argotunnel\")\n",
        "    else:\n",
        "      remocolab.setupSSHD(tunnel = \"argotunnel\")\n",
        "\n",
        "def main(): \n",
        "  if config.storage.use: mount_GDrive()\n",
        "  update_components()\n",
        "  install_remoColab()\n",
        "  establish_connection()\n",
        "\n",
        "main()"
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "collapsed_sections": [],
      "name": "Desktop@CoLab.ipynb",
      "provenance": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3 (ipykernel)",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "name": "python",
      "version": "3.10.4"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}