import re, strutils, terminal, tables, sequtils

import ./[types]

proc validatePort*(portStr: string): bool = 
    try:
        let port: int = portStr.parseInt
        return port >= 1 and port <= 65535
    except ValueError:
        return false

# Table border characters
const topLeft = "╭"
const topMid  = "┬"
const topRight= "╮"
const midLeft = "├"
const midMid  = "┼"
const midRight= "┤"
const botLeft = "╰"
const botMid  = "┴"
const botRight= "╯"
const hor     = "─"
const vert    = "│"

# Format border
proc border(left, mid, right: string, widths: seq[int]): string =
    var line = left
    for i, w in widths:
        line.add(hor.repeat(w))
        line.add(if i < widths.len - 1: mid else: right)
    return line

# Format a row of data
proc row(cells: seq[string], widths: seq[int]): string =
    var row = vert
    for i, cell in cells:
        # Truncate content of a cell with "..." when the value to be inserted is longer than the designated width
        let w = widths[i] - 2
        let c = if cell.len > w:
            if w >= 3:
                cell[0 ..< w - 3] & "..."
            else:
                ".".repeat(max(0, w))
        else:
            cell
        row.add(" " & c.alignLeft(w) & " " & vert)
    return row

proc drawTable*(cq: Conquest, listeners: seq[Listener]) = 

    # Column headers and widths
    let headers = @["Name", "Address", "Port", "Protocol", "Agents"]
    let widths = @[10, 17, 7, 10, 8]

    cq.writeLine(border(topLeft, topMid, topRight, widths))
    cq.writeLine(row(headers, widths))
    cq.writeLine(border(midLeft, midMid, midRight, widths))

    for l in listeners:
        # Get number of agents connected to the listener
        let connectedAgents = cq.agents.values.countIt(it.listener == l.name)

        let row = @[l.name, l.address, $l.port, $l.protocol, $connectedAgents]
        cq.writeLine(row(row, widths)) 

    cq.writeLine(border(botLeft, botMid, botRight, widths)) 


proc drawTable*(cq: Conquest, agents: seq[Agent]) = 
    
    let headers: seq[string] = @["Name", "Address", "Username", "Hostname", "Operating System", "Process", "PID"]
    let widths = @[10, 17, 20, 20, 20, 15, 7]

    cq.writeLine(border(topLeft, topMid, topRight, widths))
    cq.writeLine(row(headers, widths))
    cq.writeLine(border(midLeft, midMid, midRight, widths))

    for a in agents:
        let row = @[a.name, a.ip, a.username, a.hostname, a.os, a.process, $a.pid] 
        
        # Highlight agents running within elevated processes
        if a.elevated: 
            cq.writeLine(bgRed, fgBlack, row(row, widths)) 
        else: 
            cq.writeLine(row(row, widths)) 


    cq.writeLine(border(botLeft, botMid, botRight, widths))