lua.go

@@ -2,10 +2,10 @@ package main
 
 import (
 	"fmt"
-	hooks "hilbish/golibs/bait"
-	cmds "hilbish/golibs/commander"
-	lfs "hilbish/golibs/fs"
 	"os"
+	lfs "hilbish/golibs/fs"
+	cmds "hilbish/golibs/commander"
+	hooks "hilbish/golibs/bait"
 
 	"github.com/yuin/gopher-lua"
 )
@@ -34,7 +34,7 @@ func LuaInit(confpath string) {
 	commander := cmds.New()
 	// When a command from Lua is added, register it for use
 	commander.Events.On("commandRegister",
-		func(cmdName string, cmd *lua.LFunction) {
+	func (cmdName string, cmd *lua.LFunction) {
 		commands[cmdName] = true
 		l.SetField(
 			l.GetTable(l.GetGlobal("commanding"),

main.go

@@ -2,12 +2,11 @@ package main
 
 import (
 	"fmt"
-	"io"
 	"os"
-	"os/signal"
 	"os/user"
+	"os/signal"
 	"strings"
-
+	"io"
 	hooks "hilbish/golibs/bait"
 
 	"github.com/akamensky/argparse"
@@ -17,21 +16,18 @@ import (
 )
 
 const version = "0.4.0-dev.2+hilbiline"
+var l *lua.LState
+// User's prompt, this will get set when lua side is initialized
+var prompt string
+var multilinePrompt = "> "
 
-var (
+// Map of builtin/custom commands defined in the commander lua module
-	l *lua.LState
+var commands = map[string]bool{}
-
+// Command aliases
-	// User's prompt, this will get set when lua side is initialized
+var aliases = map[string]string{}
-	prompt string
+var bait hooks.Bait
-	multilinePrompt = "> "
+var homedir string
-
+var running bool
-	commands = map[string]bool{}
-	aliases = map[string]string{}
-
-	bait hooks.Bait
-	homedir string
-	running bool
-)
 
 func main() {
 	homedir, _ = os.UserHomeDir()
@@ -73,9 +69,7 @@ func main() {
 	}
 
 	// Set $SHELL if the user wants to
-	if *setshflag {
+	if *setshflag { os.Setenv("SHELL", os.Args[0]) }
-		os.Setenv("SHELL", os.Args[0])
-	}
 
 	// If user's config doesn't exixt,
 	if _, err := os.Stat(defaultconfpath); os.IsNotExist(err) {
@@ -92,7 +86,7 @@ func main() {
 		}
 
 		// Create it using either default config we found
-		err = os.WriteFile(homedir+"/.hilbishrc.lua", input, 0644)
+		err = os.WriteFile(homedir + "/.hilbishrc.lua", input, 0644)
 		if err != nil {
 			// If that fails, bail
 			fmt.Println("Error creating config file")
@@ -131,19 +125,15 @@ func main() {
 			for {
 				input, err = ContinuePrompt(strings.TrimSuffix(input, "\\"))
 
-				if err != nil || !strings.HasSuffix(input, "\\") {
+				if err != nil || !strings.HasSuffix(input, "\\") { break }
-					break
-				}
 			}
 		}
 		running = true
 		RunInput(input)
 
 		termwidth, _, err := term.GetSize(0)
-		if err != nil {
+		if err != nil { continue }
-			continue
+		fmt.Printf("\u001b[7m∆\u001b[0m" + strings.Repeat(" ", termwidth - 1) + "\r")
-		}
-		fmt.Printf("\u001b[7m∆\u001b[0m" + strings.Repeat(" ", termwidth-1) + "\r")
 	}
 }
 
@@ -198,3 +188,4 @@ func HandleSignals() {
 		}
 	}
 }
+

shell.go

@@ -1,17 +1,18 @@
 package main
 
 import (
+	"fmt"
+	"os"
 	"bufio"
 	"context"
-	"fmt"
 	"io"
-	"os"
 	"strings"
 
 	"github.com/yuin/gopher-lua"
 	"layeh.com/gopher-luar"
 	"mvdan.cc/sh/v3/interp"
 	"mvdan.cc/sh/v3/syntax"
+
 )
 
 func RunInput(input string) {
@@ -30,9 +31,7 @@ func RunInput(input string) {
 
 	// If alias was found, use command alias
 	if aliases[cmdArgs[0]] != "" {
-		alias := aliases[cmdArgs[0]]
+		cmdString = aliases[cmdArgs[0]] + strings.Trim(cmdString, cmdArgs[0])
-		cmdString = alias + strings.Trim(cmdString, cmdArgs[0])
-		cmdArgs[0] = alias
 	}
 
 	// If command is defined in Lua then run it
@@ -48,11 +47,9 @@ func RunInput(input string) {
 		}, luar.New(l, cmdArgs[1:]))
 		if err != nil {
 			fmt.Fprintln(os.Stderr,
-				"Error in command:\n\n"+err.Error())
+			"Error in command:\n\n" + err.Error())
-		}
-		if cmdArgs[0] != "exit" {
-			HandleHistory(cmdString)
 		}
+		if cmdArgs[0] != "exit" { HandleHistory(cmdString) }
 		return
 	}
 
@@ -63,9 +60,7 @@ func RunInput(input string) {
 		if syntax.IsIncomplete(err) {
 			for {
 				cmdString, err = ContinuePrompt(strings.TrimSuffix(cmdString, "\\"))
-				if err != nil {
+				if err != nil { break }
-					break
-				}
 				err = execCommand(cmdString)
 					if syntax.IsIncomplete(err) || strings.HasSuffix(input, "\\") {
 					continue
@@ -80,9 +75,7 @@ func RunInput(input string) {
 		} else {
 			if code, ok := interp.IsExitStatus(err); ok {
 				bait.Em.Emit("command.exit", code)
-			} else {
+			} else { fmt.Fprintln(os.Stderr, err) }
-				fmt.Fprintln(os.Stderr, err)
-			}
 		}
 	} else {
 		bait.Em.Emit("command.exit", 0)