path: root/internal/game/render_map.go

package game

import (
	"strings"
	"unicode/utf8"

	"thehouseoficarus/internal/color"
	"thehouseoficarus/internal/net"
	"thehouseoficarus/internal/world"
)

type mapGlyphs struct {
	topLeft, topRight       rune
	bottomLeft, bottomRight rune
	side                    rune
	topFill                 rune
	connectorH, connectorV  rune
	upArrow, downArrow      rune
	leftArrow, rightArrow   rune
}

func mapGlyphsForPlayer(unicode bool) mapGlyphs {
	if unicode {
		return mapGlyphs{
			topLeft: '╔', topRight: '╗', bottomLeft: '╚', bottomRight: '╝',
			side: '║', topFill: '═', connectorH: '-', connectorV: '│',
			upArrow: '↑', downArrow: '↓', leftArrow: '←', rightArrow: '→',
		}
	}
	return mapGlyphs{
		topLeft: '.', topRight: '.', bottomLeft: ':', bottomRight: ':',
		side: ':', topFill: '.', connectorH: '-', connectorV: '|',
		upArrow: '^', downArrow: 'v', leftArrow: '<', rightArrow: '>',
	}
}

type mapCell struct {
	char rune
	spec color.ColorSpec
}

type mapGraph struct {
	posToRoom map[[2]int]int
	roomToPos map[int][2]int
}

var bfsDirs = []struct {
	dir    world.ExitDir
	dx, dy int
}{
	{world.North, 0, -1},
	{world.South, 0, 1},
	{world.East, 1, 0},
	{world.West, -1, 0},
}

func buildGraph(g *Game, startRoomID int, visited map[int]bool) *mapGraph {
	mg := &mapGraph{
		posToRoom: make(map[[2]int]int),
		roomToPos: make(map[int][2]int),
	}

	type node struct {
		roomID int
		x, y   int
	}
	queue := []node{{startRoomID, 0, 0}}
	mg.posToRoom[[2]int{0, 0}] = startRoomID
	mg.roomToPos[startRoomID] = [2]int{0, 0}

	for len(queue) > 0 {
		n := queue[0]
		queue = queue[1:]

		room, ok := loadRoom(g, n.roomID)
		if !ok {
			continue
		}

		if visited != nil && !visited[n.roomID] {
			continue
		}

		for _, d := range bfsDirs {
			targetID, ok := exitTarget(room, d.dir)
			if !ok {
				continue
			}
			if _, seen := mg.roomToPos[targetID]; seen {
				continue
			}
			nx, ny := n.x+d.dx, n.y+d.dy
			mg.posToRoom[[2]int{nx, ny}] = targetID
			mg.roomToPos[targetID] = [2]int{nx, ny}
			queue = append(queue, node{targetID, nx, ny})
		}
	}

	return mg
}

func renderMapCells(grid [][]mapCell, colorMode string, startRow, endRow int, border rune) []string {
	lines := make([]string, 0, endRow-startRow)
	for row := startRow; row < endRow; row++ {
		var sb strings.Builder
		if border != 0 {
			sb.WriteRune(border)
		}
		for _, cell := range grid[row] {
			if cell.char == ' ' {
				sb.WriteRune(' ')
			} else if !cell.spec.Empty() {
				sb.WriteString(color.Render(colorMode, cell.spec, string(cell.char)))
			} else {
				sb.WriteRune(cell.char)
			}
		}
		if border != 0 {
			sb.WriteRune(border)
		}
		lines = append(lines, sb.String())
	}
	return lines
}

