import { fireEvent, render, screen } from '@testing-library/react'
import { beforeEach, describe, expect, it, vi } from 'vitest'
import DocumentSettings from '../document-settings'

const mockPush = vi.fn()
const mockBack = vi.fn()
const mockOpenIntegrationsSetting = vi.fn()
vi.mock('@/next/navigation', () => ({
  useRouter: () => ({
    push: mockPush,
    back: mockBack,
  }),
}))

// Mock use-context-selector
vi.mock('use-context-selector', async (importOriginal) => {
  const actual = (await importOriginal()) as Record<string, unknown>
  return {
    ...actual,
    useContext: () => ({
      indexingTechnique: 'qualified',
      dataset: { id: 'dataset-1' },
    }),
  }
})

const mockInvalidDocumentList = vi.fn()
const mockInvalidDocumentDetail = vi.fn()
let mockDocumentDetail: Record<string, unknown> | null = {
  name: 'test-document',
  data_source_type: 'upload_file',
  data_source_info: {
    upload_file: { id: 'file-1', name: 'test.pdf' },
  },
}
let mockError: Error | null = null

vi.mock('@/service/knowledge/use-document', () => ({
  useDocumentDetail: () => ({
    data: mockDocumentDetail,
    error: mockError,
  }),
  useInvalidDocumentList: () => mockInvalidDocumentList,
  useInvalidDocumentDetail: () => mockInvalidDocumentDetail,
}))

// Mock useDefaultModel
vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({
  useDefaultModel: () => ({
    data: { model: 'text-embedding-ada-002' },
  }),
}))

vi.mock('@/app/components/base/app-unavailable', () => ({
  default: ({ code, unknownReason }: { code?: number; unknownReason?: string }) => (
    <div data-testid="app-unavailable">
      <span data-testid="error-code">{code}</span>
      <span data-testid="error-reason">{unknownReason}</span>
    </div>
  ),
}))

vi.mock('@/app/components/base/loading', () => ({
  default: ({ type }: { type?: string }) => (
    <div data-testid="loading" data-type={type}>
      Loading...
    </div>
  ),
}))

vi.mock('@/app/components/datasets/create/step-two', () => ({
  default: ({
    isAPIKeySet,
    onSetting,
    datasetId,
    dataSourceType,
    files,
    onSave,
    onCancel,
    isSetting,
  }: {
    isAPIKeySet?: boolean
    onSetting?: () => void
    datasetId?: string
    dataSourceType?: string
    files?: unknown[]
    onSave?: () => void
    onCancel?: () => void
    isSetting?: boolean
  }) => (
    <div data-testid="step-two">
      <span data-testid="api-key-set">{isAPIKeySet ? 'true' : 'false'}</span>
      <span data-testid="dataset-id">{datasetId}</span>
      <span data-testid="data-source-type">{dataSourceType}</span>
      <span data-testid="is-setting">{isSetting ? 'true' : 'false'}</span>
      <span data-testid="files-count">{files?.length || 0}</span>
      <button onClick={onSetting} data-testid="setting-btn">
        Setting
      </button>
      <button onClick={onSave} data-testid="save-btn">
        Save
      </button>
      <button onClick={onCancel} data-testid="cancel-btn">
        Cancel
      </button>
    </div>
  ),
}))

vi.mock('@/app/components/header/account-setting/use-integrations-setting', () => ({
  useIntegrationsSetting: () => mockOpenIntegrationsSetting,
}))

