#!/usr/bin/env nu
use common.nu *

def run-test-instance [name: string, scenario_closure: closure] {
    let port = 3004
    let debug_port = $port + 1
    let url = $"http://localhost:($port)"
    let debug_url = $"http://localhost:($debug_port)"
    let db_path = (mktemp -d -t hydrant_gc_test.XXXXXX)

    print $"--- running scenario: ($name) ---"
    print $"database path: ($db_path)"

    let binary = build-hydrant
    let instance = start-hydrant $binary $db_path $port

    try {
        if not (wait-for-api $url) {
            error make {msg: "api failed to start"}
        }

        do $scenario_closure $url $debug_url
        
        print $"PASSED: ($name)\n"
    } catch { |e|
        print $"test failed: ($e.msg)"
        try { kill $instance.pid }
        exit 1
    }

    try { kill $instance.pid }
}

def wait-for-blocks [debug_url: string] {
    print "waiting for blocks to appear..."
    mut blocks = {}
    mut count = 0
    for i in 1..30 {
        $blocks = (http get $"($debug_url)/debug/iter?partition=blocks&limit=1000")
        $count = ($blocks.items | length)
        if $count > 0 {
            break
        }
        sleep 2sec
    }
    if $count == 0 {
        error make {msg: "FAILED: no blocks found after backfill"}
    }
    $count
}

def compact-and-check-blocks [debug_url: string, expected_count: int] {
    print "triggering major compaction on blocks partition..."
    http post -H [Content-Length 0] $"($debug_url)/debug/compact?partition=blocks" ""

    let blocks_after = http get $"($debug_url)/debug/iter?partition=blocks&limit=1000"
    let after_count = ($blocks_after.items | length)

    if $after_count != $expected_count {
        error make {msg: $"FAILED: expected ($expected_count) blocks after compaction, found ($after_count)"}
    }
}

def check-repo-refcounts [debug_url: string, did: string, expected_refcount: int] {
    let refs = (http get $"($debug_url)/debug/repo_refcounts?did=($did)")
    let cids = $refs.cids
    let count = ($cids | columns | length)
    if $count == 0 {
        if $expected_refcount != 0 {
            error make {msg: $"FAILED: expected refcounts to be ($expected_refcount) but found no cids for ($did)"}
        }
        print $"blocks for ($did) completely verified as 0"
        return
    }
    
    for rcid in ($cids | transpose key value) {
        if $rcid.value < $expected_refcount {
            error make {msg: $"FAILED: expected refcount for ($rcid.key) to be >= ($expected_refcount), but found ($rcid.value)"}
        }
        if $expected_refcount == 0 and $rcid.value != 0 {
            error make {msg: $"FAILED: expected refcount for ($rcid.key) to be 0, but found ($rcid.value)"}
        }
    }
    print $"all ($count) tracked blocks for ($did) verified with refcount >= ($expected_refcount)"
}

def ack-all-events [debug_url: string, url: string] {
    print "acking all events..."
    mut total_acked = 0
    mut items = []
    
    # wait for at least some events
    for i in 1..30 {
        let events = http get $"($debug_url)/debug/iter?partition=events&limit=1000"
        $items = $events.items
        if ($items | length) > 0 {
            break
        }
        sleep 2sec
    }
    
    if ($items | length) == 0 {
        error make {msg: "FAILED: no events to ack"}
    }

    loop {
        let event_ids = ($items | each { |x| ($x | first | into int) })
        http post -t application/json $"($url)/stream/ack" { ids: $event_ids }
        $total_acked += ($event_ids | length)
        
        # getting next batch
        let next_events = http get $"($debug_url)/debug/iter?partition=events&limit=1000"
        $items = $next_events.items
        if ($items | length) == 0 {
            break
        }
    }
    
    print $"acked ($total_acked) events"
}

