#!/usr/bin/lua

-- rpm call

-- debug=true lua -- copy_jdk_configs.lua   --currentjvm "%{uniquesuffix %{nil}}" --jvmdir "%{_jvmdir %{nil}}" --origname "%{name}" --origjavaver "%{javaver}" --arch "%{_arch}"

--test call

-- debug=true lua -- copy_jdk_configs.lua   --currentjvm "java-1.8.0-openjdk-1.8.0.65-3.b17.fc22.x86_64" --jvmdir "/usr/lib/jvm" --origname "java-1.8.0-openjdk" --origjavaver "1.8.0" --arch "x86_64" --jvmDestdir /home/jvanek/Desktop

local M = {}

if (os.getenv("debug") == "true") then

    debug = true;

else

    debug = false;

end

local function debugOneLinePrint(string)

    if (debug) then

        print(string)

    end ;

end

function getPath(str, sep)

    sep = sep or '/'

    return str:match("(.*" .. sep .. ")")

end

function splitToTable(source, pattern)

    local i1 = string.gmatch(source, pattern)

    local l1 = {}

    for i in i1 do

        table.insert(l1, i)

    end

    return l1

end

local function slurp(path)

    local f = io.open(path)

    local s = f:read("*a")

    f:close()

    return s

end

function trim(s)

    return (s:gsub("^%s*(.-)%s*$", "%1"))

end

