修复cargo换源

原来使用了sed和grep导致在win上无法运行
现在直接手动解析str
This commit is contained in:
Mikachu2333
2025-10-01 15:41:44 +08:00
committed by 曾奥然
parent dec55705c7
commit 0ff604060d

View File

@@ -48,55 +48,80 @@ pl_rust_cargo_prelude (void)
} }
void
rust_cargo_note_get_src_default ()
{
if (ENGLISH)
chsrc_note2 ("No source configured in Cargo, showing default upstream source:");
else
chsrc_note2 ("Cargo 中未自定义源,显示默认源:");
Source_t default_source = chsrc_yield_source (&pl_rust_cargo_target, "upstream");
say (default_source.url);
}
void
rust_cargo_note_get_src_mirror (char *url, bool sparse)
{
if (ENGLISH)
{
chsrc_note2 ("Custom source found:");
}
else
{
chsrc_note2 ("已找到自定义源:");
}
say (xy_2strcat (url, sparse ? " (sparse)" : ""));
}
void void
pl_rust_cargo_getsrc (char *option) pl_rust_cargo_getsrc (char *option)
{ {
char *cargo_config_file = xy_normalize_path ("~/.cargo/config.toml"); char *cargo_config_file = xy_normalize_path ("~/.cargo/config.toml");
if (xy_file_exist (cargo_config_file)) char *raw_content = xy_file_to_str (cargo_config_file);
{ char *formatted_content = xy_str_gsub (raw_content, " ", "");
// 尝试提取 [source.mirror] 下的 registry URL formatted_content = xy_str_gsub (raw_content, "\"", "");
char *grep_cmd = xy_str_gsub ("grep -A1 '\\[source\\.mirror\\]' '@f@' | grep 'registry' | sed 's/[^\"]*\"\\([^\"]*\\)\".*/\\1/'", "@f@", cargo_config_file); free (raw_content);
chsrc_ensure_program ("grep");
chsrc_ensure_program ("sed");
char *mirror_url; XyStrFindResult_t result_has_mirror = xy_str_find (formatted_content, "replace-with");
int status = xy_run_get_stdout (grep_cmd, &mirror_url); if (result_has_mirror.found)
char *stripped_url = (mirror_url) ? xy_str_strip(mirror_url) : "";
if (0 == status && stripped_url && strstr(stripped_url, "http"))
{ {
// 找到配置的镜像源,如果存在 sparse+ 前缀则去除 char *mirror_name = xy_str_take_until_newline (formatted_content + result_has_mirror.end + 1);
char *clean_url = (strstr(stripped_url, "sparse+")) ? mirror_name = xy_str_delete_prefix (mirror_name, "=\"");
stripped_url + 7 : stripped_url; mirror_name = xy_str_delete_suffix (mirror_name, "\"");
say (clean_url);
XyStrFindResult_t result_mirror = xy_str_find (formatted_content, xy_strcat (3, "[source.", mirror_name, "]"));
if (!result_mirror.found)
{
rust_cargo_note_get_src_default();
return;
} }
else char *mirror_url = xy_str_take_until_newline (formatted_content + result_mirror.end + 1);
mirror_url = xy_str_delete_prefix (mirror_url, "registry=\"");
mirror_url = xy_str_delete_suffix (mirror_url, "\"");
if (xy_str_find (mirror_url, "sparse+").found)
{ {
// 配置文件存在但没有找到镜像源配置,显示默认上游源 rust_cargo_note_get_src_mirror (xy_str_delete_prefix (mirror_url, "sparse+"), true);
if (ENGLISH)
chsrc_note2 ("Config file exists but no mirror source found, showing default upstream source:");
else
chsrc_note2 ("配置文件存在但未找到镜像源配置,显示默认上游源:");
Source_t default_source = chsrc_yield_source (&pl_rust_cargo_target, "upstream");
say (default_source.url);
} }
} }
else else
{ {
// 配置文件不存在,显示默认上游源 rust_cargo_note_get_src_default();
if (ENGLISH)
chsrc_note2 ("No source configured in Cargo, showing default upstream source:");
else
chsrc_note2 ("Cargo 中未配置源,显示默认上游源:");
Source_t default_source = chsrc_yield_source (&pl_rust_cargo_target, "upstream");
say (default_source.url);
} }
} }
void
write_rust_config (const char *path, const char *url)
{
remove (path);
char *content = RAWSTR_pl_rust_cargo_config;
content = xy_str_gsub (content, "@url@", url);
chsrc_overwrite_file (content, path);
free (content);
}
/** /**
* @consult https://mirrors.tuna.tsinghua.edu.cn/help/crates.io-index/ * @consult https://mirrors.tuna.tsinghua.edu.cn/help/crates.io-index/
* @consult https://help.mirrors.cernet.edu.cn/crates.io-index * @consult https://help.mirrors.cernet.edu.cn/crates.io-index
@@ -108,6 +133,7 @@ pl_rust_cargo_setsrc (char *option)
chsrc_use_this_source (pl_rust_cargo); chsrc_use_this_source (pl_rust_cargo);
char *default_content = RAWSTR_pl_rust_cargo_config;
char *cargo_config_dir = "~/.cargo/"; char *cargo_config_dir = "~/.cargo/";
char *cargo_config_file = xy_2strcat (cargo_config_dir, "config.toml"); char *cargo_config_file = xy_2strcat (cargo_config_dir, "config.toml");
@@ -118,49 +144,51 @@ pl_rust_cargo_setsrc (char *option)
if (xy_file_exist (cargo_config_file)) if (xy_file_exist (cargo_config_file))
{ {
chsrc_backup (cargo_config_file); chsrc_backup (cargo_config_file);
char *raw_content = xy_file_to_str (cargo_config_file);
char *formatted_content = xy_str_gsub (raw_content, " ", "");
formatted_content = xy_str_gsub (raw_content, "\"", "");
XyStrFindResult_t result_has_mirror = xy_str_find (raw_content, "replace-with");
if (!result_has_mirror.found)
{
write_rust_config (cargo_config_file, source.url);
goto finish;
} }
char *content = RAWSTR_pl_rust_cargo_config; char *mirror_name = xy_str_take_until_newline (raw_content + result_has_mirror.end + 1);
content = xy_str_gsub (content, "@url@", source.url); mirror_name = xy_str_gsub (mirror_name, " ", "");
mirror_name = xy_str_gsub (mirror_name, "'", "\"");
mirror_name = xy_str_delete_prefix (mirror_name, "=\"");
mirror_name = xy_str_delete_suffix (mirror_name, "\"");
if (xy_file_exist (cargo_config_file)) XyStrFindResult_t result_mirror = xy_str_find (raw_content, xy_strcat (3, "[source.", mirror_name, "]"));
if (!result_mirror.found)
{ {
char *check_cmd = xy_str_gsub (RAWSTR_pl_rust_cargo_check_config, "@f@", cargo_config_file); write_rust_config (cargo_config_file, source.url);
chsrc_ensure_program ("grep"); goto finish;
int status = chsrc_run_directly (check_cmd);
if (0 == status)
{
char *sed_cmd;
if (xy.on_bsd || xy.on_macos)
{
sed_cmd = "sed -i '' ";
}
else
{
sed_cmd = "sed -i ";
} }
char *update_cmd = xy_str_gsub (RAWSTR_pl_rust_cargo_update_replace_with, "@sed@", sed_cmd); char *mirror_url = xy_str_take_until_newline (raw_content + result_mirror.end + 1);
update_cmd = xy_str_gsub (update_cmd, "@f@", cargo_config_file); mirror_url = xy_str_gsub (mirror_url, " ", "");
chsrc_run (update_cmd, RunOpt_Default); if (!xy_str_find (mirror_url, "registry").found)
{
write_rust_config (cargo_config_file, source.url);
goto finish;
}
mirror_url = xy_str_delete_prefix (mirror_url, "registry=\"");
mirror_url = xy_str_delete_suffix (mirror_url, "\"");
update_cmd = xy_str_gsub (RAWSTR_pl_rust_cargo_update_registry, "@sed@", sed_cmd); char *final_content = xy_str_gsub (raw_content, mirror_url, xy_2strcat ("sparse+", source.url));
update_cmd = xy_str_gsub (update_cmd, "@f@", cargo_config_file); chsrc_overwrite_file (final_content, cargo_config_file);
update_cmd = xy_str_gsub (update_cmd, "@url@", source.url); free (final_content);
chsrc_run (update_cmd, RunOpt_Default); goto finish;
}
else
{
chsrc_append_to_file ("\n", cargo_config_file);
chsrc_append_to_file (content, cargo_config_file);
}
}
else
{
chsrc_append_to_file (content, cargo_config_file);
} }
write_rust_config (cargo_config_file, source.url);
goto finish;
finish:
chsrc_determine_chgtype (ChgType_Auto); chsrc_determine_chgtype (ChgType_Auto);
chsrc_conclude (&source); chsrc_conclude (&source);
} }