Add output format option

Fixes #10

Co-Authored-by: chksome <[email protected]>

Author: mame

README.md

@@ -22,14 +22,28 @@ chmod 755 wsl2-ssh-agent
 
 **If you are using ArchLinux, you can install the [wsl2-ssh-agent](https://aur.archlinux.org/packages/wsl2-ssh-agent) package from the AUR (maintained by @Hill-98).**
 
-### 2. Modify `.bashrc` (or `.zshrc` if you are using `zsh`)
+### 2. Modify your shell's rc file
+
+#### bash or zsh
 
 Add the following line to `.bashrc` (or `.zshrc` if you are using `zsh`).
 
 ```
 eval $($HOME/wsl2-ssh-agent)
 ```
 
+#### fish
+
+Add the following lines to `config.fish`
+
+```
+if status is-login
+  $HOME/wsl2-ssh-agent | source
+end
+```
+
+### 3. Reopen your terminal
+
 Close and reopen the terminal and execute `ssh your-machine`.
 The command should communicate with ssh-agent.exe service.
 

config.go

@@ -12,12 +12,14 @@ import (
 	"os/exec"
 	"os/signal"
 	"path/filepath"
+	"strings"
 	"syscall"
 )
 
 type config struct {
 	socketPath     string
 	powershellPath string
+	format         string
 	foreground     bool
 	verbose        bool
 	stop           bool
@@ -76,6 +78,31 @@ func newConfig() *config {
 	return c
 }
 
+func getOutputFormat(format string) string {
+	if format == "auto" {
+		shell := os.Getenv("SHELL")
+		if strings.HasSuffix(shell, "fish") {
+			format = "fish"
+		} else if strings.HasSuffix(shell, "csh") {
+			format = "csh"
+		} else {
+			format = "sh"
+		}
+	}
+	switch format {
+	case "sh", "bash", "zsh":
+		return "SSH_AUTH_SOCK=%s; export SSH_AUTH_SOCK;"
+	case "csh", "tcsh":
+		return "setenv SSH_AUTH_SOCK %s"
+	case "fish":
+		return "set -x SSH_AUTH_SOCK %s"
+	default:
+		fmt.Printf("output format must be auto, bash, zsh, csh, tcsh, or fish\n")
+		os.Exit(1)
+		return ""
+	}
+}
+
 func (c *config) start() context.Context {
 	if c.version {
 		fmt.Printf("wsl2-ssh-agent %s\n", version)
@@ -86,7 +113,7 @@ func (c *config) start() context.Context {
 	parent := checkDaemonMode()
 
 	// script output
-	output := fmt.Sprintf("SSH_AUTH_SOCK=%s; export SSH_AUTH_SOCK;", c.socketPath)
+	output := fmt.Sprintf(getOutputFormat(c.format), c.socketPath)
 
 	// set up the log file
 	c.setupLogFile()