consolidate model builder type, clarify variable names

Author: joetannenbaum

src/support/docblocks.ts

@@ -9,8 +9,9 @@ interface ClassBlock {
     blocks: string[];
 }
 
-// TODO: Chunk into several files if we have a lot of models?
-// TODO: Check if doc block is already in model file, skip if it is
+const modelBuilderType = (className: string) =>
+    `\\Illuminate\\Database\\Eloquent\\Builder<${className}>|${className}`;
+
 export const writeEloquentDocBlocks = (
     models: Eloquent.Models,
     builderMethods: Eloquent.BuilderMethod[],
@@ -77,25 +78,27 @@ const getBuilderReturnType = (
     }
 
     if (["static", "self"].includes(method.return)) {
-        return `\\Illuminate\\Database\\Eloquent\\Builder<${className}>|${className}`;
+        return modelBuilderType(className);
     }
 
-    const certainOfType = ["sole", "find", "first", "firstOrFail"].includes(method.name);
+    const returnsSingleModel = [
+        "sole",
+        "find",
+        "first",
+        "firstOrFail",
+    ].includes(method.name);
 
     const returnType = method.return
-        .replace(
-            "$this",
-            `\\Illuminate\\Database\\Eloquent\\Builder<${className}>|${className}`,
-        )
+        .replace("$this", modelBuilderType(className))
         .replace("\\TReturn", "mixed")
         .replace("TReturn", "mixed")
-        .replace("\\TValue", certainOfType ? className : "mixed")
-        .replace("TValue", certainOfType ? className : "mixed")
-        .replace("object", certainOfType ? className : "mixed");
-    
+        .replace("\\TValue", returnsSingleModel ? className : "mixed")
+        .replace("TValue", returnsSingleModel ? className : "mixed")
+        .replace("object", returnsSingleModel ? className : "mixed");
+
     if (returnType.includes("mixed")) {
         return "mixed";
-    };
+    }
 
     return returnType;
 };
@@ -110,7 +113,7 @@ const getBlocks = (
         .concat(
             [...model.scopes, "newModelQuery", "newQuery", "query"].map(
                 (method) => {
-                    return `@method static \\Illuminate\\Database\\Eloquent\\Builder<${className}>|${className} ${method}()`;
+                    return `@method static ${modelBuilderType(className)} ${method}()`;
                 },
             ),
         )
@@ -202,7 +205,7 @@ const getAttributeBlocks = (
 
     if (!["accessor", "attribute"].includes(attr.cast || "")) {
         blocks.push(
-            `@method static \\Illuminate\\Database\\Eloquent\\Builder<${className}>|${className} where${attr.title_case}($value)`,
+            `@method static ${modelBuilderType(className)} where${attr.title_case}($value)`,
         );
     }