feat: add check HTTP status code in IfStmt (#37)

feat: add check HTTP status code in IfStmt

Author: sashamelentyev

pkg/analyzer/analyzer.go

@@ -55,6 +55,7 @@ func run(pass *analysis.Pass) (interface{}, error) {
 		(*ast.CallExpr)(nil),
 		(*ast.BasicLit)(nil),
 		(*ast.CompositeLit)(nil),
+		(*ast.IfStmt)(nil),
 	}
 
 	insp.Preorder(filter, func(node ast.Node) {
@@ -157,6 +158,28 @@ func run(pass *analysis.Pass) (interface{}, error) {
 					}
 				}
 			}
+
+		case *ast.IfStmt:
+			binaryExpr, ok := n.Cond.(*ast.BinaryExpr)
+			if !ok {
+				return
+			}
+
+			selectorExpr, ok := binaryExpr.X.(*ast.SelectorExpr)
+			if !ok {
+				return
+			}
+
+			if selectorExpr.Sel.Name != "StatusCode" {
+				return
+			}
+
+			basicLit, ok := binaryExpr.Y.(*ast.BasicLit)
+			if !ok {
+				return
+			}
+
+			checkHTTPStatusCode(pass, basicLit)
 		}
 	})
 

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

@@ -29,3 +29,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.StatusCode == {{ $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.StatusCode == {{ $value }} {
+        return nil
+    }
+    return nil
+}
+{{ end -}}