Config = {
    Suffixes = {
        '.c',
        '.cpp',
        '.h',
        '.hpp',
        '.inl',
    }
}

local function setup(config)
    if config then
        for key, value in pairs(config) do
            Config[key] = config[key] or value
        end
    end
end

local function file_exists(path)
    return vim.loop.fs_stat(path) ~= nil
end

local function path_pivot(path, suffix)
    local start, stop = path:find(suffix)
    if start and stop == #path then
        return path:sub(0, start - 1)
    end
end

local function revolve(back)
    local path = vim.api.nvim_buf_get_name(vim.api.nvim_get_current_buf())

    local suffix_index = nil
    local prefix = nil
    for i, s in ipairs(Config.Suffixes) do
        prefix = path_pivot(path, s)
        if prefix then
            suffix_index = i
            break
        end
    end

    if prefix then
        local start = 1
        local stop = #Config.Suffixes - 1
        local step = 1
        if back then
            start, stop = stop, start
            step = -1
        end

        for i = start, stop, step do
            local j = (i + suffix_index - 1) % #Config.Suffixes + 1
            local next_path = prefix .. Config.Suffixes[j]
            if path ~= next_path and file_exists(next_path) then
                vim.cmd(string.format('e %s', next_path))
                break
            end
        end
    end
end

return {
    revolve = revolve,
    setup = setup,
}