feat: add contextcheck linter

Author: sylvia7788

go.mod

@@ -76,6 +76,7 @@ require (
 	github.com/spf13/viper v1.9.0
 	github.com/ssgreg/nlreturn/v2 v2.2.1
 	github.com/stretchr/testify v1.7.0
+	github.com/sylvia7788/contextcheck v1.0.4
 	github.com/tdakkota/asciicheck v0.0.0-20200416200610-e657995f937b
 	github.com/tetafro/godot v1.4.11
 	github.com/timakin/bodyclose v0.0.0-20200424151742-cb6215831a94

go.sum

@@ -0,0 +1,18 @@
+package golinters
+
+import (
+	"github.com/sylvia7788/contextcheck"
+	"golang.org/x/tools/go/analysis"
+
+	"github.com/golangci/golangci-lint/pkg/golinters/goanalysis"
+)
+
+func NewContextCheck() *goanalysis.Linter {
+	analyzer := contextcheck.NewAnalyzer()
+	return goanalysis.NewLinter(
+		"contextcheck",
+		"check for using context.Background() and context.TODO() directly",
+		[]*analysis.Analyzer{analyzer},
+		nil,
+	).WithLoadMode(goanalysis.LoadModeTypesInfo)
+}

pkg/golinters/contextcheck.go

@@ -522,6 +522,11 @@ func (m Manager) GetAllSupportedLinterConfigs() []*linter.Config {
 			WithLoadForGoAnalysis().
 			WithURL("https://github.com/Antonboom/nilnil").
 			WithSince("v1.43.0"),
+		linter.NewConfig(golinters.NewContextCheck()).
+			WithPresets(linter.PresetBugs).
+			WithLoadForGoAnalysis().
+			WithURL("https://github.com/sylvia7788/contextcheck").
+			WithSince("v1.0.0"),
 
 		// nolintlint must be last because it looks at the results of all the previous linters for unused nolint directives
 		linter.NewConfig(golinters.NewNoLintLint()).

pkg/lint/lintersdb/manager.go

@@ -0,0 +1,40 @@
+//args: -Econtextcheck
+package testdata
+
+import "context"
+
+type MyString string
+
+func contextcheckCase1(ctx context.Context) {
+	funcWithoutCtx() // ERROR "Function `funcWithoutCtx` should pass the context parameter"
+}
+
+func contextcheckCase2(ctx context.Context) {
+	ctx = context.WithValue(ctx, MyString("aaa"), "aaaaaa")
+	funcWithCtx(ctx)
+
+	defer func() {
+		funcWithCtx(ctx)
+	}()
+
+	func(ctx context.Context) {
+		funcWithCtx(ctx)
+	}(ctx)
+
+	funcWithCtx(context.Background()) // ERROR `The context param may be context.TODO\(\) or context.Background\(\), please replace it with another way, such as context.WithValue\(ctx, key, val\)`
+}
+
+func contextcheckCase3(ctx context.Context) {
+	func() {
+		funcWithCtx(ctx)
+	}()
+
+	ctx = context.Background() // ERROR `Invalid call to get new context, please replace it with another way, such as context.WithValue\(ctx, key, val\)`
+	funcWithCtx(ctx)
+}
+
+func funcWithCtx(ctx context.Context) {}
+
+func funcWithoutCtx() {
+	funcWithCtx(context.TODO())
+}