blog / scripts / thumb.jl

include("ImageProcessing.jl")
using .ImageProcessing

percentage_position = 0.1

function process_videos_recursive(dir::String)
    for item in readdir(dir)
        if item == ".DS_Store"
            continue
        end

        path = joinpath(dir, item)

        if isdir(path)
            process_videos_recursive(path)
        else
            ext = lowercase(splitext(item)[2])

            if ext in [".mp4", ".avi", ".mov", ".mkv", ".webm", ".m4v"]
                output_name = splitext(item)[1] * ".png"
                output_path = joinpath("static/video/thumb", output_name)

                if isfile(output_path)
                    continue
                end

                try
                    extract_video_frame(path, output_path, percentage_position)
                catch e
                    @error "Failed to process video" path error = e
                end
            end
        end
    end
end

function main(args::Vector{String})
    mkpath("static/video/thumb")

    for dir in args
        process_videos_recursive(dir)
    end
end

if abspath(PROGRAM_FILE) == @__FILE__
    main(ARGS)
end