diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/lpm.c | 16 | ||||
| -rw-r--r-- | src/lpm.lua | 20 |
2 files changed, 27 insertions, 9 deletions
@@ -1503,6 +1503,22 @@ int main(int argc, char* argv[]) { #endif lua_setglobal(L, "SYSTMPDIR"); + #if _WIN32 + wchar_t selfpath[MAX_PATH] = {0}; + if (GetModuleFileNameW(0, selfpath, MAX_PATH - 1)) + lua_toutf8(L, selfpath); + else + lua_pushnil(L); + #else + char selfpath[MAX_PATH] = {0}; + int length = readlink("/proc/self/exe", selfpath, MAX_PATH); + if (length > 0) + lua_pushlstring(L, selfpath, length); + else + lua_pushnil(L); + #endif + lua_setglobal(L, "EXEFILE"); + lua_pushliteral(L, LITE_ARCH_TUPLE); lua_setglobal(L, "ARCH"); lua_pushliteral(L, LPM_DEFAULT_REPOSITORY); diff --git a/src/lpm.lua b/src/lpm.lua index 1fffb39..e1c751a 100644 --- a/src/lpm.lua +++ b/src/lpm.lua @@ -411,7 +411,11 @@ end function common.dirname(path) local s = path:reverse():find("[/\\]") if not s then return path end return path:sub(1, #path - s) end function common.basename(path) local s = path:reverse():find("[/\\]") if not s then return path end return path:sub(#path - s + 2) end -function common.path(exec) return common.first(common.map({ common.split(":", os.getenv("PATH")) }, function(e) return e .. PATHSEP .. exec end), function(e) local s = system.stat(e) return s and s.type ~= "dir" and s.mode and s.mode & 73 and (not s.symlink or system.stat(s.symlink)) end) end +function common.path(exec) + -- On windows, in theory to resolve things, we also check the working directory even without a PATHSEP. + if exec:find(PATHSEP) or PLATFORM == "windows" and system.stat(exec) then return exec end + return common.first(common.map({ common.split(":", os.getenv("PATH")) }, function(e) return e .. PATHSEP .. exec end), function(e) local s = system.stat(e) return s and s.type ~= "dir" and s.mode and s.mode & 73 and (not s.symlink or system.stat(s.symlink)) end) +end function common.normalize_path(path) if PLATFORM == "windows" and path then path = path:gsub("/", PATHSEP) end if not path or not path:find("^~") then return path end return os.getenv("HOME") .. path:sub(2) end function common.rmrf(root) local info = root and root ~= "" and system.stat(root) @@ -2031,23 +2035,21 @@ end function lpm.self_upgrade(release) if not DEFAULT_RELEASE_URL or #DEFAULT_RELEASE_URL == 0 then error("self-upgrade has been disabled on lpm version " .. VERSION .. "; please upgrade it however you installed it") end - local path = ARGS[1]:find(PATHSEP) and system.stat(ARGS[1]) and ARGS[1] or common.path(ARGS[1]) - if not path then error("can't find path to lpm") end release = release or "latest" local release_url = release and release:find("^https://") and release or (DEFAULT_RELEASE_URL:gsub("%%r", release)) - local stat = system.stat(path) - if not stat then error("can't find lpm at " .. path) end + local stat = EXEFILE and system.stat(EXEFILE) + if not stat then error("can't find lpm at " .. EXEFILE) end local new_temporary_file = SYSTMPDIR .. PATHSEP .. "lpm.upgrade" local old_temporary_file = SYSTMPDIR .. PATHSEP .. "lpm.backup" common.rmrf(new_temporary_file) common.rmrf(old_temporary_file) local status, err = pcall(common.get, release_url, { cache = SYSTMPDIR, target = new_temporary_file, callback = write_progress_bar }) if not status then error("can't find release for lpm at " .. release_url .. (VERBOSE and (": " .. err) or "")) end - if common.is_path_different(new_temporary_file, path) then - status, err = pcall(common.rename, path, old_temporary_file) + if common.is_path_different(new_temporary_file, EXEFILE) then + status, err = pcall(common.rename, EXEFILE, old_temporary_file) if not status then error("can't move lpm executable; do you need to " .. (PLATFORM == "windows" and "run as administrator" or "be root") .. "?" .. (VERBOSE and ": " .. err or "")) end - common.rename(new_temporary_file, path) - system.chmod(path, stat.mode) + common.rename(new_temporary_file, EXEFILE) + system.chmod(EXEFILE, stat.mode) if PLATFORM ~= "windows" then -- because we can't delete the running executbale on windows common.rmrf(old_temporary_file) end |