import type { LanguageAdapter } from '../types';
import { declarationSymbols } from './shared';

const kinds = new Map([
  ['function_definition', 'function'],
  ['class_definition', 'class'],
  ['decorated_definition', 'decorated'],
]);
const nestedKinds = new Map([
  ['function_definition', 'method'],
  ['decorated_definition', 'decorated'],
]);
const containers = new Set(['class_definition']);

export const pythonAdapter: LanguageAdapter = {
  extensions: ['.py'],
  grammarName: 'python',
  symbols: (node, source) => declarationSymbols(node, source, kinds, nestedKinds, containers),
};