package apple2

import (
	"fmt"
)

type ioC0Page struct {
	softSwitches     [128]softSwitch
	softSwitchesData [128]uint8
	keyboard         keyboardProvider
	mmu              *memoryManager
}

type softSwitch func(io *ioC0Page, isWrite bool, value uint8) uint8

// TODO: change interface to func
type keyboardProvider interface {
	getKey() (key uint8, ok bool)
}

// See https://www.kreativekorp.com/miscpages/a2info/iomemory.shtml
// See https://stason.org/TULARC/pc/apple2/programmer/004-I-d-like-to-do-some-serious-Apple-II-programming-Whe.html

const (
	ssOn  uint8 = 0x80
	ssOff uint8 = 0x00
)

const (
	ioDataKeyboard uint8 = 0x10

	ioFlagGraphics     uint8 = 0x50
	ioFlagMixed        uint8 = 0x52
	ioFlagSecondPage   uint8 = 0x54
	ioFlagHiRes        uint8 = 0x56
	ioFlagAnnunciator0 uint8 = 0x58
	ioFlagAnnunciator1 uint8 = 0x5a
	ioFlagAnnunciator2 uint8 = 0x5c
	ioFlagAnnunciator3 uint8 = 0x5e

	ioDataCassette uint8 = 0x60
	ioFlagButton0  uint8 = 0x61
	ioFlagButton1  uint8 = 0x62
	ioFlagButton2  uint8 = 0x63
	ioDataPaddle0  uint8 = 0x64
	ioDataPaddle1  uint8 = 0x65
	ioDataPaddle2  uint8 = 0x66
	ioDataPaddle3  uint8 = 0x67
)

func (p *ioC0Page) isSoftSwitchExtActive(ioFlag uint8) bool {
	return (p.softSwitchesData[ioFlag] & ssOn) == ssOn
}

func newIoC0Page(mmu *memoryManager) *ioC0Page {
	var p ioC0Page
	p.mmu = mmu
	ss := &p.softSwitches

	ss[0x00] = getKeySoftSwitch         // Keyboard
	ss[0x10] = strobeKeyboardSoftSwitch // Keyboard Strobe
	ss[0x20] = notImplementedSoftSwitch // Cassette Output
	ss[0x30] = notImplementedSoftSwitch // Speaker
	ss[0x40] = notImplementedSoftSwitch // Game connector Strobe
	// Note: Some sources indicate that all these cover 16 positions
	// for read and write. But the Apple2e take over some of them, with
	// the prevention on acting only on writes.

	ss[0x50] = getSoftSwitch(ioFlagGraphics, false)
	ss[0x51] = getSoftSwitch(ioFlagGraphics, true)
	ss[0x52] = getSoftSwitch(ioFlagMixed, false)
	ss[0x53] = getSoftSwitch(ioFlagMixed, true)
	ss[0x54] = getSoftSwitch(ioFlagSecondPage, false)
	ss[0x55] = getSoftSwitch(ioFlagSecondPage, true)
	ss[0x56] = getSoftSwitch(ioFlagHiRes, false)
	ss[0x57] = getSoftSwitch(ioFlagHiRes, true)
	ss[0x58] = getSoftSwitch(ioFlagAnnunciator0, false)
	ss[0x59] = getSoftSwitch(ioFlagAnnunciator0, true)
	ss[0x5a] = getSoftSwitch(ioFlagAnnunciator1, false)
	ss[0x5b] = getSoftSwitch(ioFlagAnnunciator1, true)
	ss[0x5c] = getSoftSwitch(ioFlagAnnunciator2, false)
	ss[0x5d] = getSoftSwitch(ioFlagAnnunciator2, true)
	ss[0x5e] = getSoftSwitch(ioFlagAnnunciator3, false)
	ss[0x5f] = getSoftSwitch(ioFlagAnnunciator3, true)

	ss[0x60] = notImplementedSoftSwitch // Cassetter Input
	ss[0x61] = getStatusSoftSwitch(ioFlagButton0)
	ss[0x62] = getStatusSoftSwitch(ioFlagButton1)
	ss[0x63] = getStatusSoftSwitch(ioFlagButton2)
	ss[0x64] = getStatusSoftSwitch(ioDataPaddle0)
	ss[0x65] = getStatusSoftSwitch(ioDataPaddle1)
	ss[0x66] = getStatusSoftSwitch(ioDataPaddle2)
	ss[0x67] = getStatusSoftSwitch(ioDataPaddle3)
	ss[0x68] = ss[0x60]
	ss[0x69] = ss[0x61]
	ss[0x6A] = ss[0x62]
	ss[0x6B] = ss[0x63]
	ss[0x6C] = ss[0x64]
	ss[0x6D] = ss[0x65]
	ss[0x6E] = ss[0x66]
	ss[0x6F] = ss[0x67]
	ss[0x70] = notImplementedSoftSwitch // Game controllers reset

	return &p
}

func (p *ioC0Page) setKeyboardProvider(kb keyboardProvider) {
	p.keyboard = kb
}

func (p *ioC0Page) Peek(address uint8) uint8 {
	//fmt.Printf("Peek on $C0%02x ", address)
	return p.access(address, false, 0)
}

func (p *ioC0Page) Poke(address uint8, value uint8) {
	//fmt.Printf("Poke on $C0%02x with %02x ", address, value)
	p.access(address, true, value)
}

func (p *ioC0Page) access(address uint8, isWrite bool, value uint8) uint8 {
	// The second half of the pages is reserved for slots
	if address >= 0x90 {
		// TODO reserved slots data
		return 0
	}

	ss := p.softSwitches[address]
	if ss == nil {
		panic(fmt.Sprintf("Unknown softswitch 0xC0%02x", address))
	}

	return ss(p, isWrite, value)
}

func getStatusSoftSwitch(ioFlag uint8) softSwitch {
	return func(io *ioC0Page, isWrite bool, value uint8) uint8 {
		return io.softSwitchesData[ioFlag]
	}
}

func getSoftSwitch(ioFlag uint8, isSet bool) softSwitch {
	return func(io *ioC0Page, isWrite bool, value uint8) uint8 {
		if isSet {
			io.softSwitchesData[ioFlag] = ssOn
		} else {
			io.softSwitchesData[ioFlag] = ssOff
		}
		return 0
	}
}

func notImplementedSoftSwitch(*ioC0Page, bool, uint8) uint8 {
	return 0
}

func getKeySoftSwitch(p *ioC0Page, _ bool, _ uint8) uint8 {
	strobed := (p.softSwitchesData[ioDataKeyboard] & (1 << 7)) == 0
	if strobed && p.keyboard != nil {
		if key, ok := p.keyboard.getKey(); ok {
			p.softSwitchesData[ioDataKeyboard] = key + (1 << 7)
		}
	}
	return p.softSwitchesData[ioDataKeyboard]
}

func strobeKeyboardSoftSwitch(p *ioC0Page, _ bool, _ uint8) uint8 {
	result := p.softSwitchesData[ioDataKeyboard]
	p.softSwitchesData[ioDataKeyboard] &^= 1 << 7
	return result
}