#!/bin/bash

CheckStartupType() {
  # echo "Checking process autostart system..."
  starttype=`ps -p 1 | awk '/1/ {print $4}'`
  # Systemd
  if [[ $starttype == 'systemd' ]]; then return 1;
  elif [[ $starttype == 'init' ]];
      then
         # Upstart
         if [[ `/sbin/init --version` =~ upstart ]]; then return 2;
         # Sysv-init
         return 3;
      fi
  fi
  return 0;
}

# Add "StartupType=(type)" to .msh file
UpdateMshFile() {
  # Remove all lines that start with "StartupType="
  sed '/^StartupType=/ d' < /usr/local/mesh/meshagent.msh >> /usr/local/mesh/meshagent2.msh
  # Add the startup type to the file
  echo "StartupType=$starttype" >> /usr/local/mesh/meshagent2.msh
  mv /usr/local/mesh/meshagent2.msh /usr/local/mesh/meshagent.msh
}

CheckInstallAgent() {
  # echo "Checking mesh identifier..."
  if [ -e "/usr/local" ]
  then
    installpath="/usr/local/mesh"
  else
    installpath="/usr/mesh"
  fi
  if [ $# -ge 2 ]
  then
    url=$1
    meshid=$2
    meshidlen=${#meshid}
    if [ $meshidlen -eq 64 ]
    then
      machineid=0
      machinetype=$( uname -m )

      # If we have 3 arguments...
      if [ $# -ge 3 ]
      then
        # echo "Computer type is specified..."
        machineid=$3
      else
        # echo "Detecting computer type..."
        if [ $machinetype == 'x86_64' ] || [ $machinetype == 'amd64' ]
        then
          # Linux x86, 64 bit
          machineid=6
        fi
        if [ $machinetype == 'x86' ] || [ $machinetype == 'i686' ] || [ $machinetype == 'i586' ]
        then
          # Linux x86, 32 bit
          machineid=5
        fi
        if [ $machinetype == 'armv6l' ] || [ $machinetype == 'armv7l' ]
        then
          # RaspberryPi 1 (armv6l) or RaspberryPi 2/3 (armv7l)
          machineid=25
        fi
        if [ $machinetype == 'aarch64' ]
        then
          # RaspberryPi 3B+ running Ubuntu 64 (aarch64)
          machineid=26
        fi
        # Add more machine types, detect KVM support... here.
      fi

      if [ $machineid -eq 0 ]
      then
        echo "Unsupported machine type: $machinetype."
      else
        DownloadAgent $url $meshid $machineid
      fi

    else
      echo "MeshID is not correct, must be 64 characters long."
    fi
  else
    echo "URI and/or MeshID have not been specified, must be passed in as arguments."
    return 0;
  fi
}

DownloadAgent() {
  url=$1
  meshid=$2
  machineid=$3
  # Create folder
  mkdir -p /usr/local/mesh
  cd /usr/local/mesh
  echo "Downloading Mesh agent #$machineid..."
  wget $url/meshagents?id=$machineid --no-check-certificate {{{noproxy}}}-O /usr/local/mesh/meshagent

  # If it did not work, try again using http
  if [ $? != 0 ]
  then
    url=${url/"https://"/"http://"}
    wget $url/meshagents?id=$machineid {{{noproxy}}}-O /usr/local/mesh/meshagent
  fi

  if [ $? -eq 0 ]
  then
    echo "Mesh agent downloaded."
    # TODO: We could check the meshagent sha256 hash, but best to authenticate the server.
    chmod 755 /usr/local/mesh/meshagent
    wget $url/meshsettings?id=$meshid --no-check-certificate {{{noproxy}}}-O /usr/local/mesh/meshagent.msh

    # If it did not work, try again using http
    if [ $? -ne 0 ]
    then
      wget $url/meshsettings?id=$meshid {{{noproxy}}}-O /usr/local/mesh/meshagent.msh
    fi

    if [ $? -eq 0 ]
    then
      UpdateMshFile
      if [ $starttype -eq 1 ]
      then
        # systemd
        if [ -d "/lib/systemd/system/" ]
        then
            echo -e "[Unit]\nDescription=MeshCentral Agent\n[Service]\nExecStart=/usr/local/mesh/meshagent\nStandardOutput=null\nRestart=always\nRestartSec=3\n[Install]\nWantedBy=multi-user.target\nAlias=meshagent.service\n" > /lib/systemd/system/meshagent.service
        else
            # Some distros have the systemd folder at a different place
            if [ -d "/usr/lib/systemd/system/" ]
            then
                echo -e "[Unit]\nDescription=MeshCentral Agent\n[Service]\nExecStart=/usr/local/mesh/meshagent\nStandardOutput=null\nRestart=always\nRestartSec=3\n[Install]\nWantedBy=multi-user.target\nAlias=meshagent.service\n" > /usr/lib/systemd/system/meshagent.service
            else
                echo "Unable to find systemd folder."
            fi
        fi
        systemctl enable meshagent
        systemctl start meshagent
        echo 'meshagent installed as systemd service.'
        echo 'To start service: sudo systemctl start meshagent'
        echo 'To stop service: sudo systemctl stop meshagent'
      elif [ $starttype -eq 3 ]
          then
          # initd
          wget $url/meshagents?script=2 --no-check-certificate {{{noproxy}}}-O /etc/init.d/meshagent
          chmod +x /etc/init.d/meshagent
          # creates symlinks for rc.d
          update-rc.d meshagent defaults
          service meshagent start
          echo 'meshagent installed as init.d service.'
          echo 'To start service: sudo service meshagent start'
          echo 'To stop service: sudo service meshagent stop'
      elif [ $starttype -eq 2 ]
          then
          # upstart
          echo -e "start on runlevel [2345]\nstop on runlevel [016]\n\nrespawn\n\nchdir /usr/local/mesh\nexec /usr/local/mesh/meshagent\n\n" > /etc/init/meshagent.conf
          initctl start meshagent
          echo 'meshagent installed as upstart/init.d service.'
          echo 'To start service: sudo initctl start meshagent'
          echo 'To stop service: sudo initctl stop meshagent'
      else
          # unknown
          echo "Unknown Service Platform Type. (ie: init, systemd, etc)"
          echo "Unable to install meshagent as service."
      fi
      echo "Mesh agent started."
    else
      echo "Unable to download mesh settings at: $url/meshsettings?id=$meshid."
    fi
  else
    echo "Unable to download mesh agent at: $url/meshagents?id=$machineid."
  fi
}

UninstallAgent() {
# Uninstall agent
  if [ -e "/usr/local" ]
  then
    installpath="/usr/local/mesh"
  else
    installpath="/usr/mesh"
  fi

  if [ $starttype -eq 1 ]
  then
    # systemd
    systemctl disable meshagent
    systemctl stop meshagent
    rm -f /sbin/meshcmd /lib/systemd/system/meshagent.service
    systemctl stop meshagentDiagnostic &> /dev/null
    rm -f /lib/systemd/system/meshagentDiagnostic.service &> /dev/null
  else
    if [ $starttype -eq 3 ]; then
        # initd
        service meshagent stop
        update-rc.d -f meshagent remove
        rm -f /sbin/meshcmd /etc/init.d/meshagent
        service meshagentDiagnostic stop &> /dev/null
        rm -f /etc/init.d/meshagentDiagnostic &> /dev/null
    elif [ $starttype -eq 2 ]; then
        # upstart 
        initctl stop meshagent
        rm -f /sbin/meshcmd 
        rm -f /etc/init/meshagent.conf
        rm -f /etc/rc2.d/S20mesh /etc/rc3.d/S20mesh /etc/rc5.d/S20mesh
        initctl stop meshagentDiagnostic &> /dev/null
        rm -f /etc/init/meshagentDiagnostic.conf &> /dev/null
    fi
  fi

  if [ -e $installpath ]
  then
    rm -rf $installpath/*
    rmdir $installpath
  fi
  rm -rf /usr/local/mesh_services/meshagentDiagnostic &> /dev/null
  rm -f /etc/cron.d/meshagentDiagnostic_periodicStart &> /dev/null
  echo "Agent uninstalled."
}


CheckStartupType
starttype=$?
#echo "Type: $starttype"

currentuser=$( whoami )
if [ $currentuser == 'root' ]
then
  if [ $# -eq 0 ]
  then
    echo -e "This script will install or uninstall a mesh agent, usage:\n  $0 [serverurl] [meshid] (machineid)\n  $0 uninstall"
  else
    if [ $# -eq 1 ]
    then
      if [ $1 == 'uninstall' ] || [ $1 == 'UNINSTALL' ]
      then
        UninstallAgent
      fi
    else
      UninstallAgent
      CheckInstallAgent $1 $2 $3
    fi
  fi
else
  echo "Must be root to install or uninstall mesh agent."
fi
