...
This commit is contained in:
parent
507953c348
commit
82d232c948
2 changed files with 57 additions and 54 deletions
|
@ -6,5 +6,5 @@ if v < 55 {
|
||||||
println("it fucking works")
|
println("it fucking works")
|
||||||
}
|
}
|
||||||
|
|
||||||
rcmd("goblin", "ls", "-r", "100", "src") && println("it works also")
|
Rcmd("goblin", "ls", "-r", "100", "src") || println("it works also") && Rcmd("goblin", "cat", "check.anko")
|
||||||
|
|
||||||
|
|
|
@ -20,10 +20,10 @@ import (
|
||||||
"os/exec"
|
"os/exec"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Cmd struct {
|
type Command struct {
|
||||||
input io.ReadCloser
|
input io.Reader
|
||||||
*exec.Cmd
|
*exec.Cmd
|
||||||
output io.WriteCloser
|
output io.Writer
|
||||||
}
|
}
|
||||||
|
|
||||||
const version = "0.1.8"
|
const version = "0.1.8"
|
||||||
|
@ -33,78 +33,71 @@ var (
|
||||||
file string
|
file string
|
||||||
args []string
|
args []string
|
||||||
e *env.Env
|
e *env.Env
|
||||||
flag *mtool.Flags
|
//flag *mtool.Flags
|
||||||
)
|
)
|
||||||
|
|
||||||
func Run(flagSet *mtool.Flags) {
|
func Run(flagSet *mtool.Flags) {
|
||||||
var exitCode int
|
|
||||||
|
|
||||||
flag = flagSet
|
printVersion := flagSet.Bool("v", false, "prints out the version and then exits")
|
||||||
|
flagSet.StringVar(&flagExecute, "e", "", "execute the Anko code")
|
||||||
parseFlags()
|
flagSet.Parse()
|
||||||
setupEnv()
|
args := flagSet.Args()
|
||||||
exitCode = runNonInteractive()
|
|
||||||
//if flagExecute != "" || flag.NArg() > 0 {
|
if *printVersion {
|
||||||
//}
|
|
||||||
|
|
||||||
os.Exit(exitCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseFlags() {
|
|
||||||
flagVersion := flag.Bool("v", false, "prints out the version and then exits")
|
|
||||||
flag.StringVar(&flagExecute, "e", "", "execute the Anko code")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
if *flagVersion {
|
|
||||||
fmt.Println(version)
|
fmt.Println(version)
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
|
||||||
if flagExecute != "" || flag.NArg() < 1 {
|
setupEnv()
|
||||||
args = flag.Args()
|
if flagExecute != "" {
|
||||||
return
|
exitCode := runNonInteractive()
|
||||||
|
os.Exit(exitCode)
|
||||||
}
|
}
|
||||||
|
|
||||||
file = flag.Arg(0)
|
if len(args) < 1 {
|
||||||
args = flag.Args()[1:]
|
file = "#stdin"
|
||||||
|
} else {
|
||||||
|
file = args[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
exitCode := runNonInteractive()
|
||||||
|
os.Exit(exitCode)
|
||||||
}
|
}
|
||||||
|
|
||||||
func setupEnv() {
|
func Cmd(args ...string) *Command {
|
||||||
e = env.NewEnv()
|
|
||||||
cmd := func(args ...string) *Cmd {
|
|
||||||
if len(args) < 1 {
|
if len(args) < 1 {
|
||||||
panic("too few arguments")
|
panic("too few arguments")
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd := exec.Command(args[0], args[1:]...)
|
cmd := exec.Command(args[0], args[1:]...)
|
||||||
|
|
||||||
return &Cmd{
|
return &Command{ Cmd: cmd }
|
||||||
Cmd: cmd,
|
|
||||||
input: os.Stdin,
|
|
||||||
output: os.Stdout,
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
e.Define("args", args)
|
|
||||||
e.Define("cmd", cmd)
|
func (cmd *Command)IO(input io.Reader, output io.Writer) *Command {
|
||||||
e.Define("rcmd", func(args ...string) bool {
|
cmd.input = input
|
||||||
rcmd := cmd(args...)
|
cmd.output = output
|
||||||
|
return cmd
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cmd *Command) Run() error {
|
||||||
|
input := cmd.input
|
||||||
|
output := cmd.output
|
||||||
|
|
||||||
input := rcmd.input
|
stdin, err := cmd.StdinPipe()
|
||||||
output := rcmd.output
|
|
||||||
|
|
||||||
stdin, err := rcmd.StdinPipe()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
stdout, err := rcmd.StdoutPipe()
|
stdout, err := cmd.StdoutPipe()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
err = rcmd.Start()
|
err = cmd.Start()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
return err
|
||||||
}
|
}
|
||||||
go func(){
|
go func(){
|
||||||
_, err = io.Copy(stdin, input)
|
_, err = io.Copy(stdin, input)
|
||||||
|
@ -112,6 +105,7 @@ func setupEnv() {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
_, err = io.Copy(output, stdout)
|
_, err = io.Copy(output, stdout)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -119,13 +113,22 @@ func setupEnv() {
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
err = rcmd.Wait()
|
err = cmd.Wait()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
return false
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
return true
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func setupEnv() {
|
||||||
|
e = env.NewEnv()
|
||||||
|
e.Define("Args", args)
|
||||||
|
e.Define("Cmd", Cmd)
|
||||||
|
e.Define("Rcmd", func(args ...string) bool {
|
||||||
|
cmd := Cmd(args...).IO(os.Stdin, os.Stdout)
|
||||||
|
return cmd.Run() == nil
|
||||||
})
|
})
|
||||||
core.Import(e)
|
core.Import(e)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue