package com.htmlism.firepower.demo

import scala.util.chaining.*

import com.htmlism.firepower.core.Definable

object Easy6502:
  enum Color:
    case Black, White, Red, Cyan, Purple, Green, Blue, Yellow, Orange, Brown, LightRed, DarkGrey, Grey, LightGreen,
      LightBlue, LightGrey

  object Color:
    given Definable[Color] with
      def table(x: Color): Definable.Table =
        Color
          .values
          .iterator
          .map { c =>
            "COLOR_" + c.toString -> c.ordinal
          }
          .toList
          .pipe(Definable.Table("Colors that each screen pixel can be set to", _))

      extension (x: Color)
        def toComment: String =
          x.toString

      extension (x: Color)
        def toValue: Int =
          x.ordinal

      extension (x: Color)
        def toDefine: String =
          "COLOR_" + x.toString

      extension (x: Color)
        def toDefineWithMath: String =
          "COLOR_" + x.toString

  class Screen(baseAddr: Int):
    def apply(offset: Int): Screen.Pixel =
      Screen.Pixel(baseAddr, offset)

  object Screen:
    case class Pixel(baseAddr: Int, offset: Int)

    object Pixel:
      given Definable[Pixel] with
        def table(x: Pixel): Definable.Table =
          // define table needs to be a function of an instance
          Definable.Table("The screen as a collection of pixels", List("SCREEN" -> x.baseAddr))

        extension (x: Pixel)
          def toComment: String =
            s"Screen(${x.offset})"

        extension (x: Pixel)
          def toValue: Int =
            x.baseAddr + x.offset

        extension (x: Pixel)
          def toDefine: String =
            if x.offset == 0 then "SCREEN"
            else "TODO"

        extension (x: Pixel)
          def toDefineWithMath: String =
            "SCREEN+" + x.offset