Skip to content

Add fingerprint caching to restore and share commands #71

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
greynewell merged 1 commit into from
Apr 8, 2026
Merged

Add fingerprint caching to restore and share commands #71

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 15 additions & 0 deletions cmd/restore.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ import (
"github.com/spf13/cobra"

"github.com/supermodeltools/cli/internal/api"
"github.com/supermodeltools/cli/internal/build"
"github.com/supermodeltools/cli/internal/cache"
"github.com/supermodeltools/cli/internal/config"
"github.com/supermodeltools/cli/internal/restore"
Expand Down Expand Up @@ -112,6 +113,15 @@ func runRestore(cmd *cobra.Command, dir string, localMode bool, maxTokens int) e
}

func restoreViaAPI(cmd *cobra.Command, cfg *config.Config, rootDir, projectName string) (*restore.ProjectGraph, error) {
// Fast-path: check fingerprint cache before uploading.
if fp, err := cache.RepoFingerprint(rootDir); err == nil {
key := cache.AnalysisKey(fp, "restore", build.Version)
var cached api.SupermodelIR
if hit, _ := cache.GetJSON(key, &cached); hit {
return restore.FromSupermodelIR(&cached, projectName), nil
}
}

zipPath, err := restoreCreateZip(rootDir)
if err != nil {
return nil, fmt.Errorf("create archive: %w", err)
Expand All @@ -134,6 +144,11 @@ func restoreViaAPI(cmd *cobra.Command, cfg *config.Config, rootDir, projectName
return nil, err
}

if fp, err := cache.RepoFingerprint(rootDir); err == nil {
key := cache.AnalysisKey(fp, "restore", build.Version)
_ = cache.PutJSON(key, ir)
}

graph := restore.FromSupermodelIR(ir, projectName)
return graph, nil
}
Expand Down
68 changes: 52 additions & 16 deletions cmd/share.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import (

"github.com/supermodeltools/cli/internal/api"
"github.com/supermodeltools/cli/internal/audit"
"github.com/supermodeltools/cli/internal/build"
"github.com/supermodeltools/cli/internal/cache"
"github.com/supermodeltools/cli/internal/config"
)
Expand Down Expand Up @@ -49,15 +50,20 @@ func runShare(cmd *cobra.Command, dir string) error {
return err
}

// Run the full audit pipeline.
ir, err := shareAnalyze(cmd, cfg, rootDir, projectName)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
defer cancel()

fp, _ := cache.RepoFingerprint(rootDir)

// Run the full audit pipeline (shares cache with `supermodel audit`).
ir, err := shareAnalyze(ctx, cmd, cfg, rootDir, projectName, fp)
if err != nil {
return err
}

report := audit.Analyze(ir, projectName)

impact, err := runImpactForShare(cmd, cfg, rootDir)
impact, err := runImpactForShare(ctx, cmd, cfg, rootDir, fp)
if err != nil {
fmt.Fprintf(cmd.ErrOrStderr(), "Warning: impact analysis unavailable: %v\n", err)
} else {
Expand All @@ -70,11 +76,11 @@ func runShare(cmd *cobra.Command, dir string) error {

// Upload and get public URL.
client := api.New(cfg)
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
defer cancel()
uploadCtx, uploadCancel := context.WithTimeout(context.Background(), 2*time.Minute)
defer uploadCancel()

fmt.Fprintln(cmd.ErrOrStderr(), "Uploading report…")
url, err := client.Share(ctx, api.ShareRequest{
url, err := client.Share(uploadCtx, api.ShareRequest{
ProjectName: projectName,
Status: string(report.Status),
Content: buf.String(),
Expand All @@ -92,11 +98,20 @@ func runShare(cmd *cobra.Command, dir string) error {
return nil
}

func shareAnalyze(cmd *cobra.Command, cfg *config.Config, rootDir, projectName string) (*api.SupermodelIR, error) {
func shareAnalyze(ctx context.Context, cmd *cobra.Command, cfg *config.Config, rootDir, projectName, fp string) (*api.SupermodelIR, error) {
if err := cfg.RequireAPIKey(); err != nil {
return nil, err
}

// Check fingerprint cache (shares results with `supermodel audit`).
if fp != "" {
key := cache.AnalysisKey(fp, "audit-domains", build.Version)
var cached api.SupermodelIR
if hit, _ := cache.GetJSON(key, &cached); hit {
return &cached, nil
}
}

fmt.Fprintln(cmd.ErrOrStderr(), "Creating repository archive…")
zipPath, err := audit.CreateZip(rootDir)
if err != nil {
Expand All @@ -110,14 +125,29 @@ func shareAnalyze(cmd *cobra.Command, cfg *config.Config, rootDir, projectName s
}

client := api.New(cfg)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
defer cancel()

fmt.Fprintf(cmd.ErrOrStderr(), "Analyzing %s…\n", projectName)
return client.AnalyzeDomains(ctx, zipPath, "share-"+hash[:16])
ir, err := client.AnalyzeDomains(ctx, zipPath, "share-"+hash[:16])
if err != nil {
return nil, err
}

if fp != "" {
key := cache.AnalysisKey(fp, "audit-domains", build.Version)
_ = cache.PutJSON(key, ir)
}
return ir, nil
}

func runImpactForShare(cmd *cobra.Command, cfg *config.Config, rootDir string) (*api.ImpactResult, error) {
func runImpactForShare(ctx context.Context, cmd *cobra.Command, cfg *config.Config, rootDir, fp string) (*api.ImpactResult, error) {
// Check fingerprint cache (shares results with `supermodel audit` and `supermodel blast-radius`).
if fp != "" {
key := cache.AnalysisKey(fp, "impact", build.Version)
var cached api.ImpactResult
if hit, _ := cache.GetJSON(key, &cached); hit {
return &cached, nil
}
}

zipPath, err := audit.CreateZip(rootDir)
if err != nil {
return nil, err
Expand All @@ -130,11 +160,17 @@ func runImpactForShare(cmd *cobra.Command, cfg *config.Config, rootDir string) (
}

client := api.New(cfg)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
defer cancel()

fmt.Fprintln(cmd.ErrOrStderr(), "Running impact analysis…")
return client.Impact(ctx, zipPath, "share-impact-"+hash[:16], "", "")
result, err := client.Impact(ctx, zipPath, "share-impact-"+hash[:16], "", "")
if err != nil {
return nil, err
}

if fp != "" {
key := cache.AnalysisKey(fp, "impact", build.Version)
_ = cache.PutJSON(key, result)
}
return result, nil
}

// resolveAuditDir and findGitRoot are defined in cmd/audit.go (same package).
Loading