func buildTinyMap(g *Game, sess *net.Session, roomID int, mg mapGlyphs) []string {
	visited := roomsVisited(sess)
	bg := buildGraph(g, roomID, visited)

	colorMode := colorModeFor(sess)
	atSpec := resolveMapAt(g, sess)
	dimSpec := resolveDim(g, sess)
	ctx := &mapRenderCtx{
		g: g, sess: sess, bg: bg, visited: visited, currentRoom: roomID,
		atSpec: atSpec, dimSpec: dimSpec, blockedSpec: resolveMapBlocked(g, sess), mg: mg,
	}

	grid := make([][]mapCell, 5)
	for i := range grid {
		grid[i] = make([]mapCell, 5)
		for j := range grid[i] {
			grid[i][j] = mapCell{char: ' '}
		}
	}

	for y := -1; y <= 1; y++ {
		for x := -1; x <= 1; x++ {
			pos := [2]int{x, y}
			rid, ok := bg.posToRoom[pos]
			if !ok {
				continue
			}
			gr := (y + 1) * 2
			gc := (x + 1) * 2
			if rid == roomID {
				grid[gr][gc] = mapCell{char: '@', spec: atSpec}
			} else {
				unvisited := visited != nil && !visited[rid]
				ch, spec := roomMapSymbol(g, sess, rid, unvisited)
				if unvisited {
					spec = dimSpec
				}
				grid[gr][gc] = mapCell{char: ch, spec: spec}
			}
		}
	}

	for y := -1; y <= 1; y++ {
		for x := -1; x <= 0; x++ {
			leftPos := [2]int{x, y}
			rightPos := [2]int{x + 1, y}
			leftRoom, leftOK := bg.posToRoom[leftPos]
			rightRoom, rightOK := bg.posToRoom[rightPos]
			if !leftOK || !rightOK {
				continue
			}
			if cell, ok := ctx.connectorCell(leftRoom, rightRoom, world.East, world.West); ok {
				grid[(y+1)*2][(x+1)*2+1] = cell
			}
		}
	}

	for y := -1; y <= 0; y++ {
		for x := -1; x <= 1; x++ {
			topPos := [2]int{x, y}
			bottomPos := [2]int{x, y + 1}
			topRoom, topOK := bg.posToRoom[topPos]
			bottomRoom, bottomOK := bg.posToRoom[bottomPos]
			if !topOK || !bottomOK {
				continue
			}
			if cell, ok := ctx.connectorCell(topRoom, bottomRoom, world.South, world.North); ok {
				grid[(y+1)*2+1][(x+1)*2] = cell
			}
		}
	}

	cur, _ := loadRoom(g, roomID)
	if cur != nil {
		if _, ok := exitTarget(cur, world.Up); ok {
			grid[1][3] = mapCell{char: mg.upArrow, spec: color.NoColor()}
		}
		if _, ok := exitTarget(cur, world.Down); ok {
			grid[3][1] = mapCell{char: mg.downArrow, spec: color.NoColor()}
		}
	}

	topFill := strings.Repeat(string(mg.topFill), 5)
	lines := make([]string, 7)
	lines[0] = string(mg.topLeft) + topFill + string(mg.topRight)
	inner := renderMapCells(grid, colorMode, 0, 5, mg.side)
	copy(lines[1:], inner)
	botFill := strings.Repeat(string(mg.topFill), 5)
	lines[6] = string(mg.bottomLeft) + botFill + string(mg.bottomRight)

	return lines
}

func buildFullMap(g *Game, sess *net.Session, roomID, mapWidth, mapHeight int, mg mapGlyphs) []string {
	visited := roomsVisited(sess)
	bg := buildGraph(g, roomID, visited)

	colorMode := colorModeFor(sess)
	atSpec := resolveMapAt(g, sess)
	dimSpec := resolveDim(g, sess)
	ctx := &mapRenderCtx{
		g: g, sess: sess, bg: bg, visited: visited, currentRoom: roomID,
		atSpec: atSpec, dimSpec: dimSpec, blockedSpec: resolveMapBlocked(g, sess), mg: mg,
	}

	grid := make([][]mapCell, mapHeight)
	for i := range grid {
		grid[i] = make([]mapCell, mapWidth)
		for j := range grid[i] {
			grid[i][j] = mapCell{char: ' '}
		}
	}

	cx := mapWidth / 2
	cy := mapHeight / 2

	for pos, rid := range bg.posToRoom {
		gr := cy + pos[1]*2
		gc := cx + pos[0]*2
		if gr < 0 || gr >= mapHeight || gc < 0 || gc >= mapWidth {
			continue
		}
		if rid == roomID {
			grid[gr][gc] = mapCell{char: '@', spec: atSpec}
		} else {
			unvisited := visited != nil && !visited[rid]
			ch, spec := roomMapSymbol(g, sess, rid, unvisited)
			if unvisited {
				spec = dimSpec
			}
			grid[gr][gc] = mapCell{char: ch, spec: spec}
		}
	}

	for pos, rid := range bg.posToRoom {
		x, y := pos[0], pos[1]

		if rightID, exists := bg.posToRoom[[2]int{x + 1, y}]; exists {
			gr := cy + y*2
			gc := cx + x*2 + 1
			if gr >= 0 && gr < mapHeight && gc >= 0 && gc < mapWidth {
				if cell, ok := ctx.connectorCell(rid, rightID, world.East, world.West); ok {
					grid[gr][gc] = cell
				}
			}
		}

		if bottomID, exists := bg.posToRoom[[2]int{x, y + 1}]; exists {
			gr := cy + y*2 + 1
			gc := cx + x*2
			if gr >= 0 && gr < mapHeight && gc >= 0 && gc < mapWidth {
				if cell, ok := ctx.connectorCell(rid, bottomID, world.South, world.North); ok {
					grid[gr][gc] = cell
				}
			}
		}
	}

	return renderMapCells(grid, colorMode, 0, mapHeight, 0)
}

