using Pkg: Pkg
using Revise: Revise

Pkg.activate("visualization")
Revise.includet("../visualization/Animation.jl")
using .Animation: Animation

function get_last_done_dir(main_dir::String)
    paths = readdir(main_dir; join=true)
    skipped_last_dir = false

    for path_ind in length(paths):-1:1
        path = paths[path_ind]
        if isdir(path)
            if !skipped_last_dir
                skipped_last_dir = true
                continue
            end

            return path
        end
    end
end

function plot_watch(main_dir::String)
    last_plotted_dir = ""

    Revise.entr([main_dir]; pause=1.0) do
        dir = get_last_done_dir(main_dir)
        if !(dir == last_plotted_dir)
            last_plotted_dir = dir
            Animation.animate(dir)
        end
    end

    return nothing
end