import type { TSESTree } from '@typescript-eslint/types'
import { createRule } from '../rule.ts'
import { findLastIndex } from '../utils.ts'

type Location = 'at-top' | 'at-bottom'

export default createRule({
  name: 'prefer-default-export',
  meta: {
    type: 'suggestion',
    docs: {
      description: 'Enforce default export location at top or bottom',
      recommended: 'stylistic',
    },
    schema: [{ type: 'string', enum: ['at-top', 'at-bottom'] }],
    messages: {
      'at-top': 'Move default export to top',
      'at-bottom': 'Move default export to bottom',
    },
  },
  defaultOptions: ['at-bottom'] as [Location],
  create(context, [location]) {
    function onProgram(program: TSESTree.Program) {
      const index = program.body.findIndex(isDefaultExport)
      if (index === -1) return
      const node = program.body[index]
      if (location === 'at-top' && !isTop(program, index)) {
        context.report({ node, messageId: 'at-top' })
      } else if (location === 'at-bottom' && !isBottom(program, index)) {
        context.report({ node, messageId: 'at-bottom' })
      }
    }
    return { Program: onProgram }
  },
})

function isTop(program: TSESTree.Program, index: number) {
  const firstIndex = program.body.findIndex(isNonDefaultExport)
  if (firstIndex === -1) return true
  return index < firstIndex
}

function isBottom(program: TSESTree.Program, index: number) {
  const lastIndex = findLastIndex(program.body, isNonDefaultExport)
  if (lastIndex === -1) return true
  return index > lastIndex
}

function isDefaultExport(node: TSESTree.Node) {
  return node.type === 'ExportDefaultDeclaration'
}

function isNonDefaultExport(node: TSESTree.Node) {
  return node.type === 'ExportAllDeclaration' || node.type === 'ExportNamedDeclaration'
}