119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package cmds
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"text/tabwriter"
|
|
|
|
"github.com/spf13/cobra"
|
|
"lmika.dev/lmika/weiro/config"
|
|
"lmika.dev/lmika/weiro/models"
|
|
"lmika.dev/lmika/weiro/services"
|
|
)
|
|
|
|
func PubTargetsAdd() *cobra.Command {
|
|
var (
|
|
siteGUID string
|
|
targetType string
|
|
targetRef string
|
|
targetKey string
|
|
baseURL string
|
|
enabled bool
|
|
)
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "add",
|
|
Short: "Add a publication target",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cfg, err := config.LoadConfig()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
svcs, err := services.New(cfg)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer svcs.Close()
|
|
|
|
ctx := context.Background()
|
|
site, err := svcs.DB.SelectSiteByGUID(ctx, siteGUID)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
target := &models.SitePublishTarget{
|
|
SiteID: site.ID,
|
|
GUID: models.NewNanoID(),
|
|
Enabled: enabled,
|
|
BaseURL: baseURL,
|
|
TargetType: targetType,
|
|
TargetRef: targetRef,
|
|
TargetKey: targetKey,
|
|
}
|
|
|
|
if err := svcs.DB.SavePublishTarget(ctx, target); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Printf("Added publish target %s\n", target.GUID)
|
|
},
|
|
}
|
|
|
|
cmd.Flags().StringVarP(&siteGUID, "site", "s", "", "Site GUID")
|
|
cmd.Flags().StringVarP(&targetType, "type", "t", "", "Target type (localfs, netlify)")
|
|
cmd.Flags().StringVarP(&targetRef, "ref", "r", "", "Target reference")
|
|
cmd.Flags().StringVarP(&targetKey, "key", "k", "", "Target key")
|
|
cmd.Flags().StringVarP(&baseURL, "url", "u", "", "Base URL")
|
|
cmd.Flags().BoolVar(&enabled, "enabled", true, "Enable target")
|
|
|
|
cmd.MarkFlagRequired("site")
|
|
cmd.MarkFlagRequired("type")
|
|
cmd.MarkFlagRequired("ref")
|
|
cmd.MarkFlagRequired("url")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func PubTargets() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "pubtargets <site-guid>",
|
|
Short: "Manage publication targets",
|
|
Args: cobra.ExactArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cfg, err := config.LoadConfig()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
svcs, err := services.New(cfg)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer svcs.Close()
|
|
|
|
ctx := context.Background()
|
|
site, err := svcs.DB.SelectSiteByGUID(ctx, args[0])
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
targets, err := svcs.DB.SelectPublishTargetsOfSite(ctx, site.ID)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
|
|
fmt.Fprintln(w, "GUID\tTARGET_TYPE\tENABLED\tTARGET_REF")
|
|
for _, target := range targets {
|
|
fmt.Fprintf(w, "%s\t%s\t%v\t%s\n", target.GUID, target.TargetType, target.Enabled, target.TargetRef)
|
|
}
|
|
w.Flush()
|
|
},
|
|
}
|
|
cmd.AddCommand(PubTargetsAdd())
|
|
return cmd
|
|
}
|