Files
fn-serverless/fn/apps.go
Chad Arimura 1e90fc4459 consts
2017-05-26 17:01:39 -07:00

351 lines
8.6 KiB
Go

package main
import (
"encoding/json"
"errors"
"fmt"
"os"
"context"
"github.com/iron-io/functions_go"
fnclient "github.com/iron-io/functions_go/client"
apiapps "github.com/iron-io/functions_go/client/apps"
"github.com/iron-io/functions_go/models"
"github.com/jmoiron/jsonq"
"github.com/urfave/cli"
"strings"
)
type appsCmd struct {
client *fnclient.Functions
}
func apps() cli.Command {
a := appsCmd{client: apiClient()}
return cli.Command{
Name: "apps",
Usage: "manage applications",
Subcommands: []cli.Command{
{
Name: "create",
Aliases: []string{"c"},
Usage: "create a new app",
ArgsUsage: "<app>",
Action: a.create,
Flags: []cli.Flag{
cli.StringSliceFlag{
Name: "config",
Usage: "application configuration",
},
},
},
{
Name: "inspect",
Aliases: []string{"i"},
Usage: "retrieve one or all apps properties",
ArgsUsage: "<app> [property.[key]]",
Action: a.inspect,
},
{
Name: "update",
Aliases: []string{"u"},
Usage: "update an `app`",
ArgsUsage: "<app>",
Action: a.update,
Flags: []cli.Flag{
cli.StringSliceFlag{
Name: "config,c",
Usage: "route configuration",
},
},
},
{
Name: "config",
Usage: "manage your apps's function configs",
Subcommands: []cli.Command{
{
Name: "set",
Aliases: []string{"s"},
Usage: "store a configuration key for this application",
ArgsUsage: "<app> <key> <value>",
Action: a.configSet,
},
{
Name: "unset",
Aliases: []string{"u"},
Usage: "remove a configuration key for this application",
ArgsUsage: "<app> <key>",
Action: a.configUnset,
},
},
},
{
Name: "list",
Aliases: []string{"l"},
Usage: "list all apps",
Action: a.list,
},
{
Name: "delete",
Usage: "delete an app",
Action: a.delete,
},
},
}
}
func (a *appsCmd) list(c *cli.Context) error {
resp, err := a.client.Apps.GetApps(&apiapps.GetAppsParams{
Context: context.Background(),
})
if err != nil {
switch err.(type) {
case *apiapps.GetAppsAppNotFound:
return fmt.Errorf("error: %v", err.(*apiapps.GetAppsAppNotFound).Payload.Error.Message)
case *apiapps.GetAppsAppDefault:
return fmt.Errorf("unexpected error: %v", err.(*apiapps.GetAppsAppDefault).Payload.Error.Message)
}
return fmt.Errorf("unexpected error: %v", err)
}
if len(resp.Payload.Apps) == 0 {
fmt.Println("no apps found")
return nil
}
for _, app := range resp.Payload.Apps {
fmt.Println(app.Name)
}
return nil
}
func (a *appsCmd) create(c *cli.Context) error {
body := &models.AppWrapper{App: &models.App{
Name: c.Args().Get(0),
Config: extractEnvConfig(c.StringSlice("config")),
}}
resp, err := a.client.Apps.PostApps(&apiapps.PostAppsParams{
Context: context.Background(),
Body: body,
})
if err != nil {
switch err.(type) {
case *apiapps.PostAppsBadRequest:
return fmt.Errorf("error: %v", err.(*apiapps.PostAppsBadRequest).Payload.Error.Message)
case *apiapps.PostAppsConflict:
return fmt.Errorf("error: %v", err.(*apiapps.PostAppsConflict).Payload.Error.Message)
case *apiapps.PostAppsDefault:
return fmt.Errorf("unexpected error: %v", err.(*apiapps.PostAppsDefault).Payload.Error.Message)
}
return fmt.Errorf("unexpected error: %v", err)
}
const createHeader = `
____ __
/ __ \_________ ______/ /__
/ / / / ___/ __ / ___/ / _ \
/ /_/ / / / /_/ / /__/ / __/
\_________ \__,_/\___/_/\____ _
/ ____/_ ______ _____/ /_(_)___ ____ _____
/ /_ / / / / __ \/ ___/ __/ / __ \/ __ \/ ___/
/ __/ / /_/ / / / / /__/ /_/ / /_/ / / / (__ )
/_/ \____/_/ /_/\___/\__/_/\____/_/ /_/____/
`
fmt.Println(createHeader)
fmt.Println("Successfully created app: ", resp.Payload.App.Name)
return nil
}
func (a *appsCmd) update(c *cli.Context) error {
appName := c.Args().First()
patchedApp := &functions.App{
Config: extractEnvConfig(c.StringSlice("config")),
}
err := a.patchApp(appName, patchedApp)
if err != nil {
return err
}
fmt.Println("app", appName, "updated")
return nil
}
func (a *appsCmd) configSet(c *cli.Context) error {
appName := c.Args().Get(0)
key := c.Args().Get(1)
value := c.Args().Get(2)
app := &functions.App{
Config: make(map[string]string),
}
app.Config[key] = value
if err := a.patchApp(appName, app); err != nil {
return fmt.Errorf("error updating app configuration: %v", err)
}
fmt.Println(appName, "updated", key, "with", value)
return nil
}
func (a *appsCmd) configUnset(c *cli.Context) error {
appName := c.Args().Get(0)
key := c.Args().Get(1)
app := &functions.App{
Config: make(map[string]string),
}
app.Config["-"+key] = ""
if err := a.patchApp(appName, app); err != nil {
return fmt.Errorf("error updating app configuration: %v", err)
}
fmt.Printf("removed key '%s' from app '%s' \n", key, appName)
return nil
}
func (a *appsCmd) patchApp(appName string, app *functions.App) error {
resp, err := a.client.Apps.GetAppsApp(&apiapps.GetAppsAppParams{
Context: context.Background(),
App: appName,
})
if err != nil {
switch err.(type) {
case *apiapps.GetAppsAppNotFound:
return fmt.Errorf("error: %v", err.(*apiapps.GetAppsAppNotFound).Payload.Error.Message)
case *apiapps.GetAppsAppDefault:
return fmt.Errorf("unexpected error: %v", err.(*apiapps.GetAppsAppDefault).Payload.Error.Message)
}
return fmt.Errorf("unexpected error: %v", err)
}
if resp.Payload.App.Config == nil {
resp.Payload.App.Config = map[string]string{}
}
resp.Payload.App.Name = ""
if app != nil {
if app.Config != nil {
for k, v := range app.Config {
if string(k[0]) == "-" {
delete(resp.Payload.App.Config, string(k[1:]))
continue
}
resp.Payload.App.Config[k] = v
}
}
}
body := &models.AppWrapper{App: resp.Payload.App}
_, err = a.client.Apps.PatchAppsApp(&apiapps.PatchAppsAppParams{
Context: context.Background(),
App: appName,
Body: body,
})
if err != nil {
switch err.(type) {
case *apiapps.PatchAppsAppBadRequest:
return errors.New(err.(*apiapps.PatchAppsAppBadRequest).Payload.Error.Message)
case *apiapps.PatchAppsAppNotFound:
return errors.New(err.(*apiapps.PatchAppsAppNotFound).Payload.Error.Message)
case *apiapps.PatchAppsAppDefault:
return errors.New(err.(*apiapps.PatchAppsAppDefault).Payload.Error.Message)
}
return fmt.Errorf("unexpected error: %v", err)
}
return nil
}
func (a *appsCmd) inspect(c *cli.Context) error {
if c.Args().Get(0) == "" {
return errors.New("error: missing app name after the inspect command")
}
appName := c.Args().First()
prop := c.Args().Get(1)
resp, err := a.client.Apps.GetAppsApp(&apiapps.GetAppsAppParams{
Context: context.Background(),
App: appName,
})
if err != nil {
switch err.(type) {
case *apiapps.GetAppsAppNotFound:
return fmt.Errorf("error: %v", err.(*apiapps.GetAppsAppNotFound).Payload.Error.Message)
case *apiapps.GetAppsAppDefault:
return fmt.Errorf("unexpected error: %v", err.(*apiapps.GetAppsAppDefault).Payload.Error.Message)
}
return fmt.Errorf("unexpected error: %v", err)
}
enc := json.NewEncoder(os.Stdout)
enc.SetIndent("", "\t")
if prop == "" {
enc.Encode(resp.Payload.App)
return nil
}
// TODO: we really need to marshal it here just to
// unmarshal as map[string]interface{}?
data, err := json.Marshal(resp.Payload.App)
if err != nil {
return fmt.Errorf("error inspect app: %v", err)
}
var inspect map[string]interface{}
err = json.Unmarshal(data, &inspect)
if err != nil {
return fmt.Errorf("error inspect app: %v", err)
}
jq := jsonq.NewQuery(inspect)
field, err := jq.Interface(strings.Split(prop, ".")...)
if err != nil {
return errors.New("failed to inspect that apps's field")
}
enc.Encode(field)
return nil
}
func (a *appsCmd) delete(c *cli.Context) error {
appName := c.Args().First()
if appName == "" {
return errors.New("error: deleting an app takes one argument, an app name")
}
_, err := a.client.Apps.DeleteAppsApp(&apiapps.DeleteAppsAppParams{
Context: context.Background(),
App: appName,
})
if err != nil {
switch err.(type) {
case *apiapps.DeleteAppsAppNotFound:
return errors.New(err.(*apiapps.DeleteAppsAppNotFound).Payload.Error.Message)
case *apiapps.DeleteAppsAppDefault:
return errors.New(err.(*apiapps.DeleteAppsAppDefault).Payload.Error.Message)
}
return fmt.Errorf("unexpected error: %v", err)
}
fmt.Println("app", appName, "deleted")
return nil
}