GitBash and MinGW support

.gitignore

@@ -2,3 +2,4 @@ vendor
 bin
 .vscode
 sshcode
+sshcode.exe

main.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"math/rand"
 	"os"
+	"runtime"
 	"strings"
 	"time"
 
@@ -80,6 +81,11 @@ func (c *rootCmd) Run(fl *pflag.FlagSet) {
 		dir = "~"
 	}
 
+	// Get linux relative path if on windows.
+	if runtime.GOOS == "windows" {
+		dir = gitbashWindowsDir(dir)
+	}
+
 	err := sshCode(host, dir, options{
 		skipSync:         c.skipSync,
 		sshFlags:         c.sshFlags,

settings.go

@@ -24,6 +24,8 @@ func configDir() (string, error) {
 		path = os.ExpandEnv("$HOME/.config/Code/User/")
 	case "darwin":
 		path = os.ExpandEnv("$HOME/Library/Application Support/Code/User/")
+	case "windows":
+		return os.ExpandEnv("/c/Users/$USERNAME/AppData/Roaming/Code/User"), nil
 	default:
 		return "", xerrors.Errorf("unsupported platform: %s", runtime.GOOS)
 	}
@@ -39,8 +41,18 @@ func extensionsDir() (string, error) {
 	switch runtime.GOOS {
 	case "linux", "darwin":
 		path = os.ExpandEnv("$HOME/.vscode/extensions/")
+	case "windows":
+		return os.ExpandEnv("/c/Users/$USERNAME/.vscode/extensions"), nil
 	default:
 		return "", xerrors.Errorf("unsupported platform: %s", runtime.GOOS)
 	}
 	return filepath.Clean(path), nil
 }
+
+func windowsVarFix(input string) string {
+	if runtime.GOOS == "windows" {
+		return input
+	}
+
+	return "~/" + input
+}

sshcode.go

@@ -3,13 +3,17 @@ package main
 import (
 	"context"
 	"fmt"
+	"log"
 	"math/rand"
 	"net"
 	"net/http"
 	"os"
 	"os/exec"
 	"os/signal"
 	"path/filepath"
+	"regexp"
+	"runtime"
+	"sort"
 	"strconv"
 	"strings"
 	"syscall"
@@ -106,7 +110,6 @@ func sshCode(host, dir string, o options) error {
 
 		// Downloads the latest code-server and allows it to be executed.
 		sshCmdStr := fmt.Sprintf("ssh %v %v '/usr/bin/env bash -l'", o.sshFlags, host)
-
 		sshCmd := exec.Command("sh", "-l", "-c", sshCmdStr)
 		sshCmd.Stdout = os.Stdout
 		sshCmd.Stderr = os.Stderr
@@ -145,10 +148,9 @@ func sshCode(host, dir string, o options) error {
 	flog.Info("Tunneling remote port %v to %v", o.remotePort, o.bindAddr)
 
 	sshCmdStr :=
-		fmt.Sprintf("ssh -tt -q -L %v:localhost:%v %v %v 'cd %v; %v --host 127.0.0.1 --auth none --port=%v'",
-			o.bindAddr, o.remotePort, o.sshFlags, host, dir, codeServerPath, o.remotePort,
+		fmt.Sprintf("ssh -tt -q -L %v:localhost:%v %v %v '%v  %v --host 127.0.0.1 --auth none --port=%v'",
+			o.bindAddr, o.remotePort, o.sshFlags, host, codeServerPath, dir, o.remotePort,
 		)
-
 	// Starts code-server and forwards the remote port.
 	sshCmd := exec.Command("sh", "-l", "-c", sshCmdStr)
 	sshCmd.Stdin = os.Stdin
@@ -266,9 +268,12 @@ func openBrowser(url string) {
 	const (
 		macPath = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
 		wslPath = "/mnt/c/Program Files (x86)/Google/Chrome/Application/chrome.exe"
+		winPath = "C:/Program Files (x86)/Google/Chrome/Application/chrome.exe"
 	)
 
 	switch {
+	case commandExists("chrome"):
+		openCmd = exec.Command("chrome", chromeOptions(url)...)
 	case commandExists("google-chrome"):
 		openCmd = exec.Command("google-chrome", chromeOptions(url)...)
 	case commandExists("google-chrome-stable"):
@@ -281,6 +286,8 @@ func openBrowser(url string) {
 		openCmd = exec.Command(macPath, chromeOptions(url)...)
 	case pathExists(wslPath):
 		openCmd = exec.Command(wslPath, chromeOptions(url)...)
+	case pathExists(winPath):
+		openCmd = exec.Command(winPath, chromeOptions(url)...)
 	default:
 		err := browser.OpenURL(url)
 		if err != nil {
@@ -335,6 +342,11 @@ func randomPort() (string, error) {
 // checkSSHDirectory performs sanity and safety checks on sshDirectory, and
 // returns a new value for o.reuseConnection depending on the checks.
 func checkSSHDirectory(sshDirectory string, reuseConnection bool) bool {
+	if runtime.GOOS == "windows" {
+		flog.Info("OS is windows, disabling connection reuse feature")
+		return false
+	}
+
 	sshDirectoryMode, err := os.Lstat(expandPath(sshDirectory))
 	if err != nil {
 		if reuseConnection {
@@ -451,7 +463,7 @@ func syncUserSettings(sshFlags string, host string, back bool) error {
 		return err
 	}
 
-	const remoteSettingsDir = "~/.local/share/code-server/User/"
+	var remoteSettingsDir = windowsVarFix(".local/share/code-server/User/")
 
 	var (
 		src  = localConfDir + "/"
@@ -477,7 +489,7 @@ func syncExtensions(sshFlags string, host string, back bool) error {
 		return err
 	}
 
-	const remoteExtensionsDir = "~/.local/share/code-server/extensions/"
+	var remoteExtensionsDir = windowsVarFix(".local/share/code-server/extensions/")
 
 	var (
 		src  = localExtensionsDir + "/"
@@ -524,7 +536,7 @@ func downloadScript(codeServerPath string) string {
 
 [ "$(uname -m)" != "x86_64" ] && echo "Unsupported server architecture $(uname -m). code-server only has releases for x86_64 systems." && exit 1
 pkill -f %v || true
-mkdir -p ~/.local/share/code-server %v
+mkdir -p $HOME/.local/share/code-server %v
 cd %v
 curlflags="-o latest-linux"
 if [ -f latest-linux ]; then
@@ -535,8 +547,8 @@ curl $curlflags https://codesrv-ci.cdr.sh/latest-linux
 ln latest-linux %v
 chmod +x %v`,
 		codeServerPath,
-		filepath.Dir(codeServerPath),
-		filepath.Dir(codeServerPath),
+		filepath.ToSlash(filepath.Dir(codeServerPath)),
+		filepath.ToSlash(filepath.Dir(codeServerPath)),
 		codeServerPath,
 		codeServerPath,
 		codeServerPath,
@@ -548,6 +560,11 @@ chmod +x %v`,
 func ensureDir(path string) error {
 	_, err := os.Stat(path)
 	if os.IsNotExist(err) {
+		// This fixes a issue where GO reads `/c/` as `C:\c\` and creates
+		// empty directories on HOST/CLIENT that dont need to exist
+		if runtime.GOOS == "windows" && strings.HasPrefix(path, "/c/") {
+			path = "C:" + path[2:]
+		}
 		err = os.MkdirAll(path, 0750)
 	}
 
@@ -608,3 +625,73 @@ func parseGCPSSHCmd(instance string) (ip, sshFlags string, err error) {
 
 	return strings.TrimSpace(userIP), sshFlags, nil
 }
+
+// gitbashMountPoints returns all mount points in a msys2 system and then trunicates
+// the mount point for `/` from the path provided (dir), this is done to fix an
+// issue with how msys2 provides file paths to go, example. When you feed a filepath
+// as user input to go, `command /opt` for example, msys2 will pass `C:\<path to msys2 directory>\opt`
+// This causes an issue whith this program where when you run `sshcode.exe user@server /opt`,
+// GO and msys2 will attempt to pass `C:\<path to msys2 directory>\opt` to the server.
+// This function is used to prevent this.
+func gitbashWindowsDir(dir string) string {
+
+	// if dir is left empty, line82:main.go will set it to `~`, this makes it so that
+	// if dir is `~`, return `~/` instead of continuing with the gitbashWindowsDir()
+	// function. this prevens windows from trying to send a litteral instead of a relative
+	if dir == "~" {
+		return "~/"
+	}
+
+	//If msys feeds a `C:` path, clense
+	if strings.HasPrefix(dir, "C:") {
+		mountPoints := gitbashMountPointsAndHome()
+
+		// Apply mount points
+		absDir, _ := filepath.Abs(dir)
+		absDir = filepath.ToSlash(absDir)
+		for _, mp := range mountPoints {
+			if strings.HasPrefix(absDir, mp[0]) {
+				resolved := strings.Replace(absDir, mp[0], mp[1], 1)
+
+				// Sometimes the resolved path can go from user input `/Workspace`
+				// to `//Workspace`, this if statement checks that and removes it
+				if strings.HasPrefix(resolved, "//") {
+					resolved = strings.TrimPrefix(resolved, "/")
+					flog.Info("Resolved windows path '%s' to '%s", dir, resolved)
+					return resolved
+				}
+
+				flog.Info("Resolved windows path '%s' to '%s", dir, resolved)
+				return resolved
+			}
+		}
+
+	}
+	return dir
+}
+
+// This function returns an array with MINGW64 mount points including relative home dir
+func gitbashMountPointsAndHome() [][]string {
+	mountPoints := [][]string{{filepath.ToSlash(os.Getenv("HOME")), "~"}}
+
+	// Load mount points
+	out, err := exec.Command("mount").Output()
+	if err != nil {
+		//log.Error(err)
+		log.Println(err)
+	}
+	lines := strings.Split(string(out), "\n")
+	var mountRx = regexp.MustCompile(`^(.*) on (.*) type`)
+	for _, line := range lines {
+		extract := mountRx.FindStringSubmatch(line)
+		if len(extract) > 0 {
+			mountPoints = append(mountPoints, []string{extract[1], extract[2]})
+		}
+	}
+
+	// Sort by size to get more restrictive mount points first
+	sort.Slice(mountPoints, func(i, j int) bool {
+		return len(mountPoints[i][0]) > len(mountPoints[j][0])
+	})
+	return mountPoints
+}