feat: Compute frozen stake

Signed-off-by: Anirudh <[email protected]>

src/AllocationOpt.jl

@@ -8,8 +8,40 @@ using TOML
 using TypedTables
 using TheGraphData
 
+import SplitApplyCombine as SAC
+
 include("configuration.jl")
 include("data.jl")
 include("domain.jl")
 
+function julia_main()::Cint
+    try
+        main(ARGS)
+    catch
+        Base.invokelatest(Base.display_error, Base.catch_stack())
+        return 1
+    end
+    return 0 # if things finished successfully
+end
+
+main(args::Vector{String}) = main(first(args))
+
+function main(path::AbstractString)
+    # Read config and set defaults
+    config = path |> readconfig |> configuredefaults!
+    return main(config)
+end
+
+function main(config::Dict)
+    # Read data
+    i, a, s, n = AllocationOpt.read(config)
+
+    # Queried data has not yet been converted to GRT, so
+    # if this isn't data that we've queried, correct the
+    # types and write the data out to CSVs
+    isnothing(config["readdir"]) && write(i, a, s, n, config)
+
+    return nothing
+end
+
 end

src/domain.jl

@@ -160,3 +160,24 @@ julia> AllocationOpt.locked(Val(:indexer), x)
 ```
 """
 locked(::Val{:indexer}, x) = x.lockedTokens |> only
+
+"""
+    frozen(a::FlexTable, config::AbstractDict)
+
+The frozen stake of the indexer with allocations `a`.
+
+```julia
+julia> using AllocationOpt
+julia> using TheGraphData
+julia> a = flextable([
+            Dict("subgraphDeployment.ipfsHash" => "Qma", "allocatedTokens" => 5),
+            Dict("subgraphDeployment.ipfsHash" => "Qmb", "allocatedTokens" => 10),
+       ])
+julia> config = Dict("frozenlist" => ["Qma", "Qmb"])
+julia> AllocationOpt.frozen(a, config)
+```
+"""
+function frozen(a::FlexTable, config::AbstractDict)
+    frozenallocs = SAC.filterview(r -> ipfshash(Val(:allocation), r) ∈ config["frozenlist"], a)
+    return stake(Val(:allocation), frozenallocs) |> sum
+end

test/domain.jl

@@ -32,4 +32,15 @@
 
     end
 
+    @testset "frozen" begin
+        a = flextable([
+            Dict("subgraphDeployment.ipfsHash" => "Qma", "allocatedTokens" => 5),
+            Dict("subgraphDeployment.ipfsHash" => "Qmb", "allocatedTokens" => 10),
+        ])
+        config = Dict("frozenlist" => ["Qma", "Qmb"])
+        @test AllocationOpt.frozen(a, config) == 15
+        config = Dict("frozenlist" => ["Qmb"])
+        @test AllocationOpt.frozen(a, config) == 10
+    end
+
 end