package main import ( "fmt" "os" "os/exec" "strings" "syscall" "time" "hilbish/golibs/bait" "hilbish/golibs/commander" "hilbish/golibs/fs" "github.com/yuin/gopher-lua" "layeh.com/gopher-luar" ) var minimalconf = ` lunacolors = require 'lunacolors' prompt(lunacolors.format( '{blue}%u {cyan}%d {green}∆{reset} ' )) ` func LuaInit() { l = lua.NewState() l.OpenLibs() l.SetGlobal("prompt", l.NewFunction(hshprompt)) l.SetGlobal("multiprompt", l.NewFunction(hshmlprompt)) l.SetGlobal("alias", l.NewFunction(hshalias)) l.SetGlobal("appendPath", l.NewFunction(hshappendPath)) l.SetGlobal("exec", l.NewFunction(hshexec)) l.SetGlobal("goro", luar.New(l, hshgoroutine)) l.SetGlobal("timeout", luar.New(l, hshtimeout)) l.SetGlobal("interval", l.NewFunction(hshinterval)) // yes this is stupid, i know l.PreloadModule("hilbish", HilbishLoader) l.DoString("hilbish = require 'hilbish'") // Add fs module to Lua l.PreloadModule("fs", fs.Loader) cmds := commander.New() // When a command from Lua is added, register it for use cmds.Events.On("commandRegister", func(cmdName string, cmd *lua.LFunction) { commands[cmdName] = cmd }) cmds.Events.On("commandDeregister", func(cmdName string) { delete(commands, cmdName) }) l.PreloadModule("commander", cmds.Loader) hooks = bait.New() l.PreloadModule("bait", hooks.Loader) // Add more paths that Lua can require from l.DoString("package.path = package.path .. " + requirePaths) err := l.DoFile("preload.lua") if err != nil { err = l.DoFile(preloadPath) if err != nil { fmt.Fprintln(os.Stderr, "Missing preload file, builtins may be missing.") } } } func RunConfig(confpath string) { if !interactive { return } err := l.DoFile(confpath) if err != nil { fmt.Fprintln(os.Stderr, err, "\nAn error has occured while loading your config! Falling back to minimal default config.") l.DoString(minimalconf) } } func RunLogin() { if _, err := os.Stat(homedir + "/.hprofile.lua"); os.IsNotExist(err) { return } if !login { return } err := l.DoFile(homedir + "/.hprofile.lua") if err != nil { fmt.Fprintln(os.Stderr, err, "\nAn error has occured while loading your login config!n") } } func hshprompt(L *lua.LState) int { prompt = L.CheckString(1) return 0 } func hshmlprompt(L *lua.LState) int { multilinePrompt = L.CheckString(1) return 0 } func hshalias(L *lua.LState) int { alias := L.CheckString(1) source := L.CheckString(2) aliases[alias] = source return 1 } func hshappendPath(L *lua.LState) int { dir := L.CheckString(1) dir = strings.Replace(dir, "~", curuser.HomeDir, 1) pathenv := os.Getenv("PATH") // if dir isnt already in $PATH, add it if !strings.Contains(pathenv, dir) { os.Setenv("PATH", pathenv + ":" + dir) } return 0 } func hshexec(L *lua.LState) int { cmd := L.CheckString(1) cmdArgs, _ := splitInput(cmd) cmdPath, err := exec.LookPath(cmdArgs[0]) if err != nil { fmt.Println(err) // if we get here, cmdPath will be nothing // therefore nothing will run } // syscall.Exec requires an absolute path to a binary // path, args, string slice of environments // TODO: alternative for windows syscall.Exec(cmdPath, cmdArgs, os.Environ()) return 0 // random thought: does this ever return? } func hshgoroutine(gofunc func()) { go gofunc() } func hshtimeout(timeoutfunc func(), ms int) { timeout := time.Duration(ms) * time.Millisecond time.AfterFunc(timeout, timeoutfunc) } func hshinterval(L *lua.LState) int { intervalfunc := L.CheckFunction(1) ms := L.CheckInt(2) interval := time.Duration(ms) * time.Millisecond ticker := time.NewTicker(interval) stop := make(chan lua.LValue) go func() { for { select { case <-ticker.C: if err := L.CallByParam(lua.P{ Fn: intervalfunc, NRet: 0, Protect: true, }); err != nil { fmt.Fprintln(os.Stderr, "Error in interval function:\n\n", err) } case <-stop: ticker.Stop() return } } }() L.Push(lua.LChannel(stop)) return 1 }