main.sh

#!/bin/bash

scrcpyflags=(--legacy-paste --turn-screen-off --stay-awake --power-off-on-close)

error() { #red text and exit 1
  [ ! -z "$1" ] && echo -e "\e[91m$1\e[0m" 1>&2
  
  local display_error=yes
  if [ ! -z $YAD_PID ];then
    #don't popup the error if the main window was running but the process does not exist (prevents multiple error dialogs)
    kill $YAD_PID 2>/dev/null || display_error=no
  fi
  [ "$display_error" == yes ] && setsid yad "${yadflags[@]}" --text="AndroidBuddy encountered an error <b>:(</b>"$'\n'"$1" --button=OK:0 &
  exit 1
}

sudo_popup() { #just like sudo on passwordless systems like PiOS, but displays a password dialog otherwise. Avoids displaying a password prompt to an invisible terminal.
  if sudo -n true; then
    # sudo is available (within sudo timer) or passwordless
    sudo "$@"
  else
    # sudo is not available (not within sudo timer)
    pkexec "$@"
  fi
}

process_exists() { #return 0 if the $1 PID is running, otherwise 1
  [ -z "$1" ] && error "process_exists(): no PID given!"
  
  if [ -f "/proc/$1/status" ];then
    return 0
  else
    return 1
  fi
}

install_gnirehtet() {
  if [ ! -d /opt/gnirehtet ] || [ ! -f /usr/bin/gnirehtet ];then
    if ! command -v cargo >/dev/null ;then
      sudo_popup apt update
      sudo_popup apt install -y cargo || error "cargo is required but it failed to install."
    fi
    
    if ! command -v rustc >/dev/null ;then
      sudo_popup apt update
      sudo_popup apt install -y rustc || error "rustc is required but it failed to install."
    fi
    
    #get apk
    rm -rf /tmp/gnirehtet-java
    wget -O /tmp/gnirehtet-java.zip 'https://github.com/Genymobile/gnirehtet/releases/download/v2.5.1/gnirehtet-java-v2.5.1.zip' || error "Failed to download gnirehtet-java-v2.5.1.zip"
    unzip /tmp/gnirehtet-java.zip -d /tmp/gnirehtet-java || error "Failed to extract gnirehtet-java-v2.5.1.zip"
    sudo_popup rm -rf /opt/gnirehtet || error "Failed to remove /opt/gnirehtet"
    sudo_popup mkdir -p /opt/gnirehtet
    sudo_popup mv /tmp/gnirehtet-java/gnirehtet-java/gnirehtet.apk /opt/gnirehtet || error "Failed to move gnirehtet.apk to /opt/gnirehtet"
    rm -rf /tmp/gnirehtet-java
    
    #compile gnirehtet command
    rm -rf /tmp/gnirehtet
    cd /tmp
    git clone https://github.com/Genymobile/gnirehtet || error "Failed to download gnirehtet github repository"
    cd /tmp/gnirehtet/relay-rust
    cargo build --release || error "Failed to build gnirehtet with cargo and rust"
    cd
    sudo_popup mv /tmp/gnirehtet/relay-rust/target/release/gnirehtet /opt/gnirehtet || error "Failed to move gnirehtet to /opt/gnirehtet"
    rm -rf /tmp/gnirehtet
    
    #make command
    echo '#!/bin/bash
cd /opt/gnirehtet
/opt/gnirehtet/gnirehtet "$@"' > /tmp/gnirehtet
    chmod +x /tmp/gnirehtet || error "Failed to mark /tmp/gnirehte as executable!"
    sudo_popup mv -f /tmp/gnirehtet /usr/bin/gnirehtet || error "Failed to move /tmp/gnirehte to /usr/bin"
  fi
  
  if [ "$gnirehtet_installed" == false ];then
    gnirehtet install || error "Failed to install the gnirehtet.apk on your phone!"
    gnirehtet_installed=true
  fi
}

wait_for_reconnect() {
  for i in {1..20} ;do
    if adb shell pm list packages &>/dev/null ;then
      return 0
    fi
    sleep 0.1
  done
  return 1
}

