Merge pull request #51 from sashamelentyev/feat/check-http-method-in-ifstmt

feat: add check HTTP method in ifstmt

Author: sashamelentyev

pkg/analyzer/analyzer.go

@@ -205,20 +205,25 @@ func run(pass *analysis.Pass) (interface{}, error) {
 				return
 			}
 
-			if selectorExpr.Sel.Name != "StatusCode" {
-				return
-			}
-
 			basicLit, ok := binaryExpr.Y.(*ast.BasicLit)
 			if !ok {
 				return
 			}
 
-			if !lookupFlag(pass, HTTPStatusCodeFlag) {
-				return
-			}
+			switch selectorExpr.Sel.Name {
+			case "StatusCode":
+				if !lookupFlag(pass, HTTPStatusCodeFlag) {
+					return
+				}
+
+				checkHTTPStatusCode(pass, basicLit)
+			case "Method":
+				if !lookupFlag(pass, HTTPMethodFlag) {
+					return
+				}
 
-			checkHTTPStatusCode(pass, basicLit)
+				checkHTTPMethod(pass, basicLit)
+			}
 		}
 	})
 

pkg/analyzer/internal/template/test-httpmethod.go.tmpl

@@ -35,3 +35,31 @@ func _() {
 	}
 {{- end }}
 }
+
+{{ range $key, $value := .Mapping }}
+func _() error {
+    resp, err := http.DefaultClient.Do(&http.Request{})
+    if err != nil {
+        return err
+    }
+    defer func() { _ = resp.Body.Close() }()
+    if resp.Request.Method == "{{ $key }}" { // want `"{{ quoteMeta $key }}" can be replaced by {{ quoteMeta $value }}`
+        return nil
+    }
+    return nil
+}
+{{ end -}}
+
+{{ range $key, $value := .Mapping }}
+func _() error {
+    resp, err := http.DefaultClient.Do(&http.Request{})
+    if err != nil {
+        return err
+    }
+    defer func() { _ = resp.Body.Close() }()
+    if resp.Request.Method == {{ $value }} {
+        return nil
+    }
+    return nil
+}
+{{ end -}}