support python plugin

python/python.lua

@@ -0,0 +1,224 @@
+---    Copyright 2024 [lihan [email protected]]
+--
+--   Licensed under the Apache License, Version 2.0 (the "License");
+--   you may not use this file except in compliance with the License.
+--   You may obtain a copy of the License at
+--
+--     http://www.apache.org/licenses/LICENSE-2.0
+--
+--   Unless required by applicable law or agreed to in writing, software
+--   distributed under the License is distributed on an "AS IS" BASIS,
+--   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+--   See the License for the specific language governing permissions and
+--   limitations under the License.
+
+local http = require("http")
+local html = require("html")
+OS_TYPE = ""
+ARCH_TYPE = ""
+
+local PYTHON_URL = "https://www.python.org/ftp/python/"
+
+local DOWNLOAD_SOURCE = {
+    --- TODO support zip or web-based installers
+    WEB_BASED = "https://www.python.org/ftp/python/%s/python-%s%s-webinstall.exe",
+    ZIP = "https://www.python.org/ftp/python/%s/python-%s-embed-%s.zip",
+    MSI = "",
+    --- Currently only exe installers are supported
+    EXE = "https://www.python.org/ftp/python/%s/python-%s%s.exe",
+    SOURCE = "https://www.python.org/ftp/python/%s/Python-%s.tar.xz"
+}
+
+PLUGIN = {
+    --- Plugin name
+    name = "python",
+    --- Plugin author
+    author = "aooohan",
+    --- Plugin version
+    version = "0.0.1",
+    description = "For Windows, only support >=3.5.0, but no restrictions for unix-like",
+    -- Update URL
+    updateUrl = "https://github.com/aooohan/version-fox-plugins/blob/main/python/python.lua",
+    minRuntimeVersion = "0.2.2",
+}
+
+function PLUGIN:PreInstall(ctx)
+    local version = ctx.version
+    if version == "latest" then
+        version = self:Available({})[1].version
+    end
+    if not checkIsReleaseVersion(version) then
+        error("The current version is not released")
+        return
+    end
+    if OS_TYPE == "windows" then
+        local url, filename = checkAvailableReleaseForWindows(version)
+        return {
+            version = version,
+            url = url,
+            note = filename
+        }
+    else
+        return {
+            version = version,
+        }
+    end
+end
+
+function checkAvailableReleaseForWindows(version)
+    local archType = ARCH_TYPE
+    if ARCH_TYPE == "386" then
+        archType = ""
+    else
+        archType = "-" .. archType
+    end
+    --- Currently only exe installers are supported
+    --- TODO support zip or web-based installers
+    local url = DOWNLOAD_SOURCE.EXE:format(version, version, archType)
+    local resp, err = http.head({
+        url = url
+    })
+    if err ~= nil or resp.status_code ~= 200 then
+        error("No available installer found for current version")
+    end
+    return url, "python-" .. version .. archType .. ".exe"
+end
+
+function linuxCompile(ctx)
+    local sdkInfo = ctx.sdkInfo['python']
+    local path = sdkInfo.path
+    local version = sdkInfo.version
+    local pyenv_url = "https://github.com/pyenv/pyenv.git"
+    local dest_pyenv_path = ctx.rootPath .. "/pyenv"
+    local status = os.execute("git clone " .. pyenv_url .. " " .. dest_pyenv_path)
+    if status ~= 0 then
+        error("git clone failed")
+    end
+    local pyenv_build_path = dest_pyenv_path .. "/plugins/python-build/bin/python-build"
+    print("Building python ...")
+    status = os.execute(pyenv_build_path .. " " .. version .. " " .. path)
+    if status ~= 0 then
+        error("python build failed")
+    end
+    print("Build python success!")
+    print("Cleaning up ...")
+    status = os.execute("rm -rf " .. dest_pyenv_path)
+    if status ~= 0 then
+        error("remove build tool failed")
+    end
+end
+
+function windowsCompile(ctx)
+    local sdkInfo = ctx.sdkInfo['python']
+    local path = sdkInfo.path
+    local filename = sdkInfo.note
+    local qInstallFile = path .. "/" .. filename
+    local qInstallPath = path
+    --local exitCode = os.execute('msiexec /quiet /a "' .. qInstallFile .. '" TargetDir="' .. qInstallPath .. '"')
+    local exitCode = os.execute('"' .. qInstallFile .. '" /quiet InstallAllUsers=0 PrependPath=0 TargetDir="' .. qInstallPath .. '"')
+    if exitCode ~= 0 then
+        error("error installing python")
+    end
+    print("Cleaning up ...")
+    os.remove(qInstallFile)
+end
+
+function PLUGIN:PostInstall(ctx)
+    if OS_TYPE == "windows" then
+        return windowsCompile(ctx)
+    else
+        return linuxCompile(ctx)
+    end
+end
+
+function PLUGIN:Available(ctx)
+    return parseVersion()
+end
+
+function checkIsReleaseVersion(version)
+    local resp, err = http.head({
+        url = DOWNLOAD_SOURCE.SOURCE:format(version, version)
+    })
+    if err ~= nil or resp.status_code ~= 200 then
+        return false
+    end
+    return true
+end
+
+function parseVersion()
+    local resp, err = http.get({
+        url = PYTHON_URL
+    })
+    if err ~= nil or resp.status_code ~= 200 then
+        error("paring release info failed." .. err)
+    end
+    local result = {}
+    html.parse(resp.body):find("a"):each(function(i, selection)
+        local href = selection:attr("href")
+        local sn = string.match(href, "^%d")
+        local es = string.match(href, "/$")
+        if sn and es then
+            local vn = string.sub(href, 1, -2)
+            if OS_TYPE == "windows" then
+                if compare_versions(vn, "3.5.0") >= 0 then
+                    table.insert(result, {
+                        version = string.sub(href, 1, -2),
+                        note = "",
+                    })
+                end
+            else
+                table.insert(result, {
+                    version = vn,
+                    note = "",
+                })
+            end
+        end
+    end)
+    table.sort(result, function(a, b)
+        return compare_versions(a.version, b.version) > 0
+    end)
+    return result
+end
+
+function PLUGIN:EnvKeys(ctx)
+    local mainPath = ctx.path
+    if OS_TYPE == "windows" then
+        return {
+            {
+                key = "PATH",
+                value = mainPath
+            }
+        }
+    else
+        return {
+            {
+                key = "PATH",
+                value = mainPath .. "/bin"
+            }
+        }
+    end
+end
+
+function compare_versions(v1, v2)
+    local v1_parts = {}
+    for part in string.gmatch(v1, "[^.]+") do
+        table.insert(v1_parts, tonumber(part))
+    end
+
+    local v2_parts = {}
+    for part in string.gmatch(v2, "[^.]+") do
+        table.insert(v2_parts, tonumber(part))
+    end
+
+    for i = 1, math.max(#v1_parts, #v2_parts) do
+        local v1_part = v1_parts[i] or 0
+        local v2_part = v2_parts[i] or 0
+        if v1_part > v2_part then
+            return 1
+        elseif v1_part < v2_part then
+            return -1
+        end
+    end
+
+    return 0
+end