rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove proboctl internal admin CLI

Browse Source 
The set-branding functionality it provided is no longer needed.

Signed-off-by: Émile Ré <emile@getprobo.com>
This commit is contained in:
Émile Ré
2026-04-22 15:17:10 +04:00
parent d71915abd4
commit c8e3d139e1
4 changed files with 1 additions and 220 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
pkg/proboctl/cookiebanner/cookiebanner.go
View File

@@ -1,71 +0,0 @@
// Copyright (c) 2026 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package cookiebanner
import (
"fmt"
"strconv"
"github.com/spf13/cobra"
"go.gearno.de/kit/pg"
"go.probo.inc/probo/pkg/cookiebanner"
"go.probo.inc/probo/pkg/gid"
)
type pgClientFactory func() (*pg.Client, error)
func NewCmdCookieBanner(newPG pgClientFactory) *cobra.Command {
cmd := &cobra.Command{
Use: "cookie-banner <command>",
Short: "Manage cookie banners",
}
cmd.AddCommand(newCmdSetBranding(newPG))
return cmd
}
func newCmdSetBranding(newPG pgClientFactory) *cobra.Command {
return &cobra.Command{
Use: "set-branding <banner-id> <true|false>",
Short: "Enable or disable 'Powered by Probo' branding on a banner",
Args: cobra.ExactArgs(2),
RunE: func(cmd *cobra.Command, args []string) error {
bannerID, err := gid.ParseGID(args[0])
if err != nil {
return fmt.Errorf("invalid banner ID: %w", err)
}
show, err := strconv.ParseBool(args[1])
if err != nil {
return fmt.Errorf("invalid boolean value %q: use true or false", args[1])
}
pgClient, err := newPG()
if err != nil {
return fmt.Errorf("cannot connect to database: %w", err)
}
defer pgClient.Close()
svc := cookiebanner.NewService(pgClient)
if err := svc.SetShowBranding(cmd.Context(), bannerID, show); err != nil {
return err
}
fmt.Fprintf(cmd.OutOrStdout(), "show_branding set to %v for banner %s\n", show, bannerID)
return nil
},
}
}

106
pkg/proboctl/root.go
View File

@@ -1,106 +0,0 @@
// Copyright (c) 2026 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package proboctl
import (
"fmt"
"math"
"os"
"strconv"
"github.com/jackc/pgx/v5"
"github.com/spf13/cobra"
"go.gearno.de/kit/pg"
"go.probo.inc/probo/pkg/proboctl/cookiebanner"
)
func NewRootCmd(version string) *cobra.Command {
var pgDSN string
cmd := &cobra.Command{
Use: "proboctl <command> [flags]",
Short: "Probo internal admin CLI",
Long: "proboctl is an internal tool for Probo staff to manage parameters not exposed through the public API.",
SilenceUsage: true,
SilenceErrors: true,
}
cmd.PersistentFlags().StringVar(&pgDSN, "pg-dsn", "", "PostgreSQL connection string (also PG_DSN env var)")
newPG := func() (*pg.Client, error) {
return newPGClient(pgDSN)
}
cmd.AddCommand(cookiebanner.NewCmdCookieBanner(newPG))
return cmd
}
func newPGClient(flagDSN string) (*pg.Client, error) {
dsn := flagDSN
if dsn == "" {
dsn = os.Getenv("PG_DSN")
}
var addr, user, pass, db string
if dsn != "" {
cfg, err := pgx.ParseConfig(dsn)
if err != nil {
return nil, fmt.Errorf("invalid DSN: %w", err)
}
addr = fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)
user = cfg.User
pass = cfg.Password
db = cfg.Database
} else {
addr = envOrDefault("PG_ADDR", "localhost:5432")
user = envOrDefault("PG_USERNAME", "postgres")
pass = envOrDefault("PG_PASSWORD", "postgres")
db = envOrDefault("PG_DATABASE", "probod")
}
poolSize := envIntOrDefault("PG_POOL_SIZE", 2)
if poolSize < 0 || poolSize > math.MaxInt32 {
return nil, fmt.Errorf("PG_POOL_SIZE %d out of range", poolSize)
}
opts := []pg.Option{
pg.WithAddr(addr),
pg.WithUser(user),
pg.WithPassword(pass),
pg.WithDatabase(db),
pg.WithPoolSize(int32(poolSize)),
}
return pg.NewClient(opts...)
}
func envOrDefault(key, fallback string) string {
if v := os.Getenv(key); v != "" {
return v
}
return fallback
}
func envIntOrDefault(key string, fallback int) int {
if v := os.Getenv(key); v != "" {
if n, err := strconv.Atoi(v); err == nil {
return n
}
fmt.Fprintf(os.Stderr, "warning: invalid %s value %q, using default %d\n", key, v, fallback)
}
return fallback
}