import { flip, runSync } from 'effect/Effect'
import { describe, expect, it } from 'vitest'
import { updatePragma } from './updatePragma.js'

describe(updatePragma.name, () => {
	it('should update the pragma of a solidity file', () => {
		const file = `
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;

contract Foo {}
`
		expect(runSync(updatePragma(file))).toMatchInlineSnapshot(`
      "
      // SPDX-License-Identifier: MIT
      pragma solidity >=0.8.0;

      contract Foo {}
      "
    `)
	})

	it('should error if no pragma is found', () => {
		const file = `
contract Foo {}
`
		expect(runSync(flip(updatePragma(file)))).toMatchInlineSnapshot(
			'[NoPragmaFoundError: No valid pragma statement found.]',
		)
	})

	const pragmaStyles = [
		'^0.8.0',
		'0.8.0',
		'>0.8.0',
		'>=0.8.0',
		'<0.8.0',
		'<=0.8.0',
		'~0.8.0',
		'>=0.8.0 <0.8.9',
		' >=0.8.0  <0.8.9 ',
		'^0.8.0 ',
	]

	it.each(pragmaStyles)("should update the pragma of a solidity file with style '%s'", (style) => {
		const file = `
    // SPDX-License-Identifier: MIT
    pragma solidity ${style};

    contract Foo {}
    `

		const result = runSync(updatePragma(file))
		expect(result).includes('pragma solidity >=0.8.0;')
	})

	it('should handle custom version override', () => {
		const file = `
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;

contract Foo {}
`
		const result = runSync(updatePragma(file, '0.8.20'))
		expect(result).includes('pragma solidity >=0.8.20;')
	})

	it('should handle custom version override with bounds pattern', () => {
		const file = `
// SPDX-License-Identifier: MIT
pragma solidity >=0.8.0 <0.9.0;

contract Foo {}
`
		const result = runSync(updatePragma(file, '0.8.20'))
		expect(result).includes('pragma solidity >=0.8.20;')
	})

	it('should handle multiple pragma directives', () => {
		const file = `
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;
pragma experimental ABIEncoderV2;

contract Foo {}
`
		const result = runSync(updatePragma(file, '0.8.20'))
		expect(result).includes('pragma solidity >=0.8.20;')
		expect(result).includes('pragma experimental ABIEncoderV2;')
	})

	it('should handle comments before pragma', () => {
		const file = `
// SPDX-License-Identifier: MIT
// This is a comment
pragma solidity ^0.8.0;

contract Foo {}
`
		const result = runSync(updatePragma(file, '0.8.20'))
		expect(result).includes('// This is a comment')
		expect(result).includes('pragma solidity >=0.8.20;')
	})

	it('should preserve style when updating pragma', () => {
		const styles = {
			'^': 'pragma solidity ^0.8.0;',
			'': 'pragma solidity 0.8.0;',
			'>': 'pragma solidity >0.8.0;',
			'>=': 'pragma solidity >=0.8.0;',
			'<': 'pragma solidity <0.8.0;',
			'<=': 'pragma solidity <=0.8.0;',
			'~': 'pragma solidity ~0.8.0;',
		}

		for (const [_, pragma] of Object.entries(styles)) {
			const file = `
// SPDX-License-Identifier: MIT
${pragma}

contract Foo {}
`
			const customVersion = '0.8.20'
			const result = runSync(updatePragma(file, customVersion))

			// The actual implementation doesn't preserve style but converts to >=
			// This test might need to change if implementation changes
			expect(result).includes('pragma solidity >=0.8.20;')
		}
	})
})