oh-my-devpod 0.14.2 → 0.14.3

package/runtime/install/update.sh

@@ -7,6 +7,8 @@ export OHMYDEVPOD_BOOTSTRAP_LIB_ONLY=1
 source "${bundle_root}/install/bootstrap.sh"
 # shellcheck source=../modules/lib/common.sh
 source "${bundle_root}/modules/lib/common.sh"
+# shellcheck source=../modules/lib/source-config.sh
+source "${bundle_root}/modules/lib/source-config.sh"
 
 omd_update_normalize_version() {
   printf '%s\n' "${1#v}"
@@ -33,7 +35,7 @@ omd_update_archive_version() {
 omd_update_snapshot_source_config() {
   local config_dir="$1" snapshot_dir="$2" name
   mkdir -p "${snapshot_dir}" || return 1
-  for name in source mirror-profile env uv.toml install-prefix bin-dir cache-dir; do
+  for name in source npm-source mirror-profile env uv.toml install-prefix bin-dir cache-dir; do
     if [[ -f "${config_dir}/${name}" ]]; then
       cp -p "${config_dir}/${name}" "${snapshot_dir}/${name}" || return 1
     else
@@ -45,7 +47,7 @@ omd_update_snapshot_source_config() {
 omd_update_restore_source_config() {
   local config_dir="$1" snapshot_dir="$2" name
   mkdir -p "${config_dir}" || return 1
-  for name in source mirror-profile env uv.toml install-prefix bin-dir cache-dir; do
+  for name in source npm-source mirror-profile env uv.toml install-prefix bin-dir cache-dir; do
     if [[ -f "${snapshot_dir}/.missing-${name}" ]]; then
       rm -f "${config_dir}/${name}" || return 1
     else
@@ -61,12 +63,24 @@ OMD_UPDATE_BREW_REMOTE=""
 OMD_UPDATE_BREW_REMOTE_PRESENT=0
 OMD_UPDATE_BREW_REMOTE_CHANGED=0
 OMD_UPDATE_BREW_LEGACY=0
+OMD_UPDATE_BREW_CORE_PREFIX=""
+OMD_UPDATE_BREW_CORE_REMOTE=""
+OMD_UPDATE_BREW_CORE_REMOTE_PRESENT=0
+OMD_UPDATE_BREW_CORE_REMOTE_CHANGED=0
 
 omd_update_capture_brew_remote() {
-  local marker_prefix
+  local marker_prefix core_prefix
   omd_module_is_managed linuxbrew || return 0
   marker_prefix="$(omd_module_marker_value linuxbrew artifact || true)"
-  [[ -n "${marker_prefix}" && -d "${marker_prefix}" ]] || return 0
+  [[ -n "${marker_prefix}" && -d "${marker_prefix}" && -x "${marker_prefix}/bin/brew" ]] || {
+    omd_warn "Managed Homebrew prefix is missing or invalid: ${marker_prefix:-<empty>}"
+    return 1
+  }
+  marker_prefix="$(cd "${marker_prefix}" 2>/dev/null && pwd -P)" || return 1
+  omd_module_brew_cmd_matches_prefix "${marker_prefix}/bin/brew" "${marker_prefix}" || {
+    omd_warn "Managed Homebrew command does not belong to its recorded prefix"
+    return 1
+  }
   command -v git >/dev/null 2>&1 || {
     omd_warn "Git is required to switch the managed Homebrew remote"
     return 1
@@ -75,50 +89,91 @@ omd_update_capture_brew_remote() {
   OMD_UPDATE_BREW_PREFIX="${marker_prefix}"
   if [[ ! -d "${marker_prefix}/.git" ]]; then
     OMD_UPDATE_BREW_LEGACY=1
-    return 0
-  fi
-  if OMD_UPDATE_BREW_REMOTE="$(git -C "${marker_prefix}" remote get-url origin 2>/dev/null)"; then
-    OMD_UPDATE_BREW_REMOTE_PRESENT=1
   else
-    OMD_UPDATE_BREW_REMOTE=""
-    OMD_UPDATE_BREW_REMOTE_PRESENT=0
+    if OMD_UPDATE_BREW_REMOTE="$(git -C "${marker_prefix}" remote get-url origin 2>/dev/null)"; then
+      OMD_UPDATE_BREW_REMOTE_PRESENT=1
+    else
+      OMD_UPDATE_BREW_REMOTE=""
+      OMD_UPDATE_BREW_REMOTE_PRESENT=0
+    fi
+  fi
+
+  core_prefix="${marker_prefix}/Library/Taps/homebrew/homebrew-core"
+  if [[ -d "${core_prefix}/.git" ]]; then
+    OMD_UPDATE_BREW_CORE_PREFIX="${core_prefix}"
+    if OMD_UPDATE_BREW_CORE_REMOTE="$(git -C "${core_prefix}" remote get-url origin 2>/dev/null)"; then
+      OMD_UPDATE_BREW_CORE_REMOTE_PRESENT=1
+    else
+      OMD_UPDATE_BREW_CORE_REMOTE=""
+      OMD_UPDATE_BREW_CORE_REMOTE_PRESENT=0
+    fi
   fi
 }
 
 omd_update_apply_brew_remote() {
-  local source="$1" target_remote
+  local source="$1" target_remote target_core_remote
   [[ -n "${OMD_UPDATE_BREW_PREFIX}" ]] || return 0
   target_remote="$(omd_source_brew_remote "${source}")" || return 1
   if [[ "${OMD_UPDATE_BREW_LEGACY}" == "1" ]]; then
     git -C "${OMD_UPDATE_BREW_PREFIX}" init -q || return 1
     OMD_UPDATE_BREW_REMOTE_CHANGED=1
     git -C "${OMD_UPDATE_BREW_PREFIX}" remote add origin "${target_remote}" || return 1
-    return 0
+  else
+    if [[ "${OMD_UPDATE_BREW_REMOTE_PRESENT}" == "1" ]]; then
+      if [[ "${OMD_UPDATE_BREW_REMOTE}" != "${target_remote}" ]]; then
+        git -C "${OMD_UPDATE_BREW_PREFIX}" remote set-url origin "${target_remote}" ||
+          return 1
+        OMD_UPDATE_BREW_REMOTE_CHANGED=1
+      fi
+    else
+      git -C "${OMD_UPDATE_BREW_PREFIX}" remote add origin "${target_remote}" || return 1
+      OMD_UPDATE_BREW_REMOTE_CHANGED=1
+    fi
   fi
-  if [[ "${OMD_UPDATE_BREW_REMOTE_PRESENT}" == "1" ]]; then
-    [[ "${OMD_UPDATE_BREW_REMOTE}" == "${target_remote}" ]] && return 0
-    git -C "${OMD_UPDATE_BREW_PREFIX}" remote set-url origin "${target_remote}" || return 1
+
+  [[ -n "${OMD_UPDATE_BREW_CORE_PREFIX}" ]] || return 0
+  target_core_remote="$(omd_source_brew_core_remote "${source}")" || return 1
+  if [[ "${OMD_UPDATE_BREW_CORE_REMOTE_PRESENT}" == "1" ]]; then
+    if [[ "${OMD_UPDATE_BREW_CORE_REMOTE}" != "${target_core_remote}" ]]; then
+      git -C "${OMD_UPDATE_BREW_CORE_PREFIX}" remote set-url origin "${target_core_remote}" ||
+        return 1
+      OMD_UPDATE_BREW_CORE_REMOTE_CHANGED=1
+    fi
   else
-    git -C "${OMD_UPDATE_BREW_PREFIX}" remote add origin "${target_remote}" || return 1
+    git -C "${OMD_UPDATE_BREW_CORE_PREFIX}" remote add origin "${target_core_remote}" ||
+      return 1
+    OMD_UPDATE_BREW_CORE_REMOTE_CHANGED=1
   fi
-  OMD_UPDATE_BREW_REMOTE_CHANGED=1
 }
 
 omd_update_restore_brew_remote() {
-  [[ "${OMD_UPDATE_BREW_REMOTE_CHANGED}" == "1" ]] || return 0
-  if [[ "${OMD_UPDATE_BREW_LEGACY}" == "1" ]]; then
-    rm -rf "${OMD_UPDATE_BREW_PREFIX}/.git"
-    return
+  local rollback_failed=0
+  if [[ "${OMD_UPDATE_BREW_CORE_REMOTE_CHANGED}" == "1" ]]; then
+    if [[ "${OMD_UPDATE_BREW_CORE_REMOTE_PRESENT}" == "1" ]]; then
+      git -C "${OMD_UPDATE_BREW_CORE_PREFIX}" remote set-url origin "${OMD_UPDATE_BREW_CORE_REMOTE}" ||
+        rollback_failed=1
+    else
+      git -C "${OMD_UPDATE_BREW_CORE_PREFIX}" remote remove origin ||
+        rollback_failed=1
+    fi
   fi
-  if [[ "${OMD_UPDATE_BREW_REMOTE_PRESENT}" == "1" ]]; then
-    git -C "${OMD_UPDATE_BREW_PREFIX}" remote set-url origin "${OMD_UPDATE_BREW_REMOTE}"
-  else
-    git -C "${OMD_UPDATE_BREW_PREFIX}" remote remove origin
+  if [[ "${OMD_UPDATE_BREW_REMOTE_CHANGED}" == "1" ]]; then
+    if [[ "${OMD_UPDATE_BREW_LEGACY}" == "1" ]]; then
+      rm -rf "${OMD_UPDATE_BREW_PREFIX}/.git" || rollback_failed=1
+    elif [[ "${OMD_UPDATE_BREW_REMOTE_PRESENT}" == "1" ]]; then
+      git -C "${OMD_UPDATE_BREW_PREFIX}" remote set-url origin "${OMD_UPDATE_BREW_REMOTE}" ||
+        rollback_failed=1
+    else
+      git -C "${OMD_UPDATE_BREW_PREFIX}" remote remove origin ||
+        rollback_failed=1
+    fi
   fi
+  return "${rollback_failed}"
 }
 
 omd_update_abort_with_rollback() {
   local message="$1" config_dir="$2" snapshot_dir="$3" rollback_failed=0
+  omd_source_config_restore "${snapshot_dir}" || rollback_failed=1
   omd_update_restore_brew_remote || rollback_failed=1
   omd_update_restore_source_config "${config_dir}" "${snapshot_dir}" || rollback_failed=1
   if [[ "${rollback_failed}" == "0" ]]; then
@@ -128,10 +183,45 @@ omd_update_abort_with_rollback() {
   omd_error "${message}; automatic rollback was incomplete; recovery snapshot: ${snapshot_dir}"
 }
 
+omd_update_current_source() {
+  local config_dir="$1" install_channel="$2" source
+  if [[ "${install_channel}" == "npm" ]]; then
+    source="${OHMYDEVPOD_NPM_SOURCE:-}"
+    if [[ -z "${source}" && -f "${config_dir}/npm-source" ]]; then
+      source="$(tr -d '[:space:]' < "${config_dir}/npm-source")"
+    fi
+    case "${source}" in
+      github|gitee)
+        printf '%s\n' "${source}"
+        return 0
+        ;;
+      *)
+        omd_warn "Invalid npm source '${source:-<empty>}'; falling back to GitHub"
+        printf 'github\n'
+        return 0
+        ;;
+    esac
+  fi
+  omd_read_saved_source "${config_dir}"
+}
+
+omd_update_persist_selected_source() {
+  local source="$1" config_dir="$2" configure_source="$3" install_channel="$4"
+  if [[ "${install_channel}" == "npm" ]]; then
+    omd_persist_source "${source}" "${config_dir}" 0 || return 1
+    printf '%s\n' "${source}" |
+      omd_atomic_write "${config_dir}/npm-source" ||
+      return 1
+    return 0
+  fi
+  omd_persist_source "${source}" "${config_dir}" "${configure_source}"
+}
+
 omd_update_main() {
   local requested_source="" current_source update_source current_version latest_tag latest_version
   local target prefix bin_dir cache_dir config_dir archive checksum archive_version installed_version
-  local snapshot_dir="" source_changed=0 configure_source=0 manage_source=0
+  local snapshot_dir="" source_changed=0 configure_source=0 manage_source=0 source_only=0
+  local install_channel="${OHMYDEVPOD_INSTALL_CHANNEL:-bootstrap}"
 
   case "$#" in
     0) ;;
@@ -148,21 +238,33 @@ omd_update_main() {
         *) omd_error "Unknown self-update option: $1" ;;
       esac
       ;;
+    2)
+      [[ "$1" == "--source-only" ]] ||
+        omd_error "Use only one of --github or --gitee"
+      source_only=1
+      configure_source=1
+      case "$2" in
+        --github) requested_source="github" ;;
+        --gitee) requested_source="gitee" ;;
+        *) omd_error "Source-only switching requires --github or --gitee" ;;
+      esac
+      ;;
     *) omd_error "Use only one of --github or --gitee" ;;
   esac
 
   omd_require_linux
-  omd_require_command curl
-  omd_require_command tar
-  omd_require_command sha256sum
+  if [[ "${source_only}" != "1" ]]; then
+    omd_require_command curl
+    omd_require_command tar
+    omd_require_command sha256sum
+  fi
 
-  target="$(omd_target_triple)"
   prefix="${OHMYDEVPOD_PREFIX:-${OMD_DEFAULT_PREFIX}}"
   bin_dir="${OHMYDEVPOD_BIN_DIR:-${OMD_DEFAULT_BIN_DIR}}"
   cache_dir="${OHMYDEVPOD_CACHE_DIR:-${OMD_DEFAULT_CACHE_DIR}}"
   config_dir="${OHMYDEVPOD_CONFIG_DIR:-${OMD_DEFAULT_CONFIG_DIR}}"
   current_version="${OHMYDEVPOD_CURRENT_VERSION:-$(tr -d '[:space:]' < "${bundle_root}/VERSION")}"
-  current_source="$(omd_read_saved_source "${config_dir}")"
+  current_source="$(omd_update_current_source "${config_dir}" "${install_channel}")"
   update_source="${requested_source:-${current_source}}"
   [[ "${current_source}" == "${update_source}" ]] || source_changed=1
   if [[ "${configure_source}" == "1" ]]; then
@@ -174,40 +276,47 @@ omd_update_main() {
 
   mkdir -p "${cache_dir}" "$(dirname "${config_dir}")"
 
-  if [[ -n "${OHMYDEVPOD_OMD_ARCHIVE:-}" ]]; then
-    archive="${OHMYDEVPOD_OMD_ARCHIVE}"
-    checksum="${OHMYDEVPOD_OMD_CHECKSUM:-${archive}.sha256}"
-    [[ -f "${checksum}" ]] ||
-      omd_error "Local archive override requires a checksum file: ${checksum}"
-    omd_verify_checksum "${archive}" "${checksum}" ||
-      omd_error "Self-update checksum verification failed; current installation and source were preserved"
-    latest_tag="$(omd_update_archive_version "${archive}" "${cache_dir}")" ||
-      omd_error "Self-update archive validation failed; current installation and source were preserved"
+  if [[ "${source_only}" == "1" ]]; then
+    archive=""
+    printf 'Current source: %s\n' "${current_source}"
+    printf 'New source: %s\n' "${update_source}"
   else
-    latest_tag="$(omd_resolve_version "${update_source}" "${OHMYDEVPOD_VERSION:-latest}")" ||
-      omd_error "Failed to query the latest ${update_source} release; current installation and source were preserved"
-  fi
-  latest_version="$(omd_update_normalize_version "${latest_tag}")"
+    target="$(omd_target_triple)"
+    if [[ -n "${OHMYDEVPOD_OMD_ARCHIVE:-}" ]]; then
+      archive="${OHMYDEVPOD_OMD_ARCHIVE}"
+      checksum="${OHMYDEVPOD_OMD_CHECKSUM:-${archive}.sha256}"
+      [[ -f "${checksum}" ]] ||
+        omd_error "Local archive override requires a checksum file: ${checksum}"
+      omd_verify_checksum "${archive}" "${checksum}" ||
+        omd_error "Self-update checksum verification failed; current installation and source were preserved"
+      latest_tag="$(omd_update_archive_version "${archive}" "${cache_dir}")" ||
+        omd_error "Self-update archive validation failed; current installation and source were preserved"
+    else
+      latest_tag="$(omd_resolve_version "${update_source}" "${OHMYDEVPOD_VERSION:-latest}")" ||
+        omd_error "Failed to query the latest ${update_source} release; current installation and source were preserved"
+    fi
+    latest_version="$(omd_update_normalize_version "${latest_tag}")"
 
-  printf 'Current version: %s\n' "${current_version}"
-  printf 'Current source: %s\n' "${current_source}"
-  printf 'Update source: %s\n' "${update_source}"
-  printf 'Latest version: %s\n' "${latest_version}"
+    printf 'Current version: %s\n' "${current_version}"
+    printf 'Current source: %s\n' "${current_source}"
+    printf 'Update source: %s\n' "${update_source}"
+    printf 'Latest version: %s\n' "${latest_version}"
 
-  if [[ "$(omd_update_normalize_version "${current_version}")" != "${latest_version}" ]]; then
-    if [[ -z "${OHMYDEVPOD_OMD_ARCHIVE:-}" ]]; then
-      OMD_DOWNLOADED_ARCHIVE=""
-      OMD_SELECTED_SOURCE=""
-      omd_download_release "${update_source}" "${latest_tag}" "${target}" "${cache_dir}" ||
-        omd_error "Self-update download or checksum verification failed; current installation and source were preserved"
-      archive="${OMD_DOWNLOADED_ARCHIVE}"
-      archive_version="$(omd_update_archive_version "${archive}" "${cache_dir}")" ||
-        omd_error "Self-update archive validation failed; current installation and source were preserved"
-      [[ "$(omd_update_normalize_version "${archive_version}")" == "${latest_version}" ]] ||
-        omd_error "Release archive version ${archive_version} does not match ${latest_tag}; current installation and source were preserved"
+    if [[ "$(omd_update_normalize_version "${current_version}")" != "${latest_version}" ]]; then
+      if [[ -z "${OHMYDEVPOD_OMD_ARCHIVE:-}" ]]; then
+        OMD_DOWNLOADED_ARCHIVE=""
+        OMD_SELECTED_SOURCE=""
+        omd_download_release "${update_source}" "${latest_tag}" "${target}" "${cache_dir}" ||
+          omd_error "Self-update download or checksum verification failed; current installation and source were preserved"
+        archive="${OMD_DOWNLOADED_ARCHIVE}"
+        archive_version="$(omd_update_archive_version "${archive}" "${cache_dir}")" ||
+          omd_error "Self-update archive validation failed; current installation and source were preserved"
+        [[ "$(omd_update_normalize_version "${archive_version}")" == "${latest_version}" ]] ||
+          omd_error "Release archive version ${archive_version} does not match ${latest_tag}; current installation and source were preserved"
+      fi
+    else
+      archive=""
     fi
-  else
-    archive=""
   fi
 
   if [[ -n "${archive}" ]]; then
@@ -228,11 +337,23 @@ omd_update_main() {
       rm -rf "${snapshot_dir}"
       omd_error "Could not snapshot source configuration"
     }
+    omd_source_config_validate "${update_source}" || {
+      rm -rf "${snapshot_dir}"
+      omd_error "Installed managed software has user-modified source configuration; current source was preserved"
+    }
+    omd_source_config_snapshot "${snapshot_dir}" || {
+      rm -rf "${snapshot_dir}"
+      omd_error "Could not snapshot installed software source configuration"
+    }
     omd_update_capture_brew_remote || {
       rm -rf "${snapshot_dir}"
       omd_error "Could not inspect the managed Homebrew remote"
     }
-    if ! omd_persist_source "${update_source}" "${config_dir}" "${configure_source}"; then
+    if ! omd_update_persist_selected_source \
+      "${update_source}" \
+      "${config_dir}" \
+      "${configure_source}" \
+      "${install_channel}"; then
       omd_update_abort_with_rollback "Could not update source configuration" "${config_dir}" "${snapshot_dir}"
     fi
     if ! omd_persist_install_paths "${prefix}" "${bin_dir}" "${cache_dir}" "${config_dir}"; then
@@ -241,6 +362,15 @@ omd_update_main() {
     if ! omd_update_apply_brew_remote "${update_source}"; then
       omd_update_abort_with_rollback "Could not switch the managed Homebrew remote" "${config_dir}" "${snapshot_dir}"
     fi
+    if ! omd_source_config_apply "${update_source}"; then
+      omd_update_abort_with_rollback "Could not switch installed software source configuration" "${config_dir}" "${snapshot_dir}"
+    fi
+  fi
+
+  if [[ "${source_only}" == "1" ]]; then
+    [[ -z "${snapshot_dir}" ]] || rm -rf "${snapshot_dir}"
+    printf 'Source configuration updated successfully.\n'
+    return 0
   fi
 
   if [[ -n "${archive}" ]]; then

package/runtime/modules/core/micromamba.sh

@@ -2,6 +2,8 @@
 set -euo pipefail
 
 source "$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)/lib/common.sh"
+# shellcheck source=../lib/source-config.sh
+source "$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)/lib/source-config.sh"
 
 component="micromamba"
 formula="micromamba"
@@ -14,7 +16,15 @@ install_or_update() {
   shift
   omd_module_formula_install_or_update "${component}" "${formula}" "${command_name}" "${action}" "$@"
 }
-uninstall() { omd_module_formula_uninstall "${component}" "${formula}" "$@"; }
+uninstall() {
+  if omd_module_dry_run "$@"; then
+    omd_module_formula_uninstall "${component}" "${formula}" "$@"
+    return
+  fi
+  omd_module_formula_uninstall_keep_marker "${component}" "${formula}" "$@" || return
+  omd_source_config_remove_component "${component}" || return
+  omd_module_unmark_managed "${component}"
+}
 
 case "${1:-}" in
   status) status ;;

package/runtime/modules/core/uv.sh

@@ -2,6 +2,8 @@
 set -euo pipefail
 
 source "$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)/lib/common.sh"
+# shellcheck source=../lib/source-config.sh
+source "$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)/lib/source-config.sh"
 
 component="uv"
 formula="uv"
@@ -14,7 +16,15 @@ install_or_update() {
   shift
   omd_module_formula_install_or_update "${component}" "${formula}" "${command_name}" "${action}" "$@"
 }
-uninstall() { omd_module_formula_uninstall "${component}" "${formula}" "$@"; }
+uninstall() {
+  if omd_module_dry_run "$@"; then
+    omd_module_formula_uninstall "${component}" "${formula}" "$@"
+    return
+  fi
+  omd_module_formula_uninstall_keep_marker "${component}" "${formula}" "$@" || return
+  omd_source_config_remove_component "${component}" || return
+  omd_module_unmark_managed "${component}"
+}
 
 case "${1:-}" in
   status) status ;;

package/runtime/modules/lib/common.sh

@@ -188,9 +188,8 @@ omd_module_brew_cmd() {
   return 1
 }
 
-omd_module_brew_formula_installed() {
-  local formula="$1" formula_name prefix cellar version
-  prefix="$(omd_module_brew_prefix)" || return 1
+omd_module_brew_formula_installed_at() {
+  local prefix="$1" formula="$2" formula_name cellar version
   formula_name="${formula##*/}"
   cellar="${prefix}/Cellar/${formula_name}"
   [[ -d "${cellar}" ]] || return 1
@@ -200,16 +199,40 @@ omd_module_brew_formula_installed() {
   return 1
 }
 
+omd_module_brew_formula_installed() {
+  local formula="$1" prefix
+  prefix="$(omd_module_brew_prefix)" || return 1
+  omd_module_brew_formula_installed_at "${prefix}" "${formula}"
+}
+
+omd_module_brew_prefix_for_cmd() {
+  local brew="$1" prefix
+  prefix="$(omd_module_brew_exec "${brew}" --prefix 2>/dev/null || true)"
+  if [[ "${prefix}" == /* && -d "${prefix}" ]]; then
+    cd "${prefix}" 2>/dev/null && pwd -P
+    return
+  fi
+  [[ "${brew}" == */bin/brew ]] || return 1
+  cd "$(dirname "${brew}")/.." 2>/dev/null && pwd -P
+}
+
+omd_module_brew_cmd_matches_prefix() {
+  local brew="$1" expected_prefix="$2" resolved_brew reported_prefix
+  expected_prefix="$(cd "${expected_prefix}" 2>/dev/null && pwd -P)" || return 1
+  if [[ -L "${brew}" ]]; then
+    command -v readlink >/dev/null 2>&1 || return 1
+    resolved_brew="$(readlink -f "${brew}" 2>/dev/null || true)"
+    [[ "${resolved_brew}" == "${expected_prefix}"/* ]] || return 1
+  fi
+  reported_prefix="$(omd_module_brew_prefix_for_cmd "${brew}")" || return 1
+  [[ "${reported_prefix}" == "${expected_prefix}" ]]
+}
+
 omd_module_brew_prefix() {
-  local brew resolved
+  local brew
   brew="$(omd_module_brew_cmd)" || return 1
-  if command -v readlink >/dev/null 2>&1; then
-    resolved="$(readlink -f "${brew}" 2>/dev/null || true)"
-    [[ -z "${resolved}" ]] || brew="${resolved}"
-  fi
-  if [[ "${brew}" == */bin/brew ]]; then
-    cd "$(dirname "${brew}")/.." 2>/dev/null && pwd -P
-    return
+  if omd_module_brew_prefix_for_cmd "${brew}"; then
+    return 0
   fi
   if [[ -n "${HOMEBREW_PREFIX:-}" && -d "${HOMEBREW_PREFIX}" ]]; then
     printf '%s\n' "${HOMEBREW_PREFIX}"
@@ -231,9 +254,10 @@ omd_module_brew_exec() {
 }
 
 omd_module_formula_status() {
-  local formula="$1" command_name="$2"
+  local formula="$1" command_name="$2" prefix
   if [[ -n "${component:-}" ]] && omd_module_formula_managed "${component}" "${formula}"; then
-    omd_module_brew_formula_installed "${formula}"
+    prefix="$(omd_module_formula_marker_prefix "${component}" "${formula}")" || return 1
+    omd_module_brew_formula_installed_at "${prefix}" "${formula}"
     return
   fi
   command -v "${command_name}" >/dev/null 2>&1 ||
@@ -241,12 +265,40 @@ omd_module_formula_status() {
 }
 
 omd_module_formula_managed() {
-  local component="$1" formula="$2"
-  omd_module_marker_matches "${component}" "${formula}"
+  local component="$1" formula="$2" kind
+  omd_module_marker_matches "${component}" "${formula}" || return 1
+  kind="$(omd_module_marker_value "${component}" kind || true)"
+  [[ "${kind}" == "brew-formula" ]]
+}
+
+omd_module_formula_marker_prefix() {
+  local component="$1" formula="$2" prefix linuxbrew_kind
+  omd_module_formula_managed "${component}" "${formula}" || return 1
+  prefix="$(omd_module_marker_value "${component}" brew_prefix || true)"
+  if [[ -z "${prefix}" ]] && omd_module_is_managed linuxbrew; then
+    linuxbrew_kind="$(omd_module_marker_value linuxbrew kind || true)"
+    if [[ "${linuxbrew_kind}" == "directory" ]]; then
+      prefix="$(omd_module_marker_value linuxbrew artifact || true)"
+      if ! omd_module_brew_formula_installed_at "${prefix}" "${formula}"; then
+        prefix=""
+      fi
+    fi
+  fi
+  [[ -n "${prefix}" && "${prefix}" == /* && ! -L "${prefix}" && -x "${prefix}/bin/brew" ]] ||
+    return 1
+  prefix="$(cd "${prefix}" 2>/dev/null && pwd -P)" || return 1
+  omd_module_brew_cmd_matches_prefix "${prefix}/bin/brew" "${prefix}" || return 1
+  printf '%s\n' "${prefix}"
+}
+
+omd_module_formula_brew_cmd() {
+  local component="$1" formula="$2" prefix
+  prefix="$(omd_module_formula_marker_prefix "${component}" "${formula}")" || return 1
+  printf '%s/bin/brew\n' "${prefix}"
 }
 
 omd_module_formula_install_or_update() {
-  local component="$1" formula="$2" command_name="$3" action="$4"
+  local component="$1" formula="$2" command_name="$3" action="$4" brew brew_prefix
   shift 4
   omd_module_reject_unknown_flags "$@" || return
 
@@ -261,24 +313,38 @@ omd_module_formula_install_or_update() {
     return 0
   fi
 
-  local brew
-  brew="$(omd_module_brew_cmd)" || {
-    omd_module_info error "Linuxbrew is required before ${component}"
+  if omd_module_formula_managed "${component}" "${formula}"; then
+    brew="$(omd_module_formula_brew_cmd "${component}" "${formula}")" || {
+      omd_module_info error "managed ${component} has no valid owning Homebrew prefix"
+      return 1
+    }
+  else
+    brew="$(omd_module_brew_cmd)" || {
+      omd_module_info error "Linuxbrew is required before ${component}"
+      return 1
+    }
+  fi
+  brew_prefix="$(omd_module_brew_prefix_for_cmd "${brew}")" || {
+    omd_module_info error "could not determine Homebrew prefix for ${component}"
     return 1
   }
 
   if [[ "${action}" == "update" ]] &&
-    omd_module_brew_exec "${brew}" list --formula "${formula}" >/dev/null 2>&1; then
+    omd_module_brew_formula_installed_at "${brew_prefix}" "${formula}"; then
     omd_module_brew_exec "${brew}" upgrade "${formula}"
   else
     omd_module_brew_exec "${brew}" install "${formula}"
   fi
-  omd_module_mark_managed "${component}" brew-formula "${formula}"
+  omd_module_mark_managed \
+    "${component}" \
+    brew-formula \
+    "${formula}" \
+    "brew_prefix=${brew_prefix}"
 }
 
-omd_module_formula_uninstall() {
-  local component="$1" formula="$2"
-  shift 2
+omd_module_formula_uninstall_impl() {
+  local component="$1" formula="$2" remove_marker="$3"
+  shift 3
   omd_module_reject_unknown_flags "$@" || return
 
   if omd_module_dry_run "$@"; then
@@ -289,11 +355,18 @@ omd_module_formula_uninstall() {
   omd_module_formula_managed "${component}" "${formula}" ||
     omd_module_refuse_unmanaged "${component}" || return
 
-  local brew dependants
-  brew="$(omd_module_brew_cmd)" || {
-    omd_module_info error "Linuxbrew is required to uninstall ${component}"
+  local brew brew_prefix dependants
+  brew="$(omd_module_formula_brew_cmd "${component}" "${formula}")" || {
+    omd_module_info error "managed ${component} has no valid owning Homebrew prefix"
     return 1
   }
+  brew_prefix="$(omd_module_brew_prefix_for_cmd "${brew}")" || return 1
+  if ! omd_module_brew_formula_installed_at "${brew_prefix}" "${formula}"; then
+    if [[ "${remove_marker}" == "1" ]]; then
+      omd_module_unmark_managed "${component}"
+    fi
+    return 0
+  fi
   if ! dependants="$(omd_module_brew_exec "${brew}" uses --installed "${formula}" 2>&1)"; then
     omd_module_info error "failed to inspect Homebrew dependants for ${component}: ${dependants}"
     return 1
@@ -303,7 +376,21 @@ omd_module_formula_uninstall() {
     return 1
   fi
   omd_module_brew_exec "${brew}" uninstall "${formula}"
-  omd_module_unmark_managed "${component}"
+  if [[ "${remove_marker}" == "1" ]]; then
+    omd_module_unmark_managed "${component}"
+  fi
+}
+
+omd_module_formula_uninstall() {
+  local component="$1" formula="$2"
+  shift 2
+  omd_module_formula_uninstall_impl "${component}" "${formula}" 1 "$@"
+}
+
+omd_module_formula_uninstall_keep_marker() {
+  local component="$1" formula="$2"
+  shift 2
+  omd_module_formula_uninstall_impl "${component}" "${formula}" 0 "$@"
 }
 
 omd_module_zsh_path() {

package/runtime/modules/lib/postflight.sh

@@ -2,12 +2,28 @@
 set -euo pipefail
 
 source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
+# shellcheck source=source-config.sh
+source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/source-config.sh"
 
 case "${1:-}" in
   install|update) ;;
   *) omd_module_unknown_action "${1:-}" ;;
 esac
 
+case "${OHMYDEVPOD_MIRROR_PROFILE:-upstream}" in
+  cn) source_name="gitee" ;;
+  upstream) source_name="github" ;;
+  *)
+    omd_module_info notice "unknown mirror profile; native source configuration was not changed"
+    source_name=""
+    ;;
+esac
+if [[ -n "${source_name}" ]] &&
+  ! omd_source_config_apply_transactional "${source_name}"; then
+  omd_module_info error "failed to apply native source configuration"
+  exit 1
+fi
+
 if ! omd_module_is_managed zsh && ! omd_module_is_managed zsh-config; then
   exit 0
 fi