check-safety/check_codegen.go
2026-06-20 12:40:06 +08:00

79 lines
2.4 KiB
Go

package main
import (
"fmt"
"os"
"strings"
)
func init() {
register(Check{
Seq: 25,
ID: "2f",
Title: "sqlc compile and buf generate",
Run: func(ctx *ScanContext, rep *Reporter) {
// Check 2f: sqlc compile and buf generate
fmt.Println("=== Check 2f: sqlc compile and buf generate ===")
codegenTargets := discoverCodegenTargets(ctx.repoRoot, ctx.backendDir, ctx.backendTargets, ctx.pluginTargets, ctx.includeCoreTargets)
if len(codegenTargets) > 0 {
completedCodegenTargets, codegenFailures, codegenErr := runCodegenChecks(codegenTargets)
if codegenErr != nil {
fmt.Printf(" ERROR: failed to run codegen checks: %v\n", codegenErr)
os.Exit(2)
}
if len(codegenFailures) > 0 {
fmt.Printf(" FAIL: %d codegen target(s) failed sqlc compile or buf generate:\n", len(codegenFailures))
for _, failure := range codegenFailures {
fmt.Printf(" [%s] %s\n", failure.stage, failure.target)
if failure.output != "" {
for line := range strings.SplitSeq(failure.output, "\n") {
fmt.Printf(" %s\n", line)
}
}
}
rep.Fail()
} else {
fmt.Printf(" OK: sqlc compile / buf generate clean for %d target(s)\n", len(completedCodegenTargets))
for _, target := range completedCodegenTargets {
fmt.Printf(" - %s\n", target)
}
}
} else {
fmt.Println(" SKIP: no sqlc or buf codegen targets found")
}
protoFreshness, protoFreshnessErr := checkProtoGeneratedFreshness(ctx.repoRoot)
if protoFreshnessErr != nil {
fmt.Printf(" ERROR: failed to run make proto freshness check: %v\n", protoFreshnessErr)
if protoFreshness.output != "" {
for line := range strings.SplitSeq(protoFreshness.output, "\n") {
fmt.Printf(" %s\n", line)
}
}
rep.Fail()
} else if protoFreshness.changed() {
fmt.Println(" FAIL: make proto changed generated outputs; run make proto and commit the generated files.")
fmt.Println(" before:")
printIndentedStatus(protoFreshness.beforeStatus)
fmt.Println(" after:")
printIndentedStatus(protoFreshness.afterStatus)
rep.Fail()
} else if protoFreshness.checked {
fmt.Println(" OK: make proto freshness check clean")
}
fmt.Println()
},
})
}
func printIndentedStatus(status string) {
if strings.TrimSpace(status) == "" {
fmt.Println(" (clean)")
return
}
for line := range strings.SplitSeq(status, "\n") {
fmt.Printf(" %s\n", line)
}
}