func roomsVisited(sess *net.Session) map[int]bool {
	if sess != nil && sess.Player != nil {
		return sess.Player.Stats.RoomsVisited
	}
	return nil
}

func colorModeFor(sess *net.Session) string {
	if sess != nil && sess.Player != nil {
		return sess.Player.OptionString("color")
	}
	return "none"
}

func resolveMapAt(g *Game, sess *net.Session) color.ColorSpec {
	if sess != nil {
		return g.resolveColor(sess, "map_at")
	}
	return color.Parse("0F")
}

func resolveDim(g *Game, sess *net.Session) color.ColorSpec {
	if sess != nil {
		return g.resolveColor(sess, "dim")
	}
	return color.Parse("F3 dim")
}

func resolveMapBlocked(g *Game, sess *net.Session) color.ColorSpec {
	if sess != nil {
		return g.resolveColor(sess, "map_blocked")
	}
	return color.Parse("C4")
}

// mapRenderCtx bundles the per-render state shared by node and connector drawing
// so the tiny and full maps build cells the same way.
type mapRenderCtx struct {
	g           *Game
	sess        *net.Session
	bg          *mapGraph
	visited     map[int]bool
	currentRoom int
	atSpec      color.ColorSpec
	dimSpec     color.ColorSpec
	blockedSpec color.ColorSpec
	mg          mapGlyphs
}

// nodeSpec returns the effective color a room's node is drawn with, mirroring
// the logic used when placing room glyphs.
func (c *mapRenderCtx) nodeSpec(roomID int) color.ColorSpec {
	if roomID == c.currentRoom {
		return c.atSpec
	}
	if c.visited != nil && !c.visited[roomID] {
		return c.dimSpec
	}
	_, spec := roomMapSymbol(c.g, c.sess, roomID, false)
	return spec
}

// connectorCell builds the link cell between two grid-adjacent rooms based on
// the per-direction traversability of the two exits joining them. ok is false
// when there is no link at all, so the caller draws nothing:
//   - both directions open       -> bidirectional bar (- / |)
//   - exactly one open           -> arrow pointing along the open direction
//   - >=1 exists but none open   -> blocked 'X'
//   - neither exit exists        -> ok == false (no cell)
//
// Bars and arrows use the normal link coloring (dim if an endpoint is unvisited,
// otherwise the gradient average); only 'X' uses the blocked color.
func (c *mapRenderCtx) connectorCell(roomA, roomB int, dirAB, dirBA world.ExitDir) (mapCell, bool) {
	fwd := exitStateTo(c.g, c.sess, roomA, dirAB, roomB) // A -> B
	bwd := exitStateTo(c.g, c.sess, roomB, dirBA, roomA) // B -> A

	if fwd == exitAbsent && bwd == exitAbsent {
		return mapCell{}, false
	}

	var glyph rune
	switch {
	case fwd == exitOpen && bwd == exitOpen:
		glyph = barGlyph(c.mg, dirAB)
	case fwd == exitOpen:
		glyph = arrowGlyph(c.mg, dirAB)
	case bwd == exitOpen:
		glyph = arrowGlyph(c.mg, dirBA)
	default:
		// At least one exit exists but none are currently traversable.
		return mapCell{char: 'X', spec: c.blockedSpec}, true
	}

	if c.visited != nil && (!c.visited[roomA] || !c.visited[roomB]) {
		return mapCell{char: glyph, spec: c.dimSpec}, true
	}
	return mapCell{char: glyph, spec: color.Average(c.nodeSpec(roomA), c.nodeSpec(roomB))}, true
}

