package main import ( "fmt" "os" "strings" "hilbish/golibs/bait" "hilbish/golibs/commander" "hilbish/golibs/fs" "github.com/yuin/gopher-lua" ) var minimalconf = ` ansikit = require 'ansikit' prompt(ansikit.format( '{blue}%u {cyan}%d {green}∆{reset} ' )) ` func LuaInit() { l = lua.NewState() l.OpenLibs() l.SetGlobal("_ver", lua.LString(version)) l.SetGlobal("_user", lua.LString(curuser.Username)) l.SetGlobal("prompt", l.NewFunction(hshprompt)) l.SetGlobal("multiprompt", l.NewFunction(hshmlprompt)) l.SetGlobal("alias", l.NewFunction(hshalias)) l.SetGlobal("appendPath", l.NewFunction(hshappendPath)) // 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] = true l.SetField( l.GetTable(l.GetGlobal("commanding"), lua.LString("__commands")), cmdName, cmd) }) 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 .. ';./libs/?/init.lua;./?/init.lua;./?/?.lua' .. ';/usr/share/hilbish/libs/?/init.lua;' .. ';/usr/share/hilbish/libs/?/?.lua;' .. os.getenv 'HOME' .. '/.local/share/hilbish/libs/?/init.lua;' .. os.getenv 'HOME' .. '/.local/share/hilbish/libs/?/?.lua;' .. os.getenv 'HOME' .. '/.local/share/hilbish/libs/?.lua' `) err := l.DoFile("/usr/share/hilbish/preload.lua") if err != nil { err = l.DoFile("preload.lua") 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.\n") 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 { path := L.CheckString(1) path = strings.Replace(path, "~", curuser.HomeDir, 1) os.Setenv("PATH", os.Getenv("PATH") + ":" + path) return 0 }