scrcpy_daemon() {
  if ! command -v scrcpy >/dev/null ;then
    error "scrcpy not installed! If you are using an ARM computer, please use Pi-Apps to install scrcpy. https://github.com/Botspot/pi-apps"
  fi
  
  local exitcode
  local tries=0
  while true;do
    scrcpy "$@" &>/dev/null &
    local pid=$!
    trap "kill $pid 2>/dev/null" EXIT
    wait $pid
    exitcode=$?
    if [ $exitcode == 2 ] ;then
      tries=0
      #give the device 2 seconds to reconnect
      if ! wait_for_reconnect ;then
        kill $mypid
        error "phone disconnected"
      fi
    elif [ $exitcode == 0 ];then
      #user closed scrcpy
      break
    else
      #exit code was 1
      tries=$((tries+1))
      [ $tries == 5 ] && error "scrcpy exited unexpectedly. Perhaps the phone disconnected."
      wait_for_reconnect || error "phone disconnected"
    fi
  done
}

yadflags=(--center --class=androidbuddy --name=androidbuddy --window-icon="$(dirname "$0")/logo.png" --title="AndroidBuddy")

if [ ! -f ~/.local/share/applications/androidbuddy.desktop ];then
  if [ -f /usr/share/applications/androidbuddy.desktop ];then
    sudo_popup rm -f /usr/share/applications/androidbuddy.desktop
  fi
  echo "[Desktop Entry]
Name=AndroidBuddy
Comment=GUI for Android device recovery and maintainence
Exec="\""$(dirname $0)/main.sh"\""
Icon=androidbuddy
StartupWMClass=androidbuddy
Terminal=false
Type=Application
Hidden=false
Categories=Utility;" > ~/.local/share/applications/androidbuddy.desktop
fi

if [ ! -f ~/.local/share/icons/androidbuddy.png ];then
  mkdir -p ~/.local/share/icons || error "failed to make icons directory!"
  cp "$(dirname "$0")/logo.png" ~/.local/share/icons/androidbuddy.png || error "failed to copy icon!"
  update-icon-caches ~/.local/share/icons
fi

if ! command -v yad >/dev/null ;then
  sudo_popup apt update
  sudo_popup apt install -y yad || error "yad is required but it failed to install."
fi

if ! command -v adb >/dev/null ;then
  sudo_popup apt update
  sudo_popup apt install -y adb || error "adb is required but it failed to install."
fi

localhash="$(cd "$(dirname "$0")" ; git rev-parse HEAD)"
latesthash="$(git ls-remote https://github.com/Botspot/androidbuddy HEAD | awk '{print $1}')"
if [ "$localhash" != "$latesthash" ] && [ ! -z "$latesthash" ] && [ ! -z "$localhash" ];then
  echo "Auto-updating AndroidBuddy for the latest features and improvements..."
  cd "$(dirname "$0")"
  git pull | cat #piping through cat makes git noninteractive
  
  if [ "${PIPESTATUS[0]}" == 0 ];then
    cd
    echo "git pull finished. Reloading script..."
    set -a #export all variables so the script can see them
    #run updated script
    "$0" "$@"
    exit $?
  else
    cd
    echo "git pull failed. Continuing..."
  fi
fi

#install gnirehtet then exit now if given the 'install' flag (for pi-apps integration)
if [ "$1" == install ];then
  install_gnirehtet || exit 1
  exit 0
fi

read_true() {
  [ "$exitcode" == first ] && return 0
  read "$@"
  true
}
#wait for phone to be plugged in
exitcode=first
yad "${yadflags[@]}" --text="To continue, connect an Android phone to this computer with a USB cable."$'\n'"Waiting for phone..." --progress --pulsate --width 400 --no-buttons < <(echo '#';sleep infinity) &
loaderpid=$!
trap "kill $loaderpid 2>/dev/null" EXIT
line=''
yaderrpid=''
while read_true -t 5 line ;do
  echo "received '$line'"
  
  installed_apks="$(adb shell pm list packages 2>/dev/null)"
  exitcode=$?
  if [ ! -z "$line" ] && [ "$exitcode" != 0 ];then
    #phone detected, but adb failed - give it 1 second
    sleep 1
    if installed_apks="$(adb shell pm list packages 2>/dev/null)" ;then
      kill "$loaderpid" 2>/dev/null
      kill "$yaderrpid" 2>/dev/null
      break
    elif [ -z "$yaderrpid" ];then
      #still failed, adb not enabled, user not yet informed what to do
      kill "$loaderpid" 2>/dev/null
      echo 'Android phone detected, but you need to enable USB debugging.
