springboard.go

package main

import (
	"fmt"
	"github.com/draxil/springboard/watch"
	"github.com/urfave/cli"
	"os"
)

const version = "0.3.2"
const author = "Joe Higton"
const authorEmail = "draxil@gmail.com"

func main() {
	app := app()
	app.Run(os.Args)
}

func app() *cli.App {
	app := cli.NewApp()
	app.Name = "springboard"
	app.Usage = "Watch a directory for files and send them places"
	commands, flags, _ := setup()
	app.Commands = commands
	app.Version = version
	app.Authors = []cli.Author{
		{
			Name:  author,
			Email: authorEmail,
		},
	}
	app.Flags = flags
	return app
}

func setup() (c []cli.Command, f []cli.Flag, cfg *watch.Config) {
	cfg = &watch.Config{}
	f = globalFlags(cfg)

	addCommand := func(cmd cli.Command) {
		cmd = wrapCmd(cfg, cmd)
		c = append(c, cmd)
	}

	addCommand(http_post_command(cfg, run_watch))
	addCommand(echo_command(cfg, run_watch))
	addCommand(run_command(cfg, run_watch))

	return
}

func wrapCmd(cfg *watch.Config, c cli.Command) cli.Command {
	a := c.Action.(func(*cli.Context))
	c.Action = func(c *cli.Context) {
		setupAction(cfg, c)
		a(c)
	}
	return c
}

func setupAction(cfg *watch.Config, c *cli.Context) {
	sparanoia := c.GlobalString("paranoia")
	switch sparanoia {
	case "off":
		cfg.Paranoia = watch.NoParanoia
	case "basic":
		cfg.Paranoia = watch.BasicParanoia
	case "extra":
		cfg.Paranoia = watch.ExtraParanoia
	default:
		fmt.Fprintln(os.Stderr, "Invalid choice of paranoia=", c)
		cli.ShowSubcommandHelp(c)
		os.Exit(1)
	}
}

func globalFlags(cfg *watch.Config) (f []cli.Flag) {

	f = []cli.Flag{
		cli.StringFlag{
			Name:        "archive",
			Usage:       "move the file to this location after successful action",
			Destination: &cfg.ArchiveDir,
		},
		cli.StringFlag{
			Name:        "error-dir",
			Usage:       "move the file to this location after a failed action",
			Destination: &cfg.ErrorDir,
		},
		cli.StringFlag{
			Name:  "paranoia",
			Usage: "Do we take extra steps to ensure the file has been completely written? See documentation for full details. Values: off, basic, extra.",
			Value: "basic",
		},
		cli.BoolFlag{
			Name:        "process-existing",
			Usage:       "Process any pre-existing files in the directory on startup. Obviously best used alongside an archive option of some kind.",
			Destination: &cfg.ProcessExistingFiles,
		},
		cli.BoolFlag{
			Name:        "debug",
			Usage:       "enable verbose debug output",
			Destination: &cfg.Debug,
		},
		cli.BoolTFlag{
			Name:        "log-errors",
			Usage:       "enable logging of errors (default=true)",
			Destination: &cfg.ReportErrors,
		},
		cli.BoolFlag{
			Name:        "log-actions",
			Usage:       "enable logging of actions",
			Destination: &cfg.ReportActions,
		},
		cli.StringSliceFlag{
			Name:  "testing",
			Usage: "Used to set testing options, usually only required for development & testing",
			Value: (*cli.StringSlice)(&cfg.TestingOptions),
		},
	}
	return
}

func run_watch(c *watch.Config) {
	e := watch.Watch(c)
	if e != nil {
		fmt.Fprintln(os.Stderr, e)
	}
}

func http_post_command(cfg *watch.Config, action func(*watch.Config)) cli.Command {
	var pa watch.PostAction
	//	var http_headers cli.StringSlice
	return cli.Command{
		Name:  "post",
		Usage: "post the file somewhere - using an HTTP POST",
		Flags: []cli.Flag{
			// cli.StringSliceFlag{
			// 	Name:  "header",
			// 	Value: &http_headers,
			// 	Usage: "Set extra http headers, format is KEY:VAL",
			// },
			cli.StringFlag{
				Name:        "mime",
				Destination: &pa.Mime,
				Usage:       "Force the mime type on the post",
			},
			cli.StringFlag{
				Name:        "uname",
				Destination: &pa.BasicAuthUsername,
				Usage:       "Triggers use of HTTP basic auth (See RFC 2617, Section 2.) with the provided username",
			},
			cli.StringFlag{
				Name:        "pass",
				Destination: &pa.BasicAuthPwd,
				Usage:       "Set the password for HTTP basic auth.",
			},
		},
		ArgsUsage: "URL DIR",
		Action: func(c *cli.Context) {

			args := c.Args()

			bail := func() {
				cli.ShowSubcommandHelp(c)
				os.Exit(1)
			}

			if !args.Present() {
				bail()
			}
			arg := 0

			next := func() string {
				val := args.Get(arg)
				arg++
				if val == "" {
					bail()
				}
				return val
			}

			pa.To = next()

			cfg.Actions = []watch.Action{
				&pa,
			}
			cfg.Dir = next()
			action(cfg)
		},
	}
}

func echo_command(cfg *watch.Config, action func(*watch.Config)) cli.Command {
	return cli.Command{
		Name:      "echo",
		Usage:     "echo the full filepath",
		ArgsUsage: "DIR",
		Action: func(c *cli.Context) {

			args := c.Args()

			bail := func() {
				cli.ShowSubcommandHelp(c)
				os.Exit(1)
			}

			if !args.Present() {
				bail()
			}

			cfg.Actions = []watch.Action{
				&watch.EchoAction{},
			}
			cfg.Dir = args.First()
			action(cfg)
		},
	}
}

func run_command(cfg *watch.Config, action func(*watch.Config)) cli.Command {
	var ra watch.RunAction
	return cli.Command{
		Name:      "run",
		Usage:     "Runs a command with the dropped filepath. Determines success like a normal shell command, so a 0 exit status. Arguments passed are send to CMD before the filename, see -postarg to send the command arguments after the filename.",
		ArgsUsage: "CMD [CMDARGS..] DIR",
		Flags: []cli.Flag{
			// cli.StringSliceFlag{
			// 	Name:  "header",
			// 	Value: &http_headers,
			// 	Usage: "Set extra http headers, format is KEY:VAL",
			// },
			cli.StringSliceFlag{
				Name:  "postarg",
				Usage: "Add arguments which are run after the filename in the command we build. So if you were doing a cp: ./springboard run --postarg /some/place cp\nNote that you can use postarg repeatedly to add more arguments.",
				Value: (*cli.StringSlice)(&ra.PostArgs),
			},
		},
		Action: func(c *cli.Context) {

			args := c.Args()

			bail := func() {
				cli.ShowSubcommandHelp(c)
				os.Exit(1)
			}

			if !args.Present() {
				bail()
			}

			ra.Cmd = args.First()
			ra.Args = args[1 : len(args)-1]

			cfg.Actions = []watch.Action{
				&ra,
			}
			cfg.Dir = args[len(args)-1]

			action(cfg)
		},
	}
}