Improved auto complete

[aszenz]

Idea Statement

Currently I see this tool manually defines completion's in the Monaco editor, I think using duckdbs own auto complete engine via the extensions may give better results.

Feature implementation brainstorm

See:
https://duckdb.org/docs/api/cli/autocomplete.html
It works on wasm

const query = `select suggestion from sql_auto_complete('${escape(input)}')`;

[caioricciuti]

I can't use the autocomplete since there's no direct call to mocaco (editor) to keeping creating the suggestions but I add the funcions and keywords directly from the duckDb settings.

 const keywordsResult: any = await executeQuery(
        "SELECT DISTINCT(keyword_name) FROM duckdb_keywords()"
      );

      const functionsResult: any = await executeQuery(
        "SELECT DISTINCT ON (function_name) * FROM DUCKDB_FUNCTIONS();"
      );

[aszenz]

I can't use the autocomplete since there's no direct call to mocaco (editor) to keeping creating the suggestions but I add the funcions and keywords directly from the duckDb settings.

Not sure I understood the reason, in my own implementation I pass the input to the auto complete extension via provideCompletionItems.

monaco.languages.registerCompletionItemProvider('sql', {
    async provideCompletionItems(model, position) {
        const word = model.getWordUntilPosition(position);
        const conn = await db.connect();
        const range = {
            startLineNumber: position.lineNumber,
            endLineNumber: position.lineNumber,
            startColumn: word.startColumn,
            endColumn: word.endColumn,
        };
        const textInRange = model.getValueInRange({
            startColumn: 0,
            endColumn: position.column,
            startLineNumber: position.lineNumber,
            endLineNumber: position.lineNumber,
        });
        const query = `select suggestion from sql_auto_complete('${escape(textInRange)}')`;
        const items = await queryNative<{ suggestion: arrow.DataType<arrow.Type.Utf8> }>(conn, query);
        if (!items) return { suggestions: [] };
        const suggestions = items.values.map((item) => {
            return {
                label: String(item.suggestion),
                kind: monaco.languages.CompletionItemKind.Field,
                insertText: String(item.suggestion),
                range,
            };
        });
        return {
            suggestions,
        };
    },
});

Note that the engine provides suggestions based on the context, and it got improved a lot in v1.2.0, so it's much better than plain matching.

Of course it also keeps suggestions in sync as the user adds db's/files etc.

[caioricciuti]

HEYYY! @aszenz genius thankss! I got it now and you are right it works like a charm, I just didn't implemented it right! Thanks for sharing your code!

PD: I'll release that on the next one!