c7e2fc209f
Replace the bare `getprobo.com` domain and the `www.getprobo.com` marketing host with `probo.com` / `www.probo.com` across the codebase. Functional subdomains (app, console, notification, custom, test, cookie-banner, compliance) keep their existing `getprobo.com` hosts, and changelog entries are left untouched. Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
113 lines
3.7 KiB
Go
113 lines
3.7 KiB
Go
// Copyright (c) 2026 Probo Inc <hello@probo.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"
|
|
|
|
"github.com/spf13/cobra"
|
|
"go.probo.inc/probo/pkg/cookiebanner"
|
|
"go.probo.inc/probo/pkg/coredata"
|
|
"go.probo.inc/probo/pkg/gid"
|
|
"go.probo.inc/probo/pkg/proboctl/cmdutil"
|
|
)
|
|
|
|
func newCmdResetTrackers(f *cmdutil.Factory) *cobra.Command {
|
|
var (
|
|
flagMappingOnly bool
|
|
flagKeyword string
|
|
flagDryRun bool
|
|
flagYes bool
|
|
)
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "reset-trackers <banner-gid>",
|
|
Short: "Rebuild a banner's tracker patterns from detections and re-arm the analysis + mapping workers",
|
|
Long: "Destructive, tenant-scoped operator action. For a banner's uncategorised, " +
|
|
"non-excluded patterns it clears catalog/vendor links, rebuilds the raw exact " +
|
|
"patterns from detected_trackers (decomposing derived globs), and re-arms the " +
|
|
"pattern-analysis and mapping workers. User-categorised and excluded patterns are " +
|
|
"preserved. With --mapping-only it skips the rebuild and only re-arms mapping. " +
|
|
"With --keyword the rebuild and mapping reset are scoped to patterns whose pattern " +
|
|
"or display name contains the substring.",
|
|
Args: cobra.ExactArgs(1),
|
|
}
|
|
|
|
cmd.Flags().BoolVar(&flagMappingOnly, "mapping-only", false, "Only re-arm mapping (skip the detection rebuild and analysis)")
|
|
cmd.Flags().StringVar(&flagKeyword, "keyword", "", "Only reset patterns whose pattern or display name contains this substring")
|
|
cmd.Flags().BoolVar(&flagDryRun, "dry-run", false, "Print the target banner without writing")
|
|
cmd.Flags().BoolVar(&flagYes, "yes", false, "Skip confirmation")
|
|
|
|
cmd.RunE = func(cmd *cobra.Command, args []string) error {
|
|
bannerID, err := gid.ParseGID(args[0])
|
|
if err != nil {
|
|
return fmt.Errorf("invalid banner GID %q: %w", args[0], err)
|
|
}
|
|
|
|
ctx := cmd.Context()
|
|
|
|
pgClient, err := f.PgClient()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
scope := coredata.NewScopeFromObjectID(bannerID)
|
|
|
|
out := f.IOStreams.Out
|
|
|
|
mode := "full reset"
|
|
if flagMappingOnly {
|
|
mode = "mapping-only reset"
|
|
}
|
|
|
|
var keyword *string
|
|
if flagKeyword != "" {
|
|
keyword = &flagKeyword
|
|
mode = fmt.Sprintf("%s scoped to keyword %q", mode, flagKeyword)
|
|
}
|
|
|
|
if flagDryRun {
|
|
_, _ = fmt.Fprintf(out, "Would run %s on banner %s.\n", mode, bannerID.String())
|
|
return nil
|
|
}
|
|
|
|
if !flagYes {
|
|
return fmt.Errorf("about to run %s on banner %s; pass --yes to proceed or --dry-run to preview", mode, bannerID.String())
|
|
}
|
|
|
|
_, _ = fmt.Fprintf(out, "Running %s on banner %s.\n", mode, bannerID.String())
|
|
|
|
result, err := cookiebanner.ResetBannerTrackers(ctx, pgClient, scope, bannerID, flagMappingOnly, keyword)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot reset banner %s: %w", bannerID, err)
|
|
}
|
|
|
|
_, _ = fmt.Fprintf(
|
|
out,
|
|
"%s: reset %d pattern(s), decomposed %d glob(s) into %d exact(s), relinked %d detection(s), analysis_requested=%t\n",
|
|
bannerID.String(),
|
|
result.PatternsReset,
|
|
result.GlobsDecomposed,
|
|
result.ExactsCreated,
|
|
result.DetectionsRelinked,
|
|
result.AnalysisRequested,
|
|
)
|
|
|
|
return nil
|
|
}
|
|
|
|
return cmd
|
|
}
|