fix: track divergent shader loop controls

Author: aidenybai

packages/oxlint-plugin-react-doctor/src/plugin/rules/r3f/three-shader-no-derivatives-in-nonuniform-flow.test.ts

@@ -15,6 +15,8 @@ describe("three-shader-no-derivatives-in-nonuniform-flow", () => {
     `import { ShaderMaterial } from "three"; new ShaderMaterial({ fragmentShader: "varying float choice; void main() { switch (int(choice)) { case 0: gl_FragColor = vec4(fwidth(choice)); break; default: break; } }" });`,
     `import { ShaderMaterial } from "three"; new ShaderMaterial({ fragmentShader: "/* #define fwidth(value) value */ varying float value; void main() { if (value > 0.0) gl_FragColor = vec4(fwidth(value)); }" });`,
     `import { ShaderMaterial } from "three"; new ShaderMaterial({ fragmentShader: "varying float value; void main() { do { gl_FragColor = vec4(fwidth(value)); } while (value > 0.0); }" });`,
+    `import { ShaderMaterial } from "three"; new ShaderMaterial({ fragmentShader: "varying float value; void main() { for (float index = value; index < 1.0; index += 0.25) { gl_FragColor = vec4(fwidth(value)); } }" });`,
+    `import { ShaderMaterial } from "three"; new ShaderMaterial({ fragmentShader: "varying float value; void main() { for (float index = 0.0; index < 2.0; index += value) { gl_FragColor = vec4(fwidth(value)); } }" });`,
   ])("reports derivatives in proven fragment-dependent control flow", (code) => {
     expect(runRule(threeShaderNoDerivativesInNonuniformFlow, code).diagnostics).toHaveLength(1);
   });

packages/oxlint-plugin-react-doctor/src/plugin/rules/r3f/three-shader-no-derivatives-in-nonuniform-flow.ts

@@ -46,7 +46,9 @@ const getControllingExpressions = (path: Path<FunctionCallNode>): AstNode[] => {
       controllingExpressions.push(parent.expression);
     }
     if (parent.type === "for_statement" && currentPath.key === "body" && parent.condition) {
-      controllingExpressions.push(parent.condition);
+      controllingExpressions.push(
+        ...[parent.init, parent.condition, parent.operation].filter(Boolean),
+      );
     }
     if (parent.type === "ternary" && (currentPath.key === "left" || currentPath.key === "right")) {
       controllingExpressions.push(parent.expression);