Support for autocompletion model attributes and relations in blade files

generatable.json

@@ -68,5 +68,12 @@
             "completion",
             "diagnostics"
         ]
+    },
+    {
+        "type": "model",
+        "features": [
+            "completion_attribute",
+            "completion"
+        ]
     }
 ]

generate-config.php

@@ -24,6 +24,7 @@
                 'hover' => "Enable hover information for {$label}.",
                 'link' => "Enable linking for {$label}.",
                 'completion' => "Enable completion for {$label}.",
+                'completion_attribute' => "Enable completion for {$label} attributes.",
                 default => null,
             },
         ];

package.json

@@ -374,6 +374,18 @@
                     "generated": true,
                     "description": "Enable completion for mix."
                 },
+                "Laravel.model.completion_attribute": {
+                    "type": "boolean",
+                    "default": true,
+                    "generated": true,
+                    "description": "Enable completion for model attributes."
+                },
+                "Laravel.model.completion": {
+                    "type": "boolean",
+                    "default": true,
+                    "generated": true,
+                    "description": "Enable completion for model."
+                },
                 "Laravel.paths.link": {
                     "type": "boolean",
                     "default": true,

php-templates/models.php

@@ -144,11 +144,20 @@ protected function getInfo($className)
 
         $data["extends"] = $this->getParentClass($reflection);
 
+        $name = str($className)->afterLast('\\');
+
+        $data['name_cases'] = array_merge(...array_map(
+            null,
+            $this->getNameCases($name),
+            $this->getNameCases($name->plural())
+        ));
+
         $existingProperties = $this->collectExistingProperties($reflection);
 
         $data['attributes'] = collect($data['attributes'])
             ->map(fn($attrs) => array_merge($attrs, [
                 'title_case' => str($attrs['name'])->title()->replace('_', '')->toString(),
+                'name_cases' => $this->getNameCases(str($attrs['name'])),  
                 'documented' => $existingProperties->contains($attrs['name']),
                 'cast' =>  $this->getCastReturnType($attrs['cast'])
             ]))
@@ -166,6 +175,20 @@ protected function getInfo($className)
             $className => $data,
         ];
     }
+
+    /**
+     * @return array<int, string>
+     */
+    private function getNameCases(\Illuminate\Support\Stringable $name): array
+    {
+        return collect([
+            $name->camel()->toString(),
+            $name->toString(),
+            $name->snake()->toString(),
+            $name->studly()->toString(),
+            $name->studly()->lower()->toString(),
+        ])->unique()->values()->toArray();
+    }     
 };
 
 $builder = new class($docblocks) {

src/completion/Registry.ts

@@ -82,6 +82,18 @@ export class Registry implements vscode.CompletionItemProvider {
             );
         };
 
