package main import ( "bufio" "fmt" "io" "os" "runtime" "strings" "os/signal" "os/user" "path/filepath" "hilbish/golibs/bait" "github.com/pborman/getopt" "github.com/yuin/gopher-lua" "layeh.com/gopher-luar" "golang.org/x/term" ) var ( l *lua.LState lr *LineReader commands = map[string]*lua.LFunction{} aliases = map[string]string{} homedir string confDir string curuser *user.User hooks bait.Bait defaultConfPath string defaultHistPath string ) func main() { homedir, _ = os.UserHomeDir() confDir = getenv("XDG_CONFIG_HOME", homedir + "/.config") curuser, _ = user.Current() preloadPath = strings.Replace(preloadPath, "~", homedir, 1) sampleConfPath = strings.Replace(sampleConfPath, "~", homedir, 1) if defaultConfDir == "" { // we'll add *our* default if its empty (wont be if its changed comptime) if _, err := os.Stat(filepath.Join(confDir, "hilbish")); os.IsNotExist(err) { defaultConfPath = filepath.Join(homedir, "/.hilbishrc.lua") } else { defaultConfPath = filepath.Join(confDir, "hilbish", "hilbishrc.lua") } } else { // else do ~ substitution defaultConfPath = filepath.Join(strings.Replace(defaultConfDir, "~", homedir, 1), ".hilbishrc.lua") } if defaultHistDir == "" { // we'll add *our* default if its empty (wont be if its changed comptime) if _, err := os.Stat(filepath.Join(confDir, "hilbish")); os.IsNotExist(err) { defaultHistPath = filepath.Join(homedir, "/.hilbish-history") } else { defaultHistPath = filepath.Join(confDir, "hilbish", ".hilbish-history") } } else { // else do ~ substitution defaultHistPath = filepath.Join(strings.Replace(defaultHistDir, "~", homedir, 1), ".hilbishrc.lua") } helpflag := getopt.BoolLong("help", 'h', "Prints Hilbish flags") verflag := getopt.BoolLong("version", 'v', "Prints Hilbish version") setshflag := getopt.BoolLong("setshellenv", 'S', "Sets $SHELL to Hilbish's executed path") cmdflag := getopt.StringLong("command", 'c', "", "Executes a command on startup") configflag := getopt.StringLong("config", 'C', defaultConfPath, "Sets the path to Hilbish's config") getopt.BoolLong("login", 'l', "Force Hilbish to be a login shell") getopt.BoolLong("interactive", 'i', "Force Hilbish to be an interactive shell") getopt.BoolLong("noexec", 'n', "Don't execute and only report Lua syntax errors") getopt.Parse() loginshflag := getopt.Lookup('l').Seen() interactiveflag := getopt.Lookup('i').Seen() noexecflag := getopt.Lookup('n').Seen() if *helpflag { getopt.PrintUsage(os.Stdout) os.Exit(0) } if *cmdflag == "" || interactiveflag { interactive = true } if fileInfo, _ := os.Stdin.Stat(); (fileInfo.Mode() & os.ModeCharDevice) == 0 { interactive = false } if getopt.NArgs() > 0 { interactive = false } if noexecflag { noexecute = true } // first arg, first character if loginshflag || os.Args[0][0] == '-' { login = true } if *verflag { fmt.Printf("Hilbish %s\n", version) os.Exit(0) } // Set $SHELL if the user wants to if *setshflag { os.Setenv("SHELL", os.Args[0]) } // If user's config doesn't exixt, if _, err := os.Stat(defaultConfPath); os.IsNotExist(err) { // Read default from current directory // (this is assuming the current dir is Hilbish's git) input, err := os.ReadFile(".hilbishrc.lua") if err != nil { // If it wasnt found, go to the real sample conf input, err = os.ReadFile(sampleConfPath) if err != nil { fmt.Println("could not find .hilbishrc.lua or", sampleConfPath) return } } // Create it using either default config we found err = os.WriteFile(defaultConfPath, input, 0644) if err != nil { // If that fails, bail fmt.Println("Error creating config file") fmt.Println(err) return } } go HandleSignals() LuaInit() RunLogin() RunConfig(*configflag) lr = NewLineReader("") if fileInfo, _ := os.Stdin.Stat(); (fileInfo.Mode() & os.ModeCharDevice) == 0 { scanner := bufio.NewScanner(bufio.NewReader(os.Stdin)) for scanner.Scan() { RunInput(scanner.Text()) } } if *cmdflag != "" { RunInput(*cmdflag) } if getopt.NArgs() > 0 { l.SetGlobal("args", luar.New(l, getopt.Args())) err := l.DoFile(getopt.Arg(0)) if err != nil { fmt.Fprintln(os.Stderr, err) os.Exit(1) } os.Exit(0) } input: for interactive { running = false lr.SetPrompt(fmtPrompt()) input, err := lr.Read() if err == io.EOF { // Exit if user presses ^D (ctrl + d) fmt.Println("") break } if err != nil { // If we get a completely random error, print fmt.Fprintln(os.Stderr, err) } input = strings.TrimSpace(input) if len(input) == 0 { continue } if strings.HasSuffix(input, "\\") { for { input, err = ContinuePrompt(strings.TrimSuffix(input, "\\")) if err != nil { goto input // continue inside nested loop } if !strings.HasSuffix(input, "\\") { break } } } running = true HandleHistory(input) RunInput(input) termwidth, _, err := term.GetSize(0) if err != nil { continue } fmt.Printf("\u001b[7m∆\u001b[0m" + strings.Repeat(" ", termwidth - 1) + "\r") } } func ContinuePrompt(prev string) (string, error) { hooks.Em.Emit("multiline", nil) lr.SetPrompt(multilinePrompt) cont, err := lr.Read() if err != nil { fmt.Println("") return "", err } cont = strings.TrimSpace(cont) return prev + strings.TrimSuffix(cont, "\n"), nil } // This semi cursed function formats our prompt (obviously) func fmtPrompt() string { host, _ := os.Hostname() cwd, _ := os.Getwd() cwd = strings.Replace(cwd, curuser.HomeDir, "~", 1) username := curuser.Username // 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, "D", filepath.Base(cwd), "h", host, "u", username, } for i, v := range args { if i % 2 == 0 { args[i] = "%" + v } } r := strings.NewReplacer(args...) nprompt := r.Replace(prompt) return nprompt } // do i even have to say func HandleSignals() { c := make(chan os.Signal) signal.Notify(c, os.Interrupt) for range c { if !running { lr.ClearInput() } } } func HandleHistory(cmd string) { lr.AddHistory(cmd) // TODO: load history again (history shared between sessions like this ye) }