script_name = "Audio Clipper"
script_description = "Extracts audio from the selected line(s).\nNeeds ffmpeg."
script_author = "garret"
script_version = "dev"

aegi = aegisub

function err(msg)
    aegi.dialog.display({{class = "label", label = msg}}, {"OK"}, {close = "OK"})
    aegi.cancel()
end

function get_vid_dir()
    local aud_dir = aegi.decode_path("?audio")
    local vid_dir = aegi.decode_path("?video")
    if aud_dir ~= "?audio" then
        return aud_dir
    elseif vid_dir ~= "?video" then -- if there is not, in fact, a video
        return vid_dir
    else
        return nil
    end
end

function get_vid()
    local aud = aegi.project_properties().audio_file -- try get the audio first, if it's loaded separately
    local vid = aegi.project_properties().video_file
    if aud ~= "" then
        return aud
    elseif vid ~= "" then
        return vid
    else
        return nil
    end
end

function get_format(copy, format, custom)
    if copy == true then -- if we do want to copy
        if format == "" then -- format is the thing from the dropdown
            err("Need a format!")
        elseif format == "AAC" then -- these are the most common audio formats i've seen
            return "m4a"
        elseif format == "Opus" then
            return "opus"
        elseif format == "FLAC" then
            return "flac" elseif format == "Custom" then -- but I am not all-knowing, so you can put your own
            return custom
        end
    else -- if we don't want to copy (i.e. we've pressed "Just make it FLAC")
        return "flac"
    end
end

function make_out_path(out_path)
    if out_path == "" then
        err("Need an output path!")
    end
    os.execute('mkdir "'..out_path..'"') -- if it doesn't exist, it makes it, and if it does, it errors.
    return out_path -- either way, the path now exists. (probably)
end -- I imagine there's a mkdir thing in lua that's better, but this works fine, so unless it breaks, i'll keep it like this.

function extract_audio(in_path, start_time, end_time, out_path, name, extension, copy)
    if copy == true then
        os.execute('ffmpeg -i "'..in_path..'" -codec copy -vn -ss '..start_time..'ms -to '..end_time..'ms "'..out_path..name..'.'..extension..'" -y')
     -- takes the video as input, copies the streams, but doesn't include video. sets it to start at the start of the selection, and end at the end of it. outputs to our chosen out path with our chosen name and extension. overwrites anything with the same name.
    elseif copy == false then
        os.execute('ffmpeg -i "'..in_path..'" -vn -ss '..start_time..'ms -to '..end_time..'ms "'..out_path..name..'.'..extension..'" -y')
     -- same as above, but doesn't copy the stream (transcodes to flac)
    end
end

function loop(subs, sel, in_path, out_path, extension, copy, delay)
    aegi.progress.title("Extracting Audio")
    local progress = 0
    local progress_increment = 100 / #sel -- increment by this for every line, and the bar will eventually reach 100
    for x, i in ipairs(sel) do -- x is the position of the line in our selection, i is the position in the whole sub file
        if aegi.progress.is_cancelled() then -- if you press the cancel button
            aegi.cancel() -- it stops (mind-blowing, i know)
        end
        aegi.progress.set(progress)
        local line = subs[i]
        local start_time = line.start_time + delay
        local end_time = line.end_time + delay
        aegi.progress.task("Extracting line "..x)
        extract_audio(in_path, start_time, end_time, out_path, x, extension, copy)
        progress = progress + progress_increment
    end
    aegi.progress.set(100) -- in case it didn't reach 100 on its own
end


function gui(subs, sel)
    local in_path = get_vid()
    local out_path = get_vid_dir()
    if in_path == nil or out_path == nil then -- if no video is loaded
        in_path = "No audio/video loaded. Specify a path."
        out_path = in_path -- both the same error message
    else
        out_path = out_path.."/audioclipper_output/" -- make the out path the one we actually want
    end

    local get_input={{class="label",x=0,y=0,label="Input's audio format:"},{class="dropdown",name="format",x=0,y=1,width=2,height=1,items={"AAC","Opus","FLAC","Custom"},value="Audio Format",hint="If you don't know, you should probably press \"Just make it FLAC\", or use mka."},{class="label",x=0,y=2,label="Custom Extension:"},{class="edit",name="custom",x=1,y=2,value="mka",hint="You'll probably be fine with mka, because matroska can contain pretty much anything"},{class="label",x=0,y=3,label="Delay (ms):"},{class="intedit",name="delay",x=1,y=3,value=0,hint="to prevent timing fuckery with weird raws"},{class="label",x=0,y=4,label="Input path:"},{class="edit",name="in_path",x=0,y=5,width=2,height=1,value=in_path,hint="where the audio comes from"},{class="label",x=0,y=6,label="Output path (will be created if it doesn't already exist):"},{class="edit",name="out_path",x=0,y=7,width=2,height=1,value=out_path,hint="where the audio goes"}}
    local pressed, results = aegi.dialog.display(get_input, {"Cancel", "OK", "Just make it FLAC"})
    -- there's probably something that can detect the format automatically, but I do not know what it is.
    if pressed == "Cancel" then
        aegi.cancel()
    elseif pressed == "OK" then
        do_copy = true
    elseif pressed == "Just make it FLAC" then
        do_copy = false
    end
    local extension = get_format(do_copy, results.format, results.custom) -- gets file extension
    local delay = results.delay
    in_path = results.in_path
    out_path = make_out_path(results.out_path)
    loop(subs, sel, in_path, out_path, extension, do_copy, delay)
end

function non_gui(subs, sel) -- no gui, so you can bind it to a hotkey
    local vid = get_vid()
    if vid == nil then
        err("Need an input!\nSpecify a path in the GUI, or load one in aegisub.")
    else
        loop(subs, sel, get_vid(), make_out_path(get_vid_dir().."/audioclipper_output/"), 'flac', false, 0)
    end
 -- sets sane defaults (takes the audio from the video file, and outputs to /the/video/dir/audioclipper_output/. transcodes to flac. no delay)
end

aegi.register_macro(script_name, script_description, gui)
aegi.register_macro(": Non-GUI macros :/"..script_name..": Just make it FLAC", script_description, non_gui) -- same section as unanimated's scripts