chsrc/src/recipe/lang/Python/uv.c

/** ------------------------------------------------------------
 * SPDX-License-Identifier: GPL-3.0-or-later
 * ------------------------------------------------------------*/

def_target(pl_python_uv, "uv");

void
pl_python_uv_prelude (void)
{
  use_this(pl_python_uv);
  chef_allow_gsr(pl_python_uv);

  chef_set_created_on   (this, "2024-12-11");
  chef_set_last_updated (this, "2025-08-09");
  chef_set_sources_last_updated (this, "2025-08-09");

  chef_set_authors (this, 1, "happy game", "happygame1024@gmail.com");
  chef_set_chef (this, NULL, NULL);
  chef_set_cooks (this, 0);
  chef_set_contributors (this, 2,
    "ccy", "icuichengyi@gmail.com",
    "Aoran Zeng", "ccmywish@qq.com");

  chef_allow_local_mode (this, FullyCan, NULL, NULL);
  chef_allow_english(this);
  chef_allow_user_define(this);

  // 使用 pl_python_group 的源
  this->sources = pl_python_group_target.sources;
  this->sources_n = pl_python_group_target.sources_n;
}


/**
 * chsrc get uv
 *
 * uv的配置优先级顺序如下(高到低):
 * 1. $workspaces/uv.toml
 * 2. $workspaces/pyproject.toml
 * 3. ~/.config/uv/uv.toml
 * 4. /etc/uv/uv.toml
 */

#define PL_Python_uv_ConfigFile       "uv.toml"
#define PL_Python_uv_Local_ConfigPath "./"
#define PL_Python_uv_User_ConfigPath  "~/.config/uv/"


char *
pl_python_find_uv_config (bool mkdir)
{
  if (chsrc_in_local_mode())
    {
      return xy_2strjoin (PL_Python_uv_Local_ConfigPath, PL_Python_uv_ConfigFile);
    }
  else
    {
      if (xy_on_windows)
        {
          /* config path on Windows */
          char *appdata = getenv ("APPDATA");

          if (!appdata)
            {
              chsrc_error2 ("未能获取 APPDATA 环境变量");
              return NULL;
            }

          char *config_dir = xy_2strjoin(appdata, "\\uv\\");
          if (mkdir)
            {
              chsrc_ensure_dir (config_dir);
            }
          return xy_2strjoin (config_dir, PL_Python_uv_ConfigFile);
        }
      else
        {
          /* config path on Linux or macOS */
          if (mkdir)
            {
              chsrc_ensure_dir (PL_Python_uv_User_ConfigPath);
            }
          return xy_2strjoin (PL_Python_uv_User_ConfigPath, PL_Python_uv_ConfigFile);
        }
    }
}

void
pl_python_uv_getsrc (char *option)
{
  char *uv_config = pl_python_find_uv_config (false);

  if (!chsrc_check_file (uv_config))
    {
      chsrc_error2 ("未找到 uv 配置文件");
      return;
    }

  /* 获取 [[index]] 配置项的 url */
  char *cmd = xy_str_gsub (RAWSTR_pl_python_get_uv_config, "@f@", uv_config);
  chsrc_run (cmd, RunOpt_Default);
}


/**
 * @consult https://docs.astral.sh/uv/configuration/files/
 *          https://github.com/RubyMetric/chsrc/issues/139
 */
void
pl_python_uv_setsrc (char *option)
{
  chsrc_ensure_program ("uv");

  Source_t source = chsrc_yield_source (&pl_python_group_target, option);
  if (chsrc_in_standalone_mode())
    chsrc_confirm_source(&source);

  char *uv_config = pl_python_find_uv_config (true);
  if (NULL==uv_config)
    {
      chsrc_error2 ("无法获取 uv 配置文件路径");
      return;
    }
  chsrc_backup (uv_config);

  const char *source_content = xy_str_gsub (RAWSTR_pl_python_uv_config_source_content, "@url@", source.url);

#if defined(XY_On_macOS) || defined(XY_On_BSD)
  char *sed_cmd = "sed -i '' ";
#else
  char *sed_cmd = "sed -i ";
#endif

  /**
   * 将 [[index]] 到 default = true 之间的 url = ".*" 替换为 url = "source.url"
   */
  char *update_config_cmd = xy_str_gsub (RAWSTR_pl_python_set_uv_config, "@sed@", sed_cmd);
        update_config_cmd = xy_str_gsub (update_config_cmd, "@f@", uv_config);
        update_config_cmd = xy_str_gsub (update_config_cmd, "@url@", source.url);

  if (!xy_file_exist (uv_config))
    {
      /* 当 uv_config 不存在，直接写入文件 */
      chsrc_append_to_file (source_content, uv_config);
    }
  else
    {
      /* 当 uv_config 存在，如果存在 [[index]] 则更新，否则追加到文件末尾 */
      char *cmd = xy_str_gsub (RAWSTR_pl_python_test_uv_if_set_source, "@f@", uv_config);
      chsrc_ensure_program ("grep");
      int status = system (cmd);
      if (0==status)
        {
          chsrc_run (update_config_cmd, RunOpt_Default);
        }
      else
        {
          chsrc_append_to_file (source_content, uv_config);
        }
    }

  if (chsrc_in_standalone_mode())
    {
      chsrc_determine_chgtype (ChgType_Auto);
      chsrc_conclude (&source);
    }
}


void
pl_python_uv_resetsrc (char *option)
{
  pl_python_uv_setsrc (option);
}