def main [] {
    let repo1 = "did:web:guestbook.gaze.systems"
    let repo2 = "did:plc:dfl62fgb7wtjj3fcbb72naae"

    run-test-instance "delete repo only" { |url, debug_url|
        print $"adding repo ($repo1) to tracking..."
        http put -t application/json $"($url)/repos" [ { did: ($repo1) } ]
        
        let before_count = (wait-for-blocks $debug_url)
        print $"found ($before_count) blocks before GC"
        
        check-repo-refcounts $debug_url $repo1 2

        print "deleting repo..."
        http delete -t application/json $"($url)/repos" --data [ { did: ($repo1), delete_data: true } ]
        sleep 1sec

        check-repo-refcounts $debug_url $repo1 1

        compact-and-check-blocks $debug_url $before_count
    }

    run-test-instance "ack events only" { |url, debug_url|
        print $"adding repo ($repo1) to tracking..."
        http put -t application/json $"($url)/repos" [ { did: ($repo1) } ]
        
        let before_count = (wait-for-blocks $debug_url)
        print $"found ($before_count) blocks before GC"
        
        check-repo-refcounts $debug_url $repo1 2

        ack-all-events $debug_url $url
        sleep 1sec

        check-repo-refcounts $debug_url $repo1 1

        compact-and-check-blocks $debug_url $before_count
    }

    run-test-instance "delete repo, ack events" { |url, debug_url|
        print $"adding repo ($repo1) to tracking..."
        http put -t application/json $"($url)/repos" [ { did: ($repo1) } ]
        
        let before_count = (wait-for-blocks $debug_url)
        print $"found ($before_count) blocks before GC"
        
        check-repo-refcounts $debug_url $repo1 2

        print "deleting repo..."
        http delete -t application/json $"($url)/repos" --data [ { did: ($repo1), delete_data: true } ]

        ack-all-events $debug_url $url
        sleep 1sec
        
        check-repo-refcounts $debug_url $repo1 0

        compact-and-check-blocks $debug_url 0
    }

    run-test-instance "delete repo, compact, ack events, compact" { |url, debug_url|
        print $"adding repo ($repo1) to tracking..."
        http put -t application/json $"($url)/repos" [ { did: ($repo1) } ]
        
        let before_count = (wait-for-blocks $debug_url)
        print $"found ($before_count) blocks before GC"
        
        check-repo-refcounts $debug_url $repo1 2

        print "deleting repo..."
        http delete -t application/json $"($url)/repos" --data [ { did: ($repo1), delete_data: true } ]
        sleep 1sec
        
        check-repo-refcounts $debug_url $repo1 1

        compact-and-check-blocks $debug_url $before_count

        ack-all-events $debug_url $url
        sleep 1sec
        
        check-repo-refcounts $debug_url $repo1 0

        compact-and-check-blocks $debug_url 0
    }

    run-test-instance "ack events, compact, delete repo, compact" { |url, debug_url|
        print $"adding repo ($repo1) to tracking..."
        http put -t application/json $"($url)/repos" [ { did: ($repo1) } ]
        
        let before_count = (wait-for-blocks $debug_url)
        print $"found ($before_count) blocks before GC"
        
        check-repo-refcounts $debug_url $repo1 2

        ack-all-events $debug_url $url
        sleep 1sec
        
        check-repo-refcounts $debug_url $repo1 1

        compact-and-check-blocks $debug_url $before_count

        print "deleting repo..."
        http delete -t application/json $"($url)/repos" --data [ { did: ($repo1), delete_data: true } ]
        sleep 1sec
        
        check-repo-refcounts $debug_url $repo1 0

        compact-and-check-blocks $debug_url 0
    }

    run-test-instance "multiple repos" { |url, debug_url|
        print $"adding repo ($repo2) to tracking..."
        http put -t application/json $"($url)/repos" [ { did: ($repo2) } ]
        let repo2_blocks = (wait-for-blocks $debug_url)
        print $"found ($repo2_blocks) blocks for repo2"

        print $"adding repo ($repo1) to tracking..."
        http put -t application/json $"($url)/repos" [ { did: ($repo1) } ]
        
        # wait a bit more for repo1 blocks to finish
        sleep 5sec
        let total_blocks = (http get $"($debug_url)/debug/iter?partition=blocks&limit=1000000" | get items | length)
        print $"found ($total_blocks) total blocks before GC"

        print $"deleting repo ($repo1)..."
        http delete -t application/json $"($url)/repos" --data [ { did: ($repo1), delete_data: true } ]

        # ack events specifically for repo1? Actually wait, the events endpoint contains all events.
        # we will ack all events to be safe. Since repo2 is NOT deleted, its refcount should be fine even if events are acked.
        ack-all-events $debug_url $url
        sleep 1sec

        # repo1 should have expected refcount 0
        check-repo-refcounts $debug_url $repo1 0
        # Wait, for repo2, we didn't delete it, but events are acked, so its refcount should be 1
        check-repo-refcounts $debug_url $repo2 1

        compact-and-check-blocks $debug_url $repo2_blocks
    }

    print "all gc tests passed!"
}