70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
|
package proxy
|
||
|
|
||
|
import (
|
||
|
"net/url"
|
||
|
"os"
|
||
|
|
||
|
"forge.cadoles.com/cadoles/bouncer/internal/client"
|
||
|
"forge.cadoles.com/cadoles/bouncer/internal/command/admin/apierr"
|
||
|
clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/flag"
|
||
|
proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/proxy/flag"
|
||
|
"forge.cadoles.com/cadoles/bouncer/internal/format"
|
||
|
"github.com/pkg/errors"
|
||
|
"github.com/urfave/cli/v2"
|
||
|
)
|
||
|
|
||
|
func CreateCommand() *cli.Command {
|
||
|
return &cli.Command{
|
||
|
Name: "create",
|
||
|
Usage: "Create proxy",
|
||
|
Flags: proxyFlag.WithProxyFlags(
|
||
|
&cli.StringFlag{
|
||
|
Name: "to",
|
||
|
Usage: "Set `TO` as proxy's destination url",
|
||
|
Value: "",
|
||
|
Required: true,
|
||
|
},
|
||
|
&cli.StringSliceFlag{
|
||
|
Name: "from",
|
||
|
Usage: "Set `FROM` as proxy's patterns to match incoming requests",
|
||
|
Value: cli.NewStringSlice("*"),
|
||
|
},
|
||
|
),
|
||
|
Action: func(ctx *cli.Context) error {
|
||
|
baseFlags := clientFlag.GetBaseFlags(ctx)
|
||
|
|
||
|
token, err := clientFlag.GetToken(baseFlags)
|
||
|
if err != nil {
|
||
|
return errors.WithStack(apierr.Wrap(err))
|
||
|
}
|
||
|
|
||
|
proxyName, err := proxyFlag.AssertProxyName(ctx)
|
||
|
if err != nil {
|
||
|
return errors.Wrap(err, "'to' parameter should be a valid url")
|
||
|
}
|
||
|
|
||
|
to, err := url.Parse(ctx.String("to"))
|
||
|
if err != nil {
|
||
|
return errors.Wrap(err, "'to' parameter should be a valid url")
|
||
|
}
|
||
|
|
||
|
from := ctx.StringSlice("from")
|
||
|
|
||
|
client := client.New(baseFlags.ServerURL, client.WithToken(token))
|
||
|
|
||
|
proxy, err := client.CreateProxy(ctx.Context, proxyName, to, from)
|
||
|
if err != nil {
|
||
|
return errors.WithStack(apierr.Wrap(err))
|
||
|
}
|
||
|
|
||
|
hints := proxyHints(baseFlags.OutputMode)
|
||
|
|
||
|
if err := format.Write(baseFlags.Format, os.Stdout, hints, proxy); err != nil {
|
||
|
return errors.WithStack(err)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
},
|
||
|
}
|
||
|
}
|