generate.go

//go:build ignore

package main

import (
	"bytes"
	"go/format"
	"log"
	"os"
	"path/filepath"
	"regexp"
	"text/template"
)

var tmpl = `
{{define "analyzers"}}
// Code generated by generate.go. DO NOT EDIT.

package {{.dir}}

import (
	"honnef.co/go/tools/analysis/lint"
{{- range $check := .checks}}
	{{- if not (index $.disabled $check) }}
	"honnef.co/go/tools/{{$.dir}}/{{$check}}"
	{{- end }}
{{- end}}
)

var Analyzers = []*lint.Analyzer{
{{- range $check := .checks}}
	{{- if not (index $.disabled $check) }}
	{{$check}}.SCAnalyzer,
	{{- end }}
{{- end}}
}
{{end}}

{{define "tests"}}
// Code generated by generate.go. DO NOT EDIT.

package {{.check}}

import (
	"testing"

	{{ if not (index .disabled .check) }}
	"honnef.co/go/tools/analysis/lint/testutil"
	{{- end }}
)

func TestTestdata(t *testing.T) {
	{{- if index .disabled .check }}
		t.Skip("check has been disabled")
	{{- else }}
		testutil.Run(t, SCAnalyzer)
	{{- end }}
}
{{end}}
`

var disabledChecks = map[string]bool{
	// SA5011 relied on SSI and no longer works correctly.
	"sa5011": true,
}

func main() {
	log.SetFlags(0)

	dir, err := os.Getwd()
	if err != nil {
		log.Fatalln("couldn't determine current directory:", err)
	}

	dir = filepath.Base(dir)

	var t template.Template
	if _, err = t.Parse(tmpl); err != nil {
		log.Fatalln("couldn't parse templates:", err)
	}

	dirs, err := filepath.Glob("*")
	if err != nil {
		log.Fatalln("couldn't enumerate checks:", err)
	}

	checkRe := regexp.MustCompile(`^[a-z]+\d{4}$`)
	out := dirs[:0]
	for _, dir := range dirs {
		if checkRe.MatchString(dir) {
			out = append(out, dir)
		}
	}
	dirs = out

	buf := bytes.NewBuffer(nil)

	if err := t.ExecuteTemplate(buf, "analyzers",
		map[string]any{
			"checks":   dirs,
			"dir":      dir,
			"disabled": disabledChecks,
		}); err != nil {
		log.Fatalln("couldn't generate analysis.go:", err)
	}

	b, err := format.Source(buf.Bytes())
	if err != nil {
		log.Fatalln("couldn't gofmt analysis.go:", err)
	}
	if err := os.WriteFile("analysis.go", b, 0666); err != nil {
		log.Fatalln("couldn't write analysis.go:", err)
	}

	for _, dir := range dirs {
		buf.Reset()
		dst := filepath.Join(dir, dir+"_test.go")
		if err := t.ExecuteTemplate(buf, "tests",
			map[string]any{
				"check":    dir,
				"disabled": disabledChecks,
			}); err != nil {
			log.Fatalf("couldn't generate %s: %s", dst, err)
		}

		b, err := format.Source(buf.Bytes())
		if err != nil {
			log.Fatalf("couldn't gofmt %s: %s", dst, err)
		}
		if err := os.WriteFile(dst, b, 0666); err != nil {
			log.Fatalf("couldn't write %s: %s", dst, err)
		}
	}
}