+        const hasName = (names: FeatureTagParam["name"]) => {
+            if (typeof names === "undefined" || names === null) {
+                return parseResult.name() === null;
+            }
+
+            if (typeof names === "string") {
+                return names === parseResult.name();
+            }
+
+            return names.find((fn) => fn === parseResult.name()) !== undefined;
+        };
+
         const isArgumentIndex = (
             argumentIndex: number | number[] | undefined,
         ) => {
@@ -117,6 +129,7 @@ export class Registry implements vscode.CompletionItemProvider {
                     (tag) =>
                         hasClass(tag.class) &&
                         hasFunc(tag.method) &&
+                        hasName(tag.name) &&
                         isArgumentIndex(tag.argumentIndex) &&
                         isNamedArg(tag.argumentName),
                 );

src/extension.ts

@@ -65,6 +65,7 @@ export async function activate(context: vscode.ExtensionContext) {
         { viteEnvCodeActionProvider },
         { hoverProviders },
         { linkProviders },
+        { completionAttributeProvider, completionModelProvider },
     ] = await Promise.all([
         import("./completion/Registry.js"),
         import("./completion/CompletionProvider.js"),
@@ -78,6 +79,7 @@ export async function activate(context: vscode.ExtensionContext) {
         import("./features/env.js"),
         import("./hover/HoverProvider.js"),
         import("./link/LinkProvider.js"),
+        import("./features/model.js"),
     ]);
 
     console.log("Laravel VS Code Started...");
@@ -135,6 +137,16 @@ export async function activate(context: vscode.ExtensionContext) {
         //     documentSelector,
         //     new BladeFormattingEditProvider(),
         // ),
+        vscode.languages.registerCompletionItemProvider(
+            BLADE_LANGUAGES,
+            new Registry(completionAttributeProvider),
+            ">",
+        ),
+        vscode.languages.registerCompletionItemProvider(
+            BLADE_LANGUAGES,
+            completionModelProvider,
+            "$",
+        ),
         vscode.languages.registerCompletionItemProvider(
             LANGUAGES,
             delegatedRegistry,

src/features/model.ts

@@ -0,0 +1,78 @@
+import AutocompleteResult from "@src/parser/AutocompleteResult";
+import { getModelByName, getModels } from "@src/repositories/models";
+import { config } from "@src/support/config";
+import * as vscode from "vscode";
+import { CompletionProvider, Eloquent, FeatureTag } from "..";
+
+export const completionModelProvider: vscode.CompletionItemProvider = {
+    provideCompletionItems(): vscode.ProviderResult<vscode.CompletionItem[]> {
+        if (!config("model.completion", true)) {
+            return undefined;
+        }
+
+        return Object.entries(getModels().items).flatMap(([, value]) => {
+            return value.name_cases.slice(0, 2).map((name) => {
+                return new vscode.CompletionItem(
+                    name,
+                    vscode.CompletionItemKind.Variable,
+                );
+            });
+        });
+    },
+};
+
+export const completionAttributeProvider: CompletionProvider = {
+    tags() {
+        return Object.values(getModels().items)
+            .flatMap((model) => {
+                return [
+                    {
+                        method: [...model.name_cases],
+                    },
+                    {
+                        name: [...model.name_cases],
+                    },
+                ];
+            })
+            .filter((item) => item !== null) as FeatureTag;
+    },
+
+    provideCompletionItems(
+        result: AutocompleteResult,
+    ): vscode.CompletionItem[] {
+        if (!config("model.completion_attribute", true)) {
+            return [];
+        }
+
+        const name = result.name() ?? result.func();
+
+        if (!name) {
+            return [];
+        }
+
+        const model = getModelByName(name);
+
+        if (!model) {
+            return [];
+        }
+
+        const createCompleteItem = (
+            item: Eloquent.Attribute | Eloquent.Relation,
+        ) => {
+            let completeItem = new vscode.CompletionItem(
+                item.name,
+                vscode.CompletionItemKind.Property,
+            );
+
+            if (item.type) {
+                completeItem.detail = item.type;
+            }
+
+            return completeItem;
+        };
+
+        return model.attributes
+            .map(createCompleteItem)
+            .concat(model.relations.map(createCompleteItem));
+    },
+};

src/index.d.ts

@@ -48,6 +48,7 @@ type LinkProvider = (
 interface FeatureTagParam {
     class?: string | string[] | null;
     method?: string | string[] | null;
+    name?: string | string[] | null;
     argumentName?: string | string[];
     classDefinition?: string;
     methodDefinition?: string;
@@ -83,6 +84,7 @@ declare namespace Eloquent {
         observers: Observer[];
         scopes: string[];
         extends: string | null;
+        name_cases: string[];
     }
 
     interface Attribute {
@@ -97,6 +99,7 @@ declare namespace Eloquent {
         appended: null;
         cast: string | null;
         title_case: string;
+        name_cases: string[];
         documented: boolean;
     }
 

src/parser/AutocompleteResult.ts

@@ -40,6 +40,19 @@ export default class AutocompleteResult {
         return this.param()?.autocompletingValue ?? false;
     }
 
+    public name() {
+        // @ts-ignore
+        return this.result.name ?? null;
+    }
+
+    public isName(name: string | string[]) {
+        if (Array.isArray(name)) {
+            return name.includes(this.name());
+        }
+
+        return this.name() === name;
+    }
+
     public class() {
         // @ts-ignore
         return this.result.className ?? null;

src/repositories/models.ts

@@ -22,6 +22,14 @@ const load = () => {
     });
 };
 
+export const getModelByName = (name: string): Eloquent.Model | undefined => {
+    const model = Object.entries(getModels().items).find(([, value]) => {
+        return value.name_cases.includes(name);
+    });
+
+    return model?.[1];
+};
+
 export const getModels = repository<Eloquent.Models>({
     load,
     pattern: modelPaths

src/support/generated-config.ts

@@ -40,6 +40,8 @@ export type GeneratedConfigKey =
     | "mix.hover"
     | "mix.link"
     | "mix.completion"
+    | "model.completion_attribute"
+    | "model.completion"
     | "paths.link"
     | "route.diagnostics"
     | "route.hover"

src/templates/models.ts

@@ -144,11 +144,20 @@ $models = new class($factory) {
 
         $data["extends"] = $this->getParentClass($reflection);
 
+        $name = str($className)->afterLast('\\\\');
+
+        $data['name_cases'] = array_merge(...array_map(
+            null,
+            $this->getNameCases($name),
+            $this->getNameCases($name->plural())
+        ));
+
         $existingProperties = $this->collectExistingProperties($reflection);
 
         $data['attributes'] = collect($data['attributes'])
             ->map(fn($attrs) => array_merge($attrs, [
                 'title_case' => str($attrs['name'])->title()->replace('_', '')->toString(),
+                'name_cases' => $this->getNameCases(str($attrs['name'])),  
                 'documented' => $existingProperties->contains($attrs['name']),
                 'cast' =>  $this->getCastReturnType($attrs['cast'])
             ]))
@@ -166,6 +175,20 @@ $models = new class($factory) {
             $className => $data,
         ];
     }
+
+    /**
+     * @return array<int, string>
+     */
+    private function getNameCases(\\Illuminate\\Support\\Stringable $name): array
+    {
+        return collect([
+            $name->camel()->toString(),
+            $name->toString(),
+            $name->snake()->toString(),
+            $name->studly()->toString(),
+            $name->studly()->lower()->toString(),
+        ])->unique()->values()->toArray();
+    }     
 };
 
 $builder = new class($docblocks) {

src/types.ts

@@ -15,7 +15,8 @@ export namespace AutocompleteParsingResult {
         | Parameter
         | ParameterValue
         | Parameters
-        | StringValue;
+        | StringValue
+        | Variable;
 
     export interface Argument {
         type: "argument";
@@ -143,4 +144,10 @@ export namespace AutocompleteParsingResult {
             column: number;
         };
     }
+
+    export interface Variable {
+        type: "variable";
+        parent: ContextValue | null;
+        name: string | null;
+    }
 }