81 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			81 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package golinters
 | |
| 
 | |
| import (
 | |
| 	"strings"
 | |
| 
 | |
| 	"github.com/kulti/thelper/pkg/analyzer"
 | |
| 	"golang.org/x/exp/maps"
 | |
| 	"golang.org/x/tools/go/analysis"
 | |
| 
 | |
| 	"github.com/golangci/golangci-lint/pkg/config"
 | |
| 	"github.com/golangci/golangci-lint/pkg/golinters/goanalysis"
 | |
| )
 | |
| 
 | |
| func NewThelper(cfg *config.ThelperSettings) *goanalysis.Linter {
 | |
| 	a := analyzer.NewAnalyzer()
 | |
| 
 | |
| 	opts := map[string]struct{}{
 | |
| 		"t_name":  {},
 | |
| 		"t_begin": {},
 | |
| 		"t_first": {},
 | |
| 
 | |
| 		"f_name":  {},
 | |
| 		"f_begin": {},
 | |
| 		"f_first": {},
 | |
| 
 | |
| 		"b_name":  {},
 | |
| 		"b_begin": {},
 | |
| 		"b_first": {},
 | |
| 
 | |
| 		"tb_name":  {},
 | |
| 		"tb_begin": {},
 | |
| 		"tb_first": {},
 | |
| 	}
 | |
| 
 | |
| 	if cfg != nil {
 | |
| 		applyTHelperOptions(cfg.Test, "t_", opts)
 | |
| 		applyTHelperOptions(cfg.Fuzz, "f_", opts)
 | |
| 		applyTHelperOptions(cfg.Benchmark, "b_", opts)
 | |
| 		applyTHelperOptions(cfg.TB, "tb_", opts)
 | |
| 	}
 | |
| 
 | |
| 	if len(opts) == 0 {
 | |
| 		linterLogger.Fatalf("thelper: at least one option must be enabled")
 | |
| 	}
 | |
| 
 | |
| 	args := maps.Keys(opts)
 | |
| 
 | |
| 	cfgMap := map[string]map[string]any{
 | |
| 		a.Name: {
 | |
| 			"checks": strings.Join(args, ","),
 | |
| 		},
 | |
| 	}
 | |
| 
 | |
| 	return goanalysis.NewLinter(
 | |
| 		a.Name,
 | |
| 		a.Doc,
 | |
| 		[]*analysis.Analyzer{a},
 | |
| 		cfgMap,
 | |
| 	).WithLoadMode(goanalysis.LoadModeTypesInfo)
 | |
| }
 | |
| 
 | |
| func applyTHelperOptions(o config.ThelperOptions, prefix string, opts map[string]struct{}) {
 | |
| 	if o.Name != nil {
 | |
| 		if !*o.Name {
 | |
| 			delete(opts, prefix+"name")
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if o.Begin != nil {
 | |
| 		if !*o.Begin {
 | |
| 			delete(opts, prefix+"begin")
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if o.First != nil {
 | |
| 		if !*o.First {
 | |
| 			delete(opts, prefix+"first")
 | |
| 		}
 | |
| 	}
 | |
| }
 | 
