mirror of https://github.com/Hilbis/Hilbish
104 lines
1.8 KiB
Go
104 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
_ "bufio"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
_ "os/user"
|
|
"syscall"
|
|
"os/signal"
|
|
"strings"
|
|
|
|
"github.com/akamensky/argparse"
|
|
"github.com/bobappleyard/readline"
|
|
"github.com/yuin/gopher-lua"
|
|
)
|
|
|
|
const version = "0.0.4"
|
|
var l *lua.LState
|
|
var prompt string
|
|
|
|
func main() {
|
|
parser := argparse.NewParser("hilbish", "A shell for lua and flower lovers")
|
|
verflag := parser.Flag("v", "version", &argparse.Options{
|
|
Required: false,
|
|
Help: "color palette to use",
|
|
})
|
|
|
|
err := parser.Parse(os.Args)
|
|
if err != nil {
|
|
fmt.Print(parser.Usage(err))
|
|
os.Exit(0)
|
|
}
|
|
|
|
if *verflag {
|
|
fmt.Printf("Hilbish v%s\n", version)
|
|
os.Exit(0)
|
|
}
|
|
|
|
os.Setenv("SHELL", os.Args[0])
|
|
HandleSignals()
|
|
LuaInit()
|
|
|
|
for {
|
|
//user, _ := user.Current()
|
|
//dir, _ := os.Getwd()
|
|
//host, _ := os.Hostname()
|
|
|
|
//reader := bufio.NewReader(os.Stdin)
|
|
|
|
//fmt.Printf(prompt)
|
|
|
|
cmdString, err := readline.String(prompt)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
}
|
|
cmdString = strings.TrimSuffix(cmdString, "\n")
|
|
err = l.DoString(cmdString)
|
|
|
|
if err == nil { continue }
|
|
|
|
cmdArgs := strings.Fields(cmdString)
|
|
|
|
if len(cmdArgs) == 0 { continue }
|
|
|
|
switch cmdArgs[0] {
|
|
case "exit":
|
|
os.Exit(0)
|
|
case "cd":
|
|
os.Chdir(strings.Trim(cmdString, "cd "))
|
|
default:
|
|
cmd := exec.Command(cmdArgs[0], cmdArgs[1:]...)
|
|
cmd.Stderr = os.Stderr
|
|
cmd.Stdout = os.Stdout
|
|
|
|
if err := cmd.Run(); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
}
|
|
}
|
|
readline.AddHistory(cmdString)
|
|
}
|
|
}
|
|
|
|
func HandleSignals() {
|
|
c := make(chan os.Signal)
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
go func() {
|
|
<-c
|
|
}()
|
|
}
|
|
|
|
func LuaInit() {
|
|
l = lua.NewState()
|
|
|
|
l.OpenLibs()
|
|
|
|
l.SetGlobal("prompt", l.NewFunction(hshprompt))
|
|
|
|
err := l.DoFile(os.Getenv("HOME") + "/.hilbishrc.lua")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|