Limbus Wordle

using ObiterDicta

function getIDArray(personality)
    Ans = Any[]
    for fn in [ObiterDicta.getSkill1, ObiterDicta.getSkill2, ObiterDicta.getSkill3]
        Skill = fn(personality)
        Sin = ObiterDicta.getSinType(Skill)
        AtkType = ObiterDicta.getAtkType(Skill)
        NumCoins = ObiterDicta.getNumCoins(Skill, 4)

        push!(Ans, Sin)
        push!(Ans, AtkType)
        push!(Ans, NumCoins)
    end

    Sinner = ObiterDicta.getSinnerName(personality)
    Details = ObiterDicta.getFullTitle(personality)
    push!(Ans, Sinner)
    push!(Ans, Details)

    Ans
end

function compareArrays(arr1, arr2)
    Ans = Int[]
    for (x, y) in zip(arr1, arr2)
        if x isa Int && y isa Int
            if x < y 
                push!(Ans, -1)
            elseif x == y
                push!(Ans, 0)
            else
                push!(Ans, 1)
            end
        else
            push!(Ans, x == y)
        end
    end

    Ans
end

function compareArrStr(val, sinner; special = false)
    Ans = String[]
    for i in 1:3
        SkillStr = String[]
        push!(SkillStr, val[3i-2] == 1 ? "O" : "X")
        push!(SkillStr, val[3i-1] == 1 ? "O" : "X")
        if val[3i] == 0
            push!(SkillStr, "=")
        elseif val[3i] < 0
            push!(SkillStr, ">")
        else
            push!(SkillStr, "<")
        end
        CombineSkillStr = join(SkillStr, ", ")
        push!(Ans, "$CombineSkillStr")
    end

    push!(Ans, val[10] == 1 ? "O" : "X")
    push!(Ans, sinner)
    if special 
        Ans[1] = "Skill 1<br>"*Ans[1]
        Ans[2] = "Skill 2<br>"*Ans[2]
        Ans[3] = "Skill 3<br>"*Ans[3]
        Ans[4] = "Sinner<br>"*Ans[4]
    end

    "| " * join(Ans, " | ") * " |"
end 

MainList = ObiterDicta.getMasterList(ObiterDicta.Personality)
ArrayData = getIDArray.(MainList)

compareRows(i, j) = compareArrays(ArrayData[i], ArrayData[j])

function OutputCollision()
    println("| Initial Guess | Conflict 1 | Conflict 2 | Disambiguated # |")
    println("| --- | --- | --- | --- |")

    for i in 1:length(ArrayData)
        countMap = Dict{Any, Vector{Int}}()
        for j in 1: length(ArrayData)
            T = compareRows(i, j)
            if haskey(countMap, T)
                push!(countMap[T], j)
            else
                countMap[T] = [j]
            end
        end

        Disambiguated = 0
        InitialGuess = ArrayData[i][end]
        Conflict1 = ""
        Conflict2 = ""
        
        for (_, val) in countMap
            if length(val) == 1
                Disambiguated += 1
            else
                Conflict1 = ArrayData[val[1]][end]
                Conflict2 = ArrayData[val[2]][end]
            end
        end
        # Disambiguated < 90 && continue
        println("| $InitialGuess | $Conflict1 | $Conflict2 | $Disambiguated |")
    end
end


function FindSecondGuess(i = 88) # Rosespanner Rodion is 88
    linebreak() = println(("| --- ")^5*" |")
    countMap = Dict{Any, Vector{Int}}()
    for j in 1: length(ArrayData)
        T = compareRows(i, j) 
        if haskey(countMap, T)
            push!(countMap[T], j)
        else
            countMap[T] = [j]
        end
    end

    println("| Skill 1 | Skill 2 | Skill 3 | Sinner | Output |")
    linebreak()

    Lines = String[]
    for (key, val) in countMap
        length(val) == 1 || continue
        SinnerName = ArrayData[val[begin]][end]
        push!(Lines, compareArrStr(key, SinnerName))
    end
    sort!(Lines)
    for line in Lines
        println(line)
    end

    for (key, val) in countMap
        length(val) == 1 && continue
        Flag = -1
        for secondGuess in val
            Comparisons = [compareRows(secondGuess, j) for j in val]
            if length(unique(Comparisons)) == length(val)
                Flag = secondGuess
                break
            end
        end

        if Flag == -1
            for secondGuess in 1:length(ArrayData)
                Comparisons = [compareRows(secondGuess, j) for j in val]
                if length(unique(Comparisons)) == length(val)
                    Flag = secondGuess
                    break
                end
            end
        end

        if Flag != -1
            println()
            NewGuess = ArrayData[Flag][end]
            println(compareArrStr(key, "Guess: $NewGuess"; special = true))
            linebreak()
            Lines = String[]
            for hiddenID in val
                SinnerName = ArrayData[hiddenID][end]
                push!(Lines, compareArrStr(compareRows(Flag, hiddenID), SinnerName))
            end
            sort!(Lines)
            for line in Lines
                println(line)
            end
        else
            @info "Cannot disambiguate $val"
        end
    end
end