87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/dokku/dokku/plugins/common"
|
|
"github.com/dokku/dokku/plugins/proxy"
|
|
|
|
flag "github.com/spf13/pflag"
|
|
)
|
|
|
|
// main entrypoint to all subcommands
|
|
func main() {
|
|
parts := strings.Split(os.Args[0], "/")
|
|
subcommand := parts[len(parts)-1]
|
|
|
|
var err error
|
|
switch subcommand {
|
|
case "build-config":
|
|
args := flag.NewFlagSet("proxy:build-config", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
err = proxy.CommandBuildConfig(appName)
|
|
case "disable":
|
|
args := flag.NewFlagSet("proxy:disable", flag.ExitOnError)
|
|
skipRestart := args.Bool("no-restart", false, "--no-restart: skip restart of the app")
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
err = proxy.CommandDisable(appName, *skipRestart)
|
|
case "enable":
|
|
args := flag.NewFlagSet("proxy:enable", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
err = proxy.CommandEnable(appName)
|
|
case "ports":
|
|
args := flag.NewFlagSet("proxy:ports", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
err = proxy.CommandPorts(appName)
|
|
case "ports-add":
|
|
args := flag.NewFlagSet("proxy:ports-add", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
_, portMaps := common.ShiftString(args.Args())
|
|
err = proxy.CommandPortsAdd(appName, portMaps)
|
|
case "ports-clear":
|
|
args := flag.NewFlagSet("proxy:ports-clear", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
err = proxy.CommandPortsClear(appName)
|
|
case "ports-remove":
|
|
args := flag.NewFlagSet("proxy:ports-remove", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
_, portMaps := common.ShiftString(args.Args())
|
|
err = proxy.CommandPortsRemove(appName, portMaps)
|
|
case "ports-set":
|
|
args := flag.NewFlagSet("proxy:ports-set", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
_, portMaps := common.ShiftString(args.Args())
|
|
err = proxy.CommandPortsSet(appName, portMaps)
|
|
case "report":
|
|
args := flag.NewFlagSet("proxy:report", flag.ExitOnError)
|
|
osArgs, infoFlag, flagErr := common.ParseReportArgs("proxy", os.Args[2:])
|
|
if flagErr == nil {
|
|
args.Parse(osArgs)
|
|
appName := args.Arg(0)
|
|
err = proxy.CommandReport(appName, infoFlag)
|
|
}
|
|
case "set":
|
|
args := flag.NewFlagSet("proxy:set", flag.ExitOnError)
|
|
args.Parse(os.Args[2:])
|
|
appName := args.Arg(0)
|
|
proxyType := args.Arg(1)
|
|
err = proxy.CommandSet(appName, proxyType)
|
|
default:
|
|
common.LogFail(fmt.Sprintf("Invalid plugin subcommand call: %s", subcommand))
|
|
}
|
|
|
|
if err != nil {
|
|
common.LogFail(err.Error())
|
|
}
|
|
}
|