import logging

from jedi import Script
from pygls.lsp.methods import COMPLETION, INITIALIZE
from pygls.lsp.types import (CompletionItem, CompletionItemKind,
                             CompletionList, CompletionOptions,
                             CompletionParams, InsertTextFormat, Position)
from pygls.server import LanguageServer
from pygls.workspace import Document

LOG = logging.getLogger()
LOG.setLevel(logging.DEBUG)
LOG.addHandler(logging.FileHandler("/tmp/a.log"))

LS = LanguageServer('italianswirls', 'v0.0.1')

JEDI_COMPLETION_TYPE_MAP = {
    "module": CompletionItemKind.Module,
    "class": CompletionItemKind.Class,
    "instance": CompletionItemKind.Variable,
    "function": CompletionItemKind.Function,
    "param": CompletionItemKind.Variable,
    "path": CompletionItemKind.File,
    "keyword": CompletionItemKind.Keyword,
    "property": CompletionItemKind.Property,
    "statement": CompletionItemKind.Variable,
}


def get_jedi_script(document: Document) -> Script:
    """Get Jedi Script object from this document and project."""
    return Script(code=document.source, path=document.path)


def get_jedi_position(position: Position) -> tuple[int, int]:
    """Translate pygls's Position to Jedi position (line is 1-based)."""
    return position.line + 1, position.character


def get_pygls_compl_kind(jedi_compl_type: str) -> CompletionItemKind:
    return JEDI_COMPLETION_TYPE_MAP.get(
        jedi_compl_type,
        CompletionItemKind.Text
    )


@LS.feature(INITIALIZE)
async def do_initialize(*args):
    LOG.debug("do_initialize 👋")


@LS.feature(COMPLETION, CompletionOptions(trigger_characters=["."]))
async def do_completion(server: LanguageServer, params: CompletionParams):
    """Return completion items."""
    document_uri = params.text_document.uri
    document = server.workspace.get_document(document_uri)
    script = get_jedi_script(document)
    jedi_position = get_jedi_position(params.position)
    jedi_completions = script.complete(*jedi_position)
    LOG.debug(f"Completions: {jedi_completions}")

    completion_items = []
    for jedi_completion in jedi_completions:
        name = jedi_completion.name
        item = CompletionItem(
            label=name,
            filter_text=name,
            kind=get_pygls_compl_kind(jedi_completion.type),
            sort_text=name,
            insert_text_name=name,
            insert_text_format=InsertTextFormat.PlainText,
        )
        completion_items.append(item)

    return CompletionList(
        is_incomplete=False,
        items=completion_items,
    )


if __name__ == "__main__":
    LS.start_io()