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"
 )

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
-var (
-	l *lua.LState
-
 // User's prompt, this will get set when lua side is initialized
-	prompt string
+var prompt string
-	multilinePrompt = "> "
+var multilinePrompt = "> "
 
-	commands = map[string]bool{}
+// Map of builtin/custom commands defined in the commander lua module
-	aliases = map[string]string{}
+var commands = map[string]bool{}
-
+// Command aliases
-	bait hooks.Bait
+var aliases = map[string]string{}
-	homedir string
+var bait hooks.Bait
-	running bool
+var homedir string
-)
+var 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) {
@@ -131,18 +125,14 @@ 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")
 	}
 }
@@ -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
@@ -50,9 +49,7 @@ func RunInput(input string) {
 			fmt.Fprintln(os.Stderr,
 			"Error in command:\n\n" + err.Error())
 		}
-		if cmdArgs[0] != "exit" {
+		if cmdArgs[0] != "exit" { HandleHistory(cmdString) }
-			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)