1. On your phone, go to Settings > "About phone"
2. Tap "Software information" if you see it
3. Tap "Build number" several times until developer mode is enabled
4. Go back to Settings and tap "Developer options", then enable "USB debugging"' | yad "${yadflags[@]}" --text-info --width 500 --height=200 --wrap --fontname="" --no-buttons &
      yaderrpid=$!
      trap "kill $yaderrpid 2>/dev/null" EXIT
    fi
  elif [ "$exitcode" == 0 ];then
    kill "$yaderrpid" 2>/dev/null
    kill "$loaderpid" 2>/dev/null
    break
  elif [ ! -f "/proc/$loaderpid/status" ] && [ ! -f "/proc/$yaderrpid/status" ];then
    echo "exiting, as loader window was closed"
    exit 0
  fi
done < <(udevadm monitor --environment | grep --line-buffered '^ID_SERIAL.*Android')

gnirehtet_installed="$(grep -xFq "package:com.genymobile.gnirehtet" <<<"$installed_apks" && echo true || echo false)"

mypid=$$

while read -r input; do
  echo "Received '$input'"
  
  if [ -z "$YAD_PID" ];then
    YAD_PID="$(echo "$input" | awk '{print $2}')"
  fi
  input="$(echo "$input" | awk '{print $1}')"
  
  case "$input" in
    scrcpy)
      #toggle scrcpy (kill it if found, otherwise run it)
      if [ ! -z "$scrcpy_daemon_pid" ] && process_exists "$scrcpy_daemon_pid" ;then
        echo "scrcpy running, killing it"
        kill "$scrcpy_daemon_pid"
      else
        scrcpy_daemon "${scrcpyflags[@]}" &
        scrcpy_daemon_pid=$!
      fi
      ;;
    gnirehtet)
        install_gnirehtet
        #make sure it is not running on host or guest already
        killall gnirehtet 2>/dev/null
        adb shell am force-stop com.genymobile.gnirehtet
        gnirehtet run &
        gnirehtet_pid=$!
        trap "kill $gnirehtet_pid 2>/dev/null" EXIT
        #give it 2 seconds to stop running and exit with error
        sleep 2
        if ! process_exists $gnirehtet_pid && ! wait $gnirehtet_pid ;then
          error "failed to run gnirehtet for reverse tethering"
        fi
      ;;
    tethering)
      adb shell settings put global tether_dun_required 0 || error "adb command failed"
      adb shell svc usb setFunctions rndis || error "adb command failed"
      ;;
    mtp)
      if [ ! -f /usr/libexec/gvfs-mtp-volume-monitor ];then
        sudo_popup apt update
        sudo_popup apt install -y gvfs-backends || error "gvfs-backends is required but it failed to install."
      fi
      adb shell svc usb setFunctions mtp || error "adb command failed"
      wait_for_reconnect || error "phone disconnected"
      
      #start mtp volume monitor if needed
      pgrep -f /usr/libexec/gvfs-mtp-volume-monitor >/dev/null || /usr/libexec/gvfs-mtp-volume-monitor &
      
      volmonpid=$!
      tries=0
      while [ ! -d /run/user/1000/gvfs/*/Phone ];do
        if [ "$tries" -lt 20 ];then
          sleep 0.2
        elif [ "$tries" -eq 20 ];then
          echo "Phone needs you to allow access to its files. Check the phone for a popup and tap \"Allow\"."
        else
          #wait for user to allow file access. Once permitted, the phone disables mtp, so enable it again
          adb shell dumpsys usb 2>/dev/null | grep -q 'kernel_function_list.*mtp' || adb shell svc usb setFunctions mtp 2>/dev/null
          sleep 1
        fi
        tries=$((tries+1))
        [ "$tries" == 20 ] && echo
      done
      #don't make the double-mount show up in file manager
      
      xdg-open /run/user/1000/gvfs/*/Phone || error "failed to open phone filesystem in file manager"
      ;;
    quickdrop)
      files="$(yad "${yadflags[@]}" --dnd --width=400 --text="Drop files here to send them to the Download folder." --text-align=center --button=OK:0 | sed 's+file://++g')"
      IFS=$'\n'
      lastfile="$(tail -1 <<<"$files")"
      for file in $files ;do
        echo "#$(basename "$file")"
        errors="$(adb push "$file" /sdcard/Download 2>&1)"
        [ $? != 0 ] && error "failed to transfer '$file' errors below:"$'\n'"$errors"
        #pause on last file so user can see progress
        if [ "$file" == "$lastfile" ];then
          echo "#Success!"
          sleep 2
        fi
      done | yad "${yadflags[@]}" --text="Transferring files..." --progress --pulsate --auto-close --auto-kill --enable-log --log-expanded --width 400 --button=Cancel:1
      
      [ "${PIPESTATUS[0]}" != 0 ] && error
      ;;
    apkinstall)
      files="$(yad "${yadflags[@]}" --dnd --width=400 --text="Drop APK files here to install them to the phone." --text-align=center --button=OK:0 | sed 's+file://++g')"
      IFS=$'\n'
      lastfile="$(tail -1 <<<"$files")"
      for file in $files ;do
        echo "#$(basename "$file")"
        errors="$(adb install "$file" 2>&1)"
        [ $? != 0 ] && error "failed to install '$file' errors below:"$'\n'"$errors"
        #pause on last file so user can see progress
        if [ "$file" == "$lastfile" ];then
          echo "#Success!"
          sleep 2
        fi
      done | yad "${yadflags[@]}" --text="Transferring files..." --progress --pulsate --auto-close --auto-kill --enable-log --log-expanded --width 400 --button=Cancel:1
      
      [ "${PIPESTATUS[0]}" != 0 ] && error
      ;;
    autostart)
      if [ -f ~/.config/autostart/androidbuddy.desktop ];then
        rm -f ~/.config/autostart/androidbuddy.desktop
        ps aux | grep "/bin/bash $(dirname $0)/autostart.sh" | grep -v grep | awk '{print $2}' | xargs kill
        yad "${yadflags[@]}" --width=400 --text="Removed the autostart file."$'\n'"From now on, AndroidBuddy will no longer launch when an Android phone is plugged in." --button=OK:0
      else
        mkdir -p ~/.config/autostart
        echo "[Desktop Entry]