type exitState int

const (
	exitAbsent exitState = iota
	exitOpen
	exitBlocked
)

// exitStateTo reports whether the exit from `from` in `dir` leads to `neighbor`
// and, if so, whether it is currently traversable for this player. A missing
// session/player (e.g. in tests or background renders) treats conditional exits
// as open so rendering never depends on player evaluation.
func exitStateTo(g *Game, sess *net.Session, from int, dir world.ExitDir, neighbor int) exitState {
	room, ok := loadRoom(g, from)
	if !ok {
		return exitAbsent
	}
	exit, ok := room.Exits[dir]
	if !ok || exit.Room != neighbor {
		return exitAbsent
	}
	if exit.Condition == nil || sess == nil || sess.Player == nil {
		return exitOpen
	}
	if sess.Player.GodMode || g.checkCondition(sess, exit.Condition) {
		return exitOpen
	}
	return exitBlocked
}

// barGlyph returns the bidirectional connector glyph for a link's orientation.
func barGlyph(mg mapGlyphs, dir world.ExitDir) rune {
	if dir == world.East || dir == world.West {
		return mg.connectorH
	}
	return mg.connectorV
}

// arrowGlyph returns the one-way arrow pointing along the direction of travel.
func arrowGlyph(mg mapGlyphs, dir world.ExitDir) rune {
	switch dir {
	case world.East:
		return mg.rightArrow
	case world.West:
		return mg.leftArrow
	case world.South:
		return mg.downArrow
	case world.North:
		return mg.upArrow
	}
	return mg.connectorH
}

// roomDefaultColorSpec resolves the room's own default map color (feature 3).
// Empty/unset returns NoColor.
func roomDefaultColorSpec(g *Game, roomID int) color.ColorSpec {
	if room, ok := loadRoom(g, roomID); ok && room.Color != "" {
		return color.Parse(room.Color)
	}
	return color.NoColor()
}

func roomMapSymbol(g *Game, sess *net.Session, roomID int, unvisited bool) (rune, color.ColorSpec) {
	roomSpec := roomDefaultColorSpec(g, roomID)
	if sess != nil && sess.Player != nil {
		if data, ok := sess.Player.MapSymbols[roomID]; ok {
			r, size := utf8.DecodeRuneInString(data.Char)
			if size > 0 && r != utf8.RuneError {
				// precedence: player symbol color > room default color > none
				spec := roomSpec
				if data.Color != "" {
					spec = color.Parse(data.Color)
				}
				// non-ASCII custom symbols fall back to 'o'
				// when unicode mode is off, but preserve the color.
				if !sess.Player.OptionBool("unicode") && r > 127 {
					return 'o', spec
				}
				return r, spec
			}
		}
		if sess.Player.OptionBool("unicode") {
			if unvisited {
				return '□', roomSpec
			}
			return '■', roomSpec
		}
		return 'o', roomSpec
	}
	return '■', roomSpec
}

func exitTarget(room *world.Room, dir world.ExitDir) (int, bool) {
	if room == nil {
		return 0, false
	}
	exit, ok := room.Exits[dir]
	if !ok {
		return 0, false
	}
	return exit.Room, true
}

func loadRoom(g *Game, roomID int) (*world.Room, bool) {
	if roomID == 0 {
		return nil, false
	}
	room, err := g.World.LoadRoom(roomID)
	if err != nil {
		return nil, false
	}
	return room, true
}

func stripBlankRows(lines []string) []string {
	var out []string
	for _, line := range lines {
		if strings.TrimSpace(line) != "" {
			out = append(out, line)
		}
	}
	return out
}

func leftTrimCommon(lines []string) []string {
	min := -1
	for _, line := range lines {
		if strings.TrimSpace(line) == "" {
			continue
		}
		n := 0
		for _, r := range line {
			if r == ' ' {
				n++
			} else {
				break
			}
		}
		if min < 0 || n < min {
			min = n
		}
	}
	if min <= 0 {
		return lines
	}
	result := make([]string, len(lines))
	for i, line := range lines {
		if len(line) <= min {
			result[i] = ""
		} else {
			result[i] = line[min:]
		}
	}
	return result
}