# Author: Liam Grinstead
# SVG tree with tier and overlay labels

from lineage_tracker import get_ancestors, get_descendants

def _row_positions(n: int, width: int, y: int):
    if n <= 0: return []
    step = width // (n + 1)
    return [(step * (i + 1), y) for i in range(n)]

def render_lineage_tree(agent_id: str, overlay_color_hue: int = 260, max_depth: int = 6) -> str:
    width, height = 980, 640
    svg = [f"<svg width='{width}' height='{height}' xmlns='http://www.w3.org/2000/svg'>"]
    svg.append("<style>text{font-family:monospace}</style>")
    svg.append(f"<text x='12' y='24' font-size='16'>RFT Lineage: {agent_id}</text>")

    root_x, root_y = width//2, 170

    ancestors = get_ancestors(agent_id)
    if ancestors:
        pos = _row_positions(len(ancestors), width, 80)
        for (x,y), aid in zip(pos, ancestors):
            svg.append(f"<circle cx='{x}' cy='{y}' r='14' fill='hsl({overlay_color_hue},60%,50%)'/>")
            svg.append(f"<text x='{x-28}' y='{y+32}' font-size='11'>{aid}</text>")
        x_last, y_last = pos[-1]
        svg.append(f"<line x1='{x_last}' y1='{y_last+14}' x2='{root_x}' y2='{root_y-20}' stroke='#555'/>")
    else:
        svg.append("<text x='12' y='80' font-size='12' fill='#888'>No ancestors</text>")

    svg.append(f"<circle cx='{root_x}' cy='{root_y}' r='18' fill='#00b894'>"
               "<animate attributeName='r' values='18;22;18' dur='2s' repeatCount='indefinite'/></circle>")
    svg.append(f"<text x='{root_x-40}' y='{root_y+40}' font-size='12'>{agent_id}</text>")

    levels = get_descendants(agent_id, depth=max_depth)
    y = 260
    gap = 90
    for d in range(1, max_depth + 1):
        nodes = levels.get(d, [])
        if not nodes: break
        pos = _row_positions(len(nodes), width, y)
        svg.append(f"<text x='12' y='{y-26}' font-size='12' fill='#888'>Gen {d}</text>")
        for (x, yy), item in zip(pos, nodes):
            hue = 40 + (15 * d)
            svg.append(f"<circle cx='{x}' cy='{yy}' r='14' fill='hsl({hue},70%,55%)'>"
                       "<animate attributeName='fill' values='hsl({h},70%,55%);hsl({h},70%,65%);hsl({h},70%,55%)' dur='2s' repeatCount='indefinite'/>"
                       .replace("{h}", str(hue)) + "</circle>")
            label = f"{item['child_id']} · {item.get('tier','?')} · {item.get('overlay','?')}"
            svg.append(f"<text x='{x-60}' y='{yy+32}' font-size='11'>{label}</text>")
            svg.append(f"<path d='M {root_x} {root_y+20} Q {(root_x+x)//2} {(root_y+yy)//2} {x} {yy-14}' stroke='rgba(0,0,0,.35)' fill='none'/>")
        y += gap

    svg.append("</svg>")
    return "".join(svg)