Name=AndroidBuddy USB service
Exec="\""$(dirname $0)/autostart.sh"\""
Terminal=false
Type=Application
X-GNOME-Autostart-enabled=true
Hidden=false
NoDisplay=false" > ~/.config/autostart/androidbuddy.desktop
        ps aux | grep "/bin/bash $(dirname $0)/autostart.sh" | grep -v grep | awk '{print $2}' | xargs kill
        setsid "$(dirname $0)/autostart.sh" &
        yad "${yadflags[@]}" --width=400 --text="Added the autostart file."$'\n'"From now on, AndroidBuddy will launch when an Android phone is plugged in." --button=OK:0
        kill $(echo "$input" | awk '{print $2}') #kill yad and exit
        break
      fi
      ;;
    *)
      error "unknown input '$input'"
      ;;
  esac
done < <(yad "${yadflags[@]}" --form --no-buttons \
  --field=$'Control screen!!View and interact with phone screen using scrcpy.\nCopy and paste should transfer automatically, and phone'\''s audio is forwarded to your computer.':FBTN 'bash -c "echo scrcpy $YAD_PID"' \
  --field='Share internet to phone!!Share this computer'\''s internet connection with the phone (reverse tethering)':FBTN 'bash -c "echo gnirehtet $YAD_PID"' \
  --field='Use phone'\''s internet!!Have this computer use the phone'\''s mobile network or WiFi connection (tethering)':FBTN 'bash -c "echo tethering $YAD_PID"' \
  --field='Browse phone'\''s files!!View the filesystem of the phone in a file manager':FBTN 'bash -c "echo mtp $YAD_PID"' \
  --field='Send files!!Quick-drop files into the phone'\''s Download folder':FBTN 'bash -c "echo quickdrop $YAD_PID"' \
  --field='Install APK!!Install an Android app from this computer to the phone':FBTN 'bash -c "echo apkinstall $YAD_PID"' \
  --field='Run when phone found!!Launch AndroidBuddy when an Android phone is plugged in':FBTN 'bash -c "echo autostart $YAD_PID"')