Exemplo n.º 1
0
// Attempt to clean up after ctrl-C otherwise
// terminal is left in bad shape
func signalCatcher(c *api.Context) {
	ch := make(chan os.Signal)
	signal.Notify(ch, syscall.SIGINT)
	<-ch
	c.Close()
	os.Exit(0)
}
Exemplo n.º 2
0
func main() {
	createDataDir()
	// Construct the context
	context := api.Context{
		W:        os.Stdout,
		Shells:   shells.Allshells(),
		Commands: commands.Allcommands(),
	}
	argParse(&context)
	if err := context.SetShell(context.Shells[option.shell]); err != nil {
		fmt.Fprintln(context.W, err)
	}
	context.Liner = liner.NewLiner()

	// Execute command if supplied through command-line
	if option.cmdstr != "" {
		if err := doCommand(&context, option.cmdstr); err != nil {
			fmt.Fprintln(context.W, err)
		}
	} else {
		option.interactive = true
	}

	go signalCatcher(&context)
	if option.interactive {
		interactiveLoop(&context)
	}
	(&context).Close()
}
Exemplo n.º 3
0
func handleShellCommand(c *api.Context) (err error) {
	shell := c.Cursh
	cmdname := api.SplitArgN(c.Line, " ", 2)[0]
	defer func() {
		if r := recover(); r != nil {
			fmt.Fprintf(c.W, "Recovered from %q: %v\n", cmdname, r)
			fmt.Println(string(debug.Stack()))
		}
	}()
	if !api.IsCommand(cmdname, c.ShellCommands(shell.Name())) {
		return fmt.Errorf("Invalid command: %v", cmdname)
	}
	if cmd := shell.GetCommand(cmdname); cmd == nil {
		err = fmt.Errorf(
			"Command %q not supported in %v", cmdname, api.SHELL_CB)
	} else {
		err = cmd.Interpret(c)
	}
	return
}
Exemplo n.º 4
0
func doCommand(c *api.Context, line string) (err error) {
	// Switch shells if need be
	switch {
	case strings.HasPrefix(line, api.SHELL_CB):
		err = c.SetShell(c.Shells[api.SHELL_CB])
	case strings.HasPrefix(line, api.SHELL_N1QL):
		err = c.SetShell(c.Shells[api.SHELL_N1QL])
	case strings.HasPrefix(line, api.SHELL_INDEX):
		err = c.SetShell(c.Shells[api.SHELL_INDEX])
	default:
		c.Line = line
		// Handle the command for the current shell
		err = handleShellCommand(c)
	}
	return
}
Exemplo n.º 5
0
func (cmd *CommandsCommand) Interpret(c *api.Context) (err error) {
	for name, cmd := range c.ShellCommands(c.Cursh.Name()) {
		fmt.Fprintf(c.W, "  %-15v %v\n", name, cmd.Description())
	}
	return
}