woodpecker/cli/handle.go
2014-09-24 21:24:12 -07:00

33 lines
668 B
Go

package main
import (
"os"
"github.com/codegangsta/cli"
"github.com/drone/drone/client"
)
type handlerFunc func(*cli.Context, *client.Client) error
// handle wraps the command function handlers and
// sets up the environment.
func handle(c *cli.Context, fn handlerFunc) {
var token = c.GlobalString("token")
var server = c.GlobalString("server")
// if no server url is provided we can default
// to the hosted Drone service.
if len(server) == 0 {
server = "http://test.drone.io"
}
// create the drone client
client := client.New(token, server)
// handle the function
if err := fn(c, client); err != nil {
println(err.Error())
os.Exit(1)
}
}