import { Circle2d, createShapeId, TLHighlightShape } from '@tldraw/editor'
import { TestEditor } from '../../../test/TestEditor'
import { createDrawSegments, pointsToBase64 } from '../../utils/test-helpers'

let editor: TestEditor

beforeEach(() => {
	editor = new TestEditor()
})

afterEach(() => {
	editor?.dispose()
})

describe('HighlightShapeUtil dot detection', () => {
	const shapeId = createShapeId('test-highlight')

	function createHighlightShape(segments: TLHighlightShape['props']['segments']): TLHighlightShape {
		editor.createShapes([
			{
				id: shapeId,
				type: 'highlight',
				props: { segments },
			},
		])
		return editor.getShape(shapeId) as TLHighlightShape
	}

	describe('empty segments', () => {
		it('treats a shape created from its default props as a dot', () => {
			editor.createShape({ id: shapeId, type: 'highlight' })
			const shape = editor.getShape<TLHighlightShape>(shapeId)!
			expect(shape.props.segments).toEqual([])
			expect(editor.getShapeGeometry(shape)).toBeInstanceOf(Circle2d)
			expect(editor.getShapeAtPoint({ x: 0, y: 0 })?.id).toBe(shapeId)
		})
	})

	describe('getIsDot behavior via hideResizeHandles', () => {
		it('treats a shape with one segment and zero points as a dot', () => {
			const shape = createHighlightShape([{ type: 'free', path: '' }])
			const util = editor.getShapeUtil('highlight')
			expect(util.hideResizeHandles(shape)).toBe(true)
		})

		it('treats a shape with one segment and one point as a dot', () => {
			const shape = createHighlightShape(createDrawSegments([[{ x: 0, y: 0, z: 0.5 }]]))
			const util = editor.getShapeUtil('highlight')
			expect(util.hideResizeHandles(shape)).toBe(true)
		})

		it('treats a shape with one segment and two points as NOT a dot', () => {
			const shape = createHighlightShape(
				createDrawSegments([
					[
						{ x: 0, y: 0, z: 0.5 },
						{ x: 10, y: 10, z: 0.5 },
					],
				])
			)
			const util = editor.getShapeUtil('highlight')
			expect(util.hideResizeHandles(shape)).toBe(false)
		})

		it('treats a shape with one segment and many points as NOT a dot', () => {
			const shape = createHighlightShape(
				createDrawSegments([
					[
						{ x: 0, y: 0, z: 0.5 },
						{ x: 10, y: 10, z: 0.5 },
						{ x: 20, y: 5, z: 0.5 },
						{ x: 30, y: 15, z: 0.5 },
					],
				])
			)
			const util = editor.getShapeUtil('highlight')
			expect(util.hideResizeHandles(shape)).toBe(false)
		})

		it('treats a shape with multiple segments as NOT a dot', () => {
			const shape = createHighlightShape(
				createDrawSegments([[{ x: 0, y: 0, z: 0.5 }], [{ x: 10, y: 10, z: 0.5 }]])
			)
			const util = editor.getShapeUtil('highlight')
			expect(util.hideResizeHandles(shape)).toBe(false)
		})
	})

	describe('hideRotateHandle mirrors hideResizeHandles for dots', () => {
		it('hides rotate handle for dots', () => {
			const shape = createHighlightShape(createDrawSegments([[{ x: 0, y: 0, z: 0.5 }]]))
			const util = editor.getShapeUtil('highlight')
			expect(util.hideRotateHandle(shape)).toBe(true)
		})

		it('shows rotate handle for non-dots', () => {
			const shape = createHighlightShape(
				createDrawSegments([
					[
						{ x: 0, y: 0, z: 0.5 },
						{ x: 10, y: 10, z: 0.5 },
					],
				])
			)
			const util = editor.getShapeUtil('highlight')
			expect(util.hideRotateHandle(shape)).toBe(false)
		})
	})

	describe('hideSelectionBoundsFg mirrors hideResizeHandles for dots', () => {
		it('hides selection bounds for dots', () => {
			const shape = createHighlightShape(createDrawSegments([[{ x: 0, y: 0, z: 0.5 }]]))
			const util = editor.getShapeUtil('highlight')
			expect(util.hideSelectionBoundsFg(shape)).toBe(true)
		})

		it('shows selection bounds for non-dots', () => {
			const shape = createHighlightShape(
				createDrawSegments([
					[
						{ x: 0, y: 0, z: 0.5 },
						{ x: 10, y: 10, z: 0.5 },
					],
				])
			)
			const util = editor.getShapeUtil('highlight')
			expect(util.hideSelectionBoundsFg(shape)).toBe(false)
		})
	})

	describe('base64 encoding boundary conditions', () => {
		it('correctly handles the boundary at exactly 24 base64 characters (2 points)', () => {
			// First point is 16 base64 chars (Float32), delta points are 8 chars each (Float16)
			// 2 points = 24 characters, which should NOT be a dot
			const twoPointsBase64 = pointsToBase64([
				{ x: 0, y: 0, z: 0.5 },
				{ x: 1, y: 1, z: 0.5 },
			])
			expect(twoPointsBase64.length).toBe(24)

			const shape = createHighlightShape([{ type: 'free', path: twoPointsBase64 }])
			const util = editor.getShapeUtil('highlight')
			expect(util.hideResizeHandles(shape)).toBe(false)
		})

		it('correctly handles the boundary at exactly 16 base64 characters (1 point)', () => {
			// 1 point = 16 characters (Float32 first point), which should be a dot
			const onePointBase64 = pointsToBase64([{ x: 0, y: 0, z: 0.5 }])
			expect(onePointBase64.length).toBe(16)

			const shape = createHighlightShape([{ type: 'free', path: onePointBase64 }])
			const util = editor.getShapeUtil('highlight')
			expect(util.hideResizeHandles(shape)).toBe(true)
		})
	})
})