describe('DocumentSettings', () => {
  beforeEach(() => {
    vi.clearAllMocks()
    mockDocumentDetail = {
      name: 'test-document',
      data_source_type: 'upload_file',
      data_source_info: {
        upload_file: { id: 'file-1', name: 'test.pdf' },
      },
    }
    mockError = null
  })

  const defaultProps = {
    datasetId: 'dataset-1',
    documentId: 'document-1',
  }

  describe('Rendering', () => {
    it('should render StepTwo component when data is loaded', () => {
      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('step-two')).toBeInTheDocument()
    })

    it('should render loading when documentDetail is not available', () => {
      mockDocumentDetail = null

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('loading')).toBeInTheDocument()
    })

    it('should render AppUnavailable when error occurs', () => {
      mockError = new Error('Error loading document')

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('app-unavailable')).toBeInTheDocument()
      expect(screen.getByTestId('error-code')).toHaveTextContent('500')
    })
  })

  // Props passing
  describe('Props Passing', () => {
    it('should pass datasetId to StepTwo', () => {
      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('dataset-id')).toHaveTextContent('dataset-1')
    })

    it('should pass isSetting true to StepTwo', () => {
      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('is-setting')).toHaveTextContent('true')
    })

    it('should pass isAPIKeySet when embedding model is available', () => {
      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('api-key-set')).toHaveTextContent('true')
    })

    it('should pass data source type to StepTwo', () => {
      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('data-source-type')).toHaveTextContent('upload_file')
    })
  })

  describe('User Interactions', () => {
    it('should call router.back when cancel is clicked', () => {
      render(<DocumentSettings {...defaultProps} />)

      fireEvent.click(screen.getByTestId('cancel-btn'))

      expect(mockBack).toHaveBeenCalled()
    })

    it('should navigate to document page when save is clicked', () => {
      render(<DocumentSettings {...defaultProps} />)

      fireEvent.click(screen.getByTestId('save-btn'))

      expect(mockInvalidDocumentList).toHaveBeenCalled()
      expect(mockInvalidDocumentDetail).toHaveBeenCalled()
      expect(mockPush).toHaveBeenCalledWith('/datasets/dataset-1/documents/document-1')
    })

    it('should open model provider integrations settings when setting button is clicked', () => {
      render(<DocumentSettings {...defaultProps} />)

      fireEvent.click(screen.getByTestId('setting-btn'))

      expect(mockOpenIntegrationsSetting).toHaveBeenCalledWith({ payload: 'provider' })
    })
  })

  // Data source types
  describe('Data Source Types', () => {
    it('should handle upload_file_id data source format', () => {
      mockDocumentDetail = {
        name: 'test-document',
        data_source_type: 'upload_file',
        data_source_info: {
          upload_file_id: '4a807f05-45d6-4fc4-b7a8-b009a4568b36',
        },
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('files-count')).toHaveTextContent('1')
    })

    it('should handle legacy upload_file data source', () => {
      mockDocumentDetail = {
        name: 'test-document',
        data_source_type: 'upload_file',
        data_source_info: {
          upload_file: { id: 'file-1', name: 'test.pdf' },
        },
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('files-count')).toHaveTextContent('1')
    })

    it('should handle website crawl data source', () => {
      mockDocumentDetail = {
        name: 'test-website',
        data_source_type: 'website_crawl',
        data_source_info: {
          title: 'Test Page',
          source_url: 'https://example.com',
          content: 'Page content',
          description: 'Page description',
        },
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('data-source-type')).toHaveTextContent('website_crawl')
    })

    it('should handle local file data source', () => {
      mockDocumentDetail = {
        name: 'local-file',
        data_source_type: 'upload_file',
        data_source_info: {
          related_id: 'file-id',
          transfer_method: 'local',
          name: 'local-file.pdf',
          extension: 'pdf',
        },
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('files-count')).toHaveTextContent('1')
    })

    it('should handle online document (Notion) data source', () => {
      mockDocumentDetail = {
        name: 'notion-page',
        data_source_type: 'notion_import',
        data_source_info: {
          workspace_id: 'ws-1',
          credential_id: 'cred-1',
          page: {
            page_id: 'page-1',
            page_name: 'Test Page',
            page_icon: '📄',
            type: 'page',
          },
        },
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('data-source-type')).toHaveTextContent('notion_import')
    })
  })

  describe('Edge Cases', () => {
    it('should handle undefined data_source_info', () => {
      mockDocumentDetail = {
        name: 'test-document',
        data_source_type: 'upload_file',
        data_source_info: undefined,
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('files-count')).toHaveTextContent('0')
    })

    it('should handle empty data_source_info object', () => {
      mockDocumentDetail = {
        name: 'test-document',
        data_source_type: 'upload_file',
        data_source_info: {},
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('files-count')).toHaveTextContent('0')
    })

    it('should maintain structure when rerendered', () => {
      const { rerender } = render(<DocumentSettings datasetId="dataset-1" documentId="doc-1" />)

      rerender(<DocumentSettings datasetId="dataset-2" documentId="doc-2" />)

      expect(screen.getByTestId('step-two')).toBeInTheDocument()
    })
  })

  describe('Files Extraction Regression Tests', () => {
    it('should correctly extract file ID from upload_file_id format', () => {
      const fileId = '4a807f05-45d6-4fc4-b7a8-b009a4568b36'
      mockDocumentDetail = {
        name: 'test-document.pdf',
        data_source_type: 'upload_file',
        data_source_info: {
          upload_file_id: fileId,
        },
      }

      render(<DocumentSettings {...defaultProps} />)

      // Verify files array is populated with correct file ID
      expect(screen.getByTestId('files-count')).toHaveTextContent('1')
    })

    it('should preserve document name when using upload_file_id format', () => {
      const documentName = 'my-uploaded-document.txt'
      mockDocumentDetail = {
        name: documentName,
        data_source_type: 'upload_file',
        data_source_info: {
          upload_file_id: 'some-file-id',
        },
      }

      render(<DocumentSettings {...defaultProps} />)

      expect(screen.getByTestId('files-count')).toHaveTextContent('1')
    })
  })
})