local function dirWithParents(path)

    local s = ""

    local dirs = splitToTable(path, "[^/]+")

    for i, d in pairs(dirs) do

        if (i == #dirs) then

            break

        end

        s = s .. "/" .. d

        local stat2 = posix.stat(s, "type");

        if (stat2 == nil) then

            debugOneLinePrint(s .. " does not exists, creating")

            if (not dry) then

                posix.mkdir(s)

            end

        else

            debugOneLinePrint(s .. " exists,not creating")

        end

    end

end

-- main function,

-- formelry main body

-- move to function resolved

-- https://bugzilla.redhat.com/show_bug.cgi?id=1892224

-- for readability not indented, todo, indent once tuned

function M.mainProgram(arg)

    debugOneLinePrint("cjc: lua debug on")

    local caredFiles = { "jre/lib/calendars.properties",

                         "jre/lib/content-types.properties",

                         "jre/lib/flavormap.properties",

                         "jre/lib/logging.properties",

                         "jre/lib/net.properties",

                         "jre/lib/psfontj2d.properties",

                         "jre/lib/sound.properties",

                         "jre/lib/deployment.properties",

                         "jre/lib/deployment.config",

                         "jre/lib/security/US_export_policy.jar",

                         "jre/lib/security/unlimited/US_export_policy.jar",

                         "jre/lib/security/limited/US_export_policy.jar",

                         "jre/lib/security/policy/unlimited/US_export_policy.jar",

                         "jre/lib/security/policy/limited/US_export_policy.jar",

                         "jre/lib/security/java.policy",

                         "jre/lib/security/java.security",

                         "jre/lib/security/local_policy.jar",

                         "jre/lib/security/unlimited/local_policy.jar",

                         "jre/lib/security/limited/local_policy.jar",

                         "jre/lib/security/policy/unlimited/local_policy.jar",

                         "jre/lib/security/policy/limited/local_policy.jar",

                         "jre/lib/security/nss.cfg",

                         "jre/lib/security/cacerts",

                         "jre/lib/security/blacklisted.certs",

                         "jre/lib/security/jssecacerts",

                         "jre/lib/security/trusted.certs",

                         "jre/lib/security/trusted.jssecerts",

                         "jre/lib/security/trusted.clientcerts",

                         "jre/lib/ext",

                         "jre/lib/security/blacklist",

                         "jre/lib/security/javaws.policy",

                         "jre/lib/security/nss.fips.cfg",

                         "lib/security",

                         "conf",

                         "lib/ext" }

    -- before import to allow run from spec

    if (arg[1] == "--list") then

        for i, file in pairs(caredFiles) do

            print(file)

        end

        return 0;

    end

    -- yum install lua-posix

    local posix = require "posix"

    -- the one we are installing

    local currentjvm = nil

    local jvmdir = nil

    local jvmDestdir = nil

    local origname = nil

    local origjavaver = nil

    local arch = nil

    local temp = nil;

    local dry = false;

    for i = 1, #arg, 2 do

        if (arg[i] == "--help" or arg[i] == "-h") then

            print("all but jvmDestdir and debug are mandatory")

            print("  --currentjvm")

            print("    NVRA of currently installed java")

            print("  --jvmdir")

            print("    Directory where to find this kind of virtual machine. Generally /usr/lib/jvm")

            print("  --origname")

            print("    convinient switch to determine jdk. Generally java-1.X.0-vendor")

            print("  --origjavaver")

            print("    convinient switch to determine jdk's version. Generally 1.X.0")

            print("  --arch")

            print("    convinient switch to determine jdk's arch")

            print("  --jvmDestdir")

            print("    Migration/testing switch. Target Mostly same as jvmdir, but you may wont to copy ouside it.")

            print("  --debug or $debug")

            print("    Enables printing out whats going on. true/false. False by default")

            print("  --temp")

            print("    optional file to save intermediate result - directory configs were copied from")

            print("  --dry")

            print("    true/fase if true, then no changes will be written to disk except one tmp file. False by default")

            print("  **** specil parasm ****")

            print("  --list")

            print("    if present on cmdline, list all cared files and exists")

            os.exit(0)

        end

        if (arg[i] == "--currentjvm") then

            currentjvm = arg[i + 1]

        end

        if (arg[i] == "--jvmdir") then

            jvmdir = arg[i + 1]

        end

        if (arg[i] == "--origname") then

            origname = arg[i + 1]

        end

        if (arg[i] == "--origjavaver") then

            origjavaver = arg[i + 1]

        end

        if (arg[i] == "--arch") then

            arch = arg[i + 1]

        end

        if (arg[i] == "--jvmDestdir") then

            jvmDestdir = arg[i + 1]

        end

        if (arg[i] == "--debug") then

            --no string, boolean, workaround

            if (arg[i + 1] == "true") then

                debug = true

            end

        end

        if (arg[i] == "--dry") then

            --no string, boolean, workaround

            if (arg[i + 1] == "true") then

                dry = true

            end

        end

        if (arg[i] == "--temp") then

            temp = arg[i + 1]

        end

    end

    if (jvmDestdir == nil) then

        jvmDestdir = jvmdir

    end

    if (debug) then

        print("--currentjvm:");

        print(currentjvm);

        print("--jvmdir:");

        print(jvmdir);

        print("--jvmDestdir:");

        print(jvmDestdir);

        print("--origname:");

        print(origname);

        print("--origjavaver:");

        print(origjavaver);

        print("--arch:");

        print(arch);

        print("--debug:");

        print(debug);

    end

    --trasnform substitute names to lua patterns

    local name = string.gsub(string.gsub(origname, "%-", "%%-"), "%.", "%%.")

    local javaver = string.gsub(origjavaver, "%.", "%%.")

    local jvms = { }

    debugOneLinePrint("started")

    foundJvms = posix.dir(jvmdir);

    if (foundJvms == nil) then

        debugOneLinePrint("no, or nothing in " .. jvmdir .. " exit")

        return

    end

    debugOneLinePrint("found " .. #foundJvms .. " jvms")

    for i, p in pairs(foundJvms) do

        -- regex similar to %{_jvmdir}/%{name}-%{javaver}*%{_arch} bash command

        if (string.find(p, name .. "%-" .. javaver .. ".*" .. arch) ~= nil) then

            debugOneLinePrint("matched:  " .. p)

            if (currentjvm == p) then

                debugOneLinePrint("this jdk is already installed. exiting lua script")

                return

            end ;

            if (string.match(p, ".*-debug$")) then

                print(p .. " matched but seems to be debug variant. Skipping")

            else

                table.insert(jvms, p)

            end

        else

            debugOneLinePrint("NOT matched:  " .. p)

        end

    end

    if (#jvms <= 0) then

        debugOneLinePrint("no matching jdk in " .. jvmdir .. " exit")

        return

    end ;

    debugOneLinePrint("matched " .. #jvms .. " jdk in " .. jvmdir)

    --full names are like java-1.7.0-openjdk-1.7.0.60-2.4.5.1.fc20.x86_64

    table.sort(jvms, function(a, b)

        -- version-sort

        -- split on non word: . -

        local l1 = splitToTable(a, "[^%.-]+")

        local l2 = splitToTable(b, "[^%.-]+")

        for x = 1, math.min(#l1, #l2) do

            local l1x = tonumber(l1[x])

            local l2x = tonumber(l2[x])

            if (l1x ~= nil and l2x ~= nil) then

                --if hunks are numbers, go with them

                if (l1x < l2x) then

                    return true;

                end

                if (l1x > l2x) then

                    return false;

                end

            else

                if (l1[x] < l2[x]) then

                    return true;

                end

                if (l1[x] > l2[x]) then

                    return false;

                end

            end

            -- if hunks are equals then move to another pair of hunks

        end

        return a < b

    end)

    if (debug) then

        print("sorted lsit of jvms")

        for i, file in pairs(jvms) do

            print(file)

        end

    end

    latestjvm = jvms[#jvms]

    if (temp ~= nil) then

        src = jvmdir .. "/" .. latestjvm

        debugOneLinePrint("temp declared as " .. temp .. " saving used dir of " .. src)

        file = io.open(temp, "w")

        file:write(src)

        file:close()

    end

    local readlinkOutput = os.tmpname()

    for i, file in pairs(caredFiles) do

        local SOURCE = jvmdir .. "/" .. latestjvm .. "/" .. file

        local DEST = jvmDestdir .. "/" .. currentjvm .. "/" .. file

        debugOneLinePrint("going to copy " .. SOURCE)

        debugOneLinePrint("to  " .. DEST)

        local stat1 = posix.stat(SOURCE, "type");

        if (stat1 ~= nil) then

            debugOneLinePrint(SOURCE .. " exists")

            dirWithParents(DEST)

            -- Copy with -a to keep everything intact

            local exe = "cp" .. " -ar " .. SOURCE .. " " .. DEST

            local linkExe = "readlink" .. " -f " .. SOURCE .. " > " .. readlinkOutput

            debugOneLinePrint("executing " .. linkExe)

            os.remove(readlinkOutput)

            os.execute(linkExe)

            local link = trim(slurp(readlinkOutput))

            debugOneLinePrint("  ...link is " .. link)

            if (not ((link) == (SOURCE))) then

                debugOneLinePrint("WARNING link " .. link .. " where file " .. SOURCE .. " expected!")

                debugOneLinePrint("Will try to copy link target rather then link itself!")

                --replacing  any NVRA by future NVRA (still execting to have NVRA for any multiple-installable targets

                -- lua stubbornly consider dash as inteval. Replacing by dot to match X-Y more correct as X.Y rather then not at all

                local linkDest = string.gsub(link, latestjvm:gsub("-", "."), currentjvm)

                debugOneLinePrint("attempting to copy " .. link .. " to " .. linkDest)

                if (link == linkDest) then

                    debugOneLinePrint("Those are identical files! Nothing to do!")

                else

                    local exe2 = "cp" .. " -ar " .. link .. " " .. linkDest

                    dirWithParents(linkDest)

                    debugOneLinePrint("executing " .. exe2)

                    if (not dry) then

                        os.execute(exe2)

                    end

                end

            else

                debugOneLinePrint("executing " .. exe)

                if (not dry) then

                    os.execute(exe)

                end

            end

        else

            debugOneLinePrint(SOURCE .. " does not exists")

        end

    end

end --unindented main function

if (arg == nil) then

    debugOneLinePrint("arg variable is nil, you have to call mainProgram manually") -- this can actually not be invoked, as the debug is set via arg

else

    M.mainProgram(arg)

end

return M