import { toMonacoRange } from '../common';

import { AbstractLanguageFacts, IFunctionBlockInfo } from './base';

import type { SyntaxNode } from 'web-tree-sitter';

/**
 * javascript 中表示代码块的节点类型
 */
export const javascriptBlockCodeTypes = [
  'function',
  'function_declaration',
  'function_expression',
  'arrow_function',
  'class_declaration',
  'method_definition',
  'if_statement',
  'switch_case',
  'while_statement',
  'do_statement',
  'for_statement',
  'for_in_statement',
  'try_statement',
  'catch_clause',
  'block',
  'export_statement',
  'lexical_declaration',
];

export const functionBlockCodeTypes = [
  'function',
  'function_declaration',
  'function_expression',
  'arrow_function',
  'method_definition',
];

const functionBlockSet = new Set(functionBlockCodeTypes);
const blockSet = new Set(javascriptBlockCodeTypes);

export class JavaScriptLanguageFacts implements AbstractLanguageFacts {
  name = 'javascript' as const;
  listCommentStyle = '// ';
  blockCommentStyle = {
    start: '/**',
    end: ' */',
    linePrefix: ' * ',
  };

  provideCodeBlocks(): Set<string> {
    return blockSet;
  }

  isCodeBlock(type: string): boolean {
    return blockSet.has(type);
  }

  isFunctionCodeBlocks(type: string): boolean {
    return functionBlockSet.has(type);
  }

  provideFunctionInfo(node: SyntaxNode): IFunctionBlockInfo | null {
    switch (node.type) {
      case 'function_declaration':
      case 'function_expression':
        return {
          infoCategory: 'function',
          type: node.type,
          name: node.firstNamedChild?.text || '',
          signatures: node.children
            .filter((child) => child.type === 'parameter')
            .map((param) => param.firstChild?.text || ''),
          range: toMonacoRange(node),
        };
      case 'arrow_function': {
        const parent = node.parent;
        // example: const a = () => {}
        if (
          parent &&
          parent.type === 'variable_declarator' &&
          parent.parent &&
          (parent.parent.type === 'lexical_declaration' ||
            parent.parent.type === 'export_statement' ||
            parent.parent.type === 'variable_declaration')
        ) {
          return {
            infoCategory: 'function',
            type: node.type,
            name: parent.firstChild?.text || '',
            signatures: node.children
              .filter((child) => child.type === 'parameter')
              .map((param) => param.firstChild?.text || ''),
            range: toMonacoRange(parent.parent),
          };
        }
        return null;
      }
      case 'method_definition':
        return {
          infoCategory: 'function',
          type: node.type,
          name: node.firstNamedChild?.text || '',
          signatures: node.children
            .filter((child) => child.type === 'parameter')
            .map((param) => param.firstChild?.text || ''),
          range: toMonacoRange(node),
        };
    }

    return null;
  }
}