describe('HighlightShapeUtil getInterpolatedProps', () => {
	const segments = createDrawSegments([
		[
			{ x: 0, y: 0, z: 0.5 },
			{ x: 10, y: 10, z: 0.5 },
		],
	])

	function createHighlightShape(id: string, props: Partial<TLHighlightShape['props']>) {
		const shapeId = createShapeId(id)
		editor.createShapes([{ id: shapeId, type: 'highlight', props: { segments, ...props } }])
		return editor.getShape(shapeId) as TLHighlightShape
	}

	it('takes discrete props from the nearer shape', () => {
		const start = createHighlightShape('start', { color: 'red', size: 's', scale: 1 })
		const end = createHighlightShape('end', { color: 'blue', size: 'xl', scale: 3 })
		const util = editor.getShapeUtil('highlight')

		expect(util.getInterpolatedProps!(start, end, 0.25)).toMatchObject({
			color: 'red',
			size: 's',
			scale: 1.5,
		})
		expect(util.getInterpolatedProps!(start, end, 0.75)).toMatchObject({
			color: 'blue',
			size: 'xl',
			scale: 2.5,
		})
	})

	it('interpolates from a shape created with default props, which has no segments', () => {
		// https://github.com/tldraw/tldraw/issues/10737
		const emptyId = createShapeId('empty')
		editor.createShapes([{ id: emptyId, type: 'highlight' }])
		const empty = editor.getShape<TLHighlightShape>(emptyId)!
		const end = createHighlightShape('end', {})
		expect(empty.props.segments).toEqual([])
		const util = editor.getShapeUtil('highlight')

		expect(() => util.getInterpolatedProps!(empty, end, 0.5)).not.toThrow()
		expect(util.getInterpolatedProps!(empty, end, 1).segments).toEqual(end.props.segments)
		expect(() => util.getInterpolatedProps!(end, empty, 0.5)).not.toThrow()
		expect(util.getInterpolatedProps!(end, empty, 0).segments).toEqual(end.props.segments)
	})
})