Add rc hook

Author: avindra

main.go

@@ -25,8 +25,15 @@ func main() {
 			cfg := dirp.FindDirs(arg0)
 			dirp.Selector(cfg)
 		} else if arg0 == "hook" {
-			if len(args) >= 2 && args[1] == "fish" {
-				dirp.PrintFishHook()
+			if len(args) >= 2 {
+				switch args[1] {
+				case "fish":
+					dirp.PrintFishHook()
+				case "rc":
+					dirp.PrintRcHook()
+				default:
+					panic("I don't know about " + args[1])
+				}
 			} else {
 				dirp.PrintHook()
 			}

src/hook.go

@@ -1,6 +1,9 @@
 package dirp
 
-import "fmt"
+import (
+	"fmt"
+	"strings"
+)
 
 // PrintHook emits shell code for Bash, ZSH, sh, BusyBox, etc
 func PrintHook() {
@@ -53,3 +56,25 @@ func PrintFishHook() {
 		pushd "$stdout"
 	end`)
 }
+
+// PrintRcHook emits code for rc, the plan 9 shell
+func PrintRcHook() {
+	src := `fn dir {
+		stdout=` + "`" + `{dirp $*};
+		if (~ $bqstatus 2 ) {
+			$EDITOR $stdout;
+			return $status;
+		};
+	
+		if (~ "x$stdout" "x" ) {
+			echo -n How are we doing @;
+			uptime;
+			return $status;
+		};
+	
+		echo Switching to $stdout;
+		cd $stdout
+	}`
+
+	fmt.Println(strings.ReplaceAll(src, "\t", "  "))
+}