hilbish.go

@@ -5,7 +5,6 @@ package main
 
 import (
 	"os"
-	"runtime"
 	"strings"
 
 	"github.com/pborman/getopt"
@@ -23,11 +22,7 @@ func HilbishLoader(L *lua.LState) int {
 	mod := L.SetFuncs(L.NewTable(), exports)
 
 	host, _ := os.Hostname()
-	username := curuser.Username
+	username := strings.Split(curuser.Username, "\\")[1] // for some reason Username includes the hostname on windows
-	// this will be baked into binary since GOOS is a constant
-	if runtime.GOOS == "windows" {
-		username = strings.Split(username, "\\")[1] // for some reason Username includes the hostname on windows
-	}
 
 	L.SetField(mod, "ver", lua.LString(version))
 	L.SetField(mod, "user", lua.LString(username))

main.go

@@ -2,20 +2,22 @@ package main
 
 import (
 	"bufio"
+	"context"
 	"fmt"
 	"io"
 	"os"
-	"runtime"
 	"strings"
 	"os/signal"
 	"os/user"
 	"path/filepath"
+	"time"
 
 	"hilbish/golibs/bait"
 
 	"github.com/pborman/getopt"
 	"github.com/yuin/gopher-lua"
 	"layeh.com/gopher-luar"
+	"mvdan.cc/sh/v3/interp"
 	"golang.org/x/term"
 )
 
@@ -31,6 +33,8 @@ var (
 
 	hooks bait.Bait
 	defaultConfPath string
+
+	runner *interp.Runner	
 )
 
 func main() {
@@ -149,6 +153,56 @@ func main() {
 		os.Exit(0)
 	}
 
+	// Execute handler for sh runner
+	exechandle := func(ctx context.Context, args []string) error {
+		hc := interp.HandlerCtx(ctx)
+		_, argstring := splitInput(strings.Join(args, " "))
+
+		// If alias was found, use command alias
+		if aliases[args[0]] != "" {
+			alias := aliases[args[0]]
+			argstring = alias + strings.TrimPrefix(argstring, args[0])
+			cmdArgs, _ := splitInput(argstring)
+			args = cmdArgs
+		}
+
+		// If command is defined in Lua then run it
+		if commands[args[0]] != nil {
+			err := l.CallByParam(lua.P{
+				Fn: commands[args[0]],
+				NRet:    1,
+				Protect: true,
+			}, luar.New(l, args[1:]))
+			luaexitcode := l.Get(-1)
+			var exitcode uint8 = 0
+
+			l.Pop(1)
+
+			if code, ok := luaexitcode.(lua.LNumber); luaexitcode != lua.LNil && ok {
+				exitcode = uint8(code)
+			}
+
+			if err != nil {
+				fmt.Fprintln(os.Stderr,
+					"Error in command:\n\n" + err.Error())
+			}
+			hooks.Em.Emit("command.exit", exitcode)
+			return interp.NewExitStatus(exitcode)
+		}
+
+		if _, err := interp.LookPathDir(hc.Dir, hc.Env, args[0]); err != nil {
+			hooks.Em.Emit("command.not-found", args[0])
+			return interp.NewExitStatus(127)
+		}
+
+		return interp.DefaultExecHandler(2 * time.Second)(ctx, args)
+	}
+	// Setup sh runner outside of input label
+	runner, _ = interp.New(
+		interp.StdIO(os.Stdin, os.Stdout, os.Stderr),
+		interp.ExecHandler(exechandle),
+	)
+
 input:
 	for interactive {
 		running = false
@@ -211,11 +265,7 @@ func fmtPrompt() string {
 	cwd, _ := os.Getwd()
 
 	cwd = strings.Replace(cwd, curuser.HomeDir, "~", 1)
-	username := curuser.Username
+	username := strings.Split(curuser.Username, "\\")[1] // for some reason Username includes the hostname on windows
-	// this will be baked into binary since GOOS is a constant
-	if runtime.GOOS == "windows" {
-		username = strings.Split(username, "\\")[1] // for some reason Username includes the hostname on windows
-	}
 
 	args := []string{
 		"d", cwd,

shell.go

@@ -5,7 +5,6 @@ import (
 	"fmt"
 	"os"
 	"strings"
-	"time"
 
 //	"github.com/bobappleyard/readline"
 	"github.com/yuin/gopher-lua"
@@ -114,54 +113,7 @@ func execCommand(cmd string) error {
 	if err != nil {
 		return err
 	}
-
+	
-	exechandle := func(ctx context.Context, args []string) error {
-		hc := interp.HandlerCtx(ctx)
-		_, argstring := splitInput(strings.Join(args, " "))
-
-		// If alias was found, use command alias
-		if aliases[args[0]] != "" {
-			alias := aliases[args[0]]
-			argstring = alias + strings.TrimPrefix(argstring, args[0])
-			cmdArgs, _ := splitInput(argstring)
-			args = cmdArgs
-		}
-
-		// If command is defined in Lua then run it
-		if commands[args[0]] != nil {
-			err := l.CallByParam(lua.P{
-				Fn: commands[args[0]],
-				NRet:    1,
-				Protect: true,
-			}, luar.New(l, args[1:]))
-			luaexitcode := l.Get(-1)
-			var exitcode uint8 = 0
-
-			l.Pop(1)
-
-			if code, ok := luaexitcode.(lua.LNumber); luaexitcode != lua.LNil && ok {
-				exitcode = uint8(code)
-			}
-
-			if err != nil {
-				fmt.Fprintln(os.Stderr,
-					"Error in command:\n\n" + err.Error())
-			}
-			hooks.Em.Emit("command.exit", exitcode)
-			return interp.NewExitStatus(exitcode)
-		}
-
-		if _, err := interp.LookPathDir(hc.Dir, hc.Env, args[0]); err != nil {
-			hooks.Em.Emit("command.not-found", args[0])
-			return interp.NewExitStatus(127)
-		}
-
-		return interp.DefaultExecHandler(2 * time.Second)(ctx, args)
-	}
-	runner, _ := interp.New(
-		interp.StdIO(os.Stdin, os.Stdout, os.Stderr),
-		interp.ExecHandler(exechandle),
-	)
 	err = runner.Run(context.TODO(), file)
 
 	return err