/* -------------------------------------------------------------- * File : chsrc.c * Authors : Aoran Zeng * Created on : <2023-08-28> * Last modified : <2023-09-01> * * chsrc: * * Change Source —— 命令行换源工具 * -------------------------------------------------------------*/ #include "chsrc.h" #define Chsrc_Version "v0.1.0.20230910.pre" /** * 检测二进制程序是否存在 * * @param[in] check_cmd 检测 `progname` 是否存在的一段命令,一般来说,填 `progname` 本身即可, * 但是某些情况下,需要使用其他命令绕过一些特殊情况,比如 python 这个命令在Windows上 * 会自动打开 Microsoft Store,需避免 * * @param[in] progname 要检测的二进制程序名 */ bool does_the_program_exist (char* check_cmd, char* progname) { char* which = check_cmd; int ret = system(which); char buf[32] = {0}; sprintf(buf, "错误码: %d", ret); if (0!=ret) { xy_warn (xy_strjoin(4, "× 命令 ", progname, " 不存在,", buf)); return false; } else { xy_success (xy_strjoin(3, "√ 命令 ", progname, " 存在")); return true; } } /** * Perl换源 * * 参考:https://help.mirrors.cernet.edu.cn/CPAN/ */ void pl_perl_setsrc (char* option) { int selected = 0; char* check_cmd, *prog = NULL; if (xy_on_windows) check_cmd = "perl --version >nul 2>nul"; else check_cmd = "perl --version 1>/dev/null 2>&1"; bool exist_b = does_the_program_exist (check_cmd, "perl"); if (!exist_b) { xy_error ("chsrc: 未找到 perl 相关命令,请检查是否存在"); return; } for (int i=0;iname; const char* source_abbr = pl_perl_sources[selected].mirror->abbr; const char* source_url = pl_perl_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); char* cmd = xy_strjoin(3, "perl -MCPAN -e 'CPAN::HandleConfig->edit(\"pushy_https\", 0); CPAN::HandleConfig->edit(\"urllist\", \"unshift\", \"", source_url, "\"); mkmyconfig'"); system(cmd); free(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * NodeJS换源 * * 参考:https://npmmirror.com/ */ void pl_nodejs_setsrc (char* option) { int selected = 0; char* check_cmd, *prog = NULL; if (xy_on_windows) check_cmd = "npm -v >nul 2>nul"; else check_cmd = "npm -v 1>/dev/null 2>&1"; bool exist_b = does_the_program_exist (check_cmd, "npm"); if (!exist_b) { xy_error ("chsrc: 未找到 npm 相关命令,请检查是否存在"); return; } for (int i=0;iname; const char* source_abbr = pl_nodejs_sources[selected].mirror->abbr; const char* source_url = pl_nodejs_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); char* cmd = xy_2strjoin("npm config set registry ", source_url); system(cmd); free(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * Python换源 * * 参考:https://mirrors.tuna.tsinghua.edu.cn/help/pypi/ * * 经测试,Windows上调用换源命令,会写入 C:\Users\RubyMetric\AppData\Roaming\pip\pip.ini */ void pl_python_setsrc (char* option) { int selected = 0; char* check_cmd, *prog = NULL; // 不要调用 python 自己,而是使用 python --version,避免Windows弹出Microsoft Store if (xy_on_windows) check_cmd = "python --version >nul 2>nul"; else check_cmd = "python --version 1>/dev/null 2>&1"; bool exist_b = does_the_program_exist (check_cmd, "python"); if (!exist_b) { if (xy_on_windows) check_cmd = "python3 --version >nul 2>nul"; else check_cmd = "python3 --version 1>/dev/null 2>&1"; exist_b = does_the_program_exist (check_cmd, "python3"); if (exist_b) prog = "python3"; } else { prog = "python"; } if (!exist_b) { xy_error ("chsrc: 未找到 Python 相关命令,请检查是否存在"); return; } for (int i=0;iname; const char* source_abbr = pl_python_sources[selected].mirror->abbr; const char* source_url = pl_python_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); char* cmd = xy_2strjoin(prog, xy_2strjoin(" -m pip config set global.index-url ", source_url)); system(cmd); free(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * Ruby换源 * * 参考:https://gitee.com/RubyKids/rbenv-cn */ void pl_ruby_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; for (int i=0;iname; const char* source_abbr = pl_ruby_sources[selected].mirror->abbr; const char* source_url = pl_ruby_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); xy_info("chsrc: 为 gem 命令换源"); system("gem source -r https://rubygems.org/"); char* cmd = xy_2strjoin("gem source -a ", source_url); system(cmd); free(cmd); if (xy_on_windows) check_cmd = "bundle -v >nul 2>nul"; else check_cmd = "bundle -v 1>/dev/null 2>&1"; exist_b = does_the_program_exist (check_cmd, "bundle"); if (!exist_b) { xy_error ("chsrc: 未找到 bundle 相关命令,请检查是否存在"); return; } cmd = xy_2strjoin("bundle config 'mirror.https://rubygems.org' ", source_url); xy_info("chsrc: 为 bundler 命令换源"); system(cmd); free(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * Go换源 * * 参考:https://goproxy.cn/ */ void pl_go_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; if (xy_on_windows) check_cmd = "go --version >nul 2>nul"; else check_cmd = "go --version 1>/dev/null 2>&1"; bool exist_b = does_the_program_exist (check_cmd, "go"); if (!exist_b) { xy_error ("chsrc: 未找到 go 相关命令,请检查是否存在"); return; } for (int i=0;iname; const char* source_abbr = pl_go_sources[selected].mirror->abbr; const char* source_url = pl_go_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); char* cmd = "go env -w GO111MODULE=on"; system(cmd); cmd = xy_strjoin(3, "go env -w GOPROXY=", source_url, ",direct"); system(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * Rust 换源 * * 参考:https://help.mirrors.cernet.edu.cn/crates.io-index.git */ void pl_rust_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; for (int i=0;iname; const char* source_abbr = pl_rust_sources[selected].mirror->abbr; const char* source_url = pl_rust_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); const char* file = xy_strjoin (3, "[source.crates-io]\n" "replace-with = 'mirror'\n\n" "[source.mirror]\n" "registry = \"", source_url, "\""); char* cmd = NULL; if (xy_on_windows) cmd = xy_strjoin(3, "echo ", file, ">> \%USERPROFILE%\\.cargo"); else cmd = xy_strjoin(3, "echo ", file, ">> $HOME/.cargo"); system(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * NuGet 换源 * */ void pl_dotnet_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; xy_error ("chsrc: 暂时无法为NuGet换源,若有需求,请您提交issue"); } /** * PHP 换源 * * 参考:https://developer.aliyun.com/composer */ void pl_php_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; if (xy_on_windows) check_cmd = "composer --version >nul 2>nul"; else check_cmd = "composer --version 1>/dev/null 2>&1"; bool exist_b = does_the_program_exist (check_cmd, "composer"); if (!exist_b) { xy_error ("chsrc: 未找到 composer 相关命令,请检查是否存在"); return; } for (int i=0;iname; const char* source_abbr = pl_php_sources[selected].mirror->abbr; const char* source_url = pl_php_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); char* cmd = xy_2strjoin("composer config repo.packagist composer ", source_url); system(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * Java 换源 * * 参考:https://developer.aliyun.com/mirror/maven */ void pl_java_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; if (xy_on_windows) check_cmd = "mvn --version >nul 2>nul"; else check_cmd = "mvn --version 1>/dev/null 2>&1"; bool mvn_exist_b = does_the_program_exist (check_cmd, "mvn"); if (xy_on_windows) check_cmd = "gradle --version >nul 2>nul"; else check_cmd = "gradle --version 1>/dev/null 2>&1"; bool gradle_exist_b = does_the_program_exist (check_cmd, "gradle"); if (!mvn_exist_b && !gradle_exist_b) { xy_error ("chsrc: maven 与 gradle 命令均未找到,请检查是否存在(其一)"); return; } for (int i=0;iname; const char* source_abbr = pl_java_sources[selected].mirror->abbr; const char* source_url = pl_java_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); if (mvn_exist_b) { const char* file = xy_strjoin(3, "\n" " aliyunmaven\n" " *\n" " 阿里云公共仓库\n" " ", source_url, "\n" ""); xy_info ("chsrc: 请在您的 maven安装目录/conf/settings.xml 中添加:\n"); puts (file); } if (gradle_exist_b) { const char* file = xy_strjoin(3, "allprojects {\n" " repositories {\n" " maven { url '", source_url, "' }\n" " mavenLocal()\n" " mavenCentral()\n" " }\n" "}"); xy_info ("chsrc: 请在您的 build.gradle 中添加:\n"); puts (file); } xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * R 换源 * * 参考:https://help.mirrors.cernet.edu.cn/CRAN/ */ void pl_r_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; for (int i=0;iname; const char* source_abbr = pl_r_sources[selected].mirror->abbr; const char* source_url = pl_r_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); const char* file = xy_strjoin (3, "options(\"repos\" = c(CRAN=\"", source_url, "\"))" ); char* cmd = NULL; // TODO: 待确认,Windows 下是否也是该文件 if (xy_on_windows) cmd = xy_strjoin(3, "echo ", file, " >> %USERPROFILE%/.Rprofile"); else cmd = xy_strjoin(3, "echo ", file, " >> ~/.Rprofile"); system(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } /** * Julia 换源 * * 参考:https://help.mirrors.cernet.edu.cn/julia/ */ void pl_julia_setsrc (char* option) { int selected = 0; char* check_cmd = NULL; for (int i=0;iname; const char* source_abbr = pl_r_sources[selected].mirror->abbr; const char* source_url = pl_r_sources[selected].url; xy_info (xy_2strjoin("chsrc: 选中镜像站:", source_abbr)); const char* file = xy_strjoin (3, "ENV[\"JULIA_PKG_SERVER\"] = \"", source_url, "\""); char* cmd = NULL; // TODO: $JULIA_DEPOT_PATH/config/startup.jl 是否要考虑环境变量 if (xy_on_windows) cmd = xy_strjoin(3, "echo ", file, " >> %USERPROFILE%/.julia/config/startup.jl"); else cmd = xy_strjoin(3, "echo ", file, " >> ~/.julia/config/startup.jl"); system(cmd); xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } void os_ubuntu_setsrc (char* option) { int selected = 0; for (int i=0;iname; const char* source_abbr = os_ubuntu_sources[selected].mirror->abbr; const char* source_url = os_ubuntu_sources[selected].url; char* backup = "cp -rf /etc/apt/sources.list /etc/apt/sources.list.bak"; system(backup); xy_info ("chsrc: 备份文件名: /etc/apt/sources.list.bak"); const char* current_url = xy_strch(source_url,'/',"\\/"); char* cmd = xy_strjoin(3, "sed -E \'s/(^[^#]* .*)http[:|\\.|\\/|a-z|A-Z]*\\/ubuntu\\//\\1", current_url, "/\'< /etc/apt/sources.list.bak | cat > /etc/apt/sources.list"); system(cmd); free(cmd); char* rm = "rm -rf /etc/apt/source.list.bak"; system(rm); // free(rm); xy_info ("chsrc: 为 ubuntu 命令换源"); xy_success (xy_2strjoin("chsrc: 感谢镜像提供方:", source_name)); } #define setsrc_fn(func) (const char const*)func #define getsrc_fn(func) (const char const*)func #define sources_ptr(ptr) (const char const*)ptr static const char const *pl_ruby [] = {"gem", "ruby", "rb", "rubygems", NULL, setsrc_fn(pl_ruby_setsrc) , NULL, sources_ptr(pl_ruby_sources) }, *pl_python[] = {"pip", "python", "py", "pypi", NULL, setsrc_fn(pl_python_setsrc), NULL, sources_ptr(pl_python_sources)}, *pl_nodejs[] = {"npm", "node", "js", "nodejs", NULL, setsrc_fn(pl_nodejs_setsrc), NULL, sources_ptr(pl_nodejs_sources)}, *pl_perl [] = {"perl", "cpan", NULL, setsrc_fn(pl_perl_setsrc), NULL, sources_ptr(pl_perl_sources)}, *pl_rust [] = {"rust", "cargo", "crate", "crates", NULL, setsrc_fn(pl_rust_setsrc), NULL, sources_ptr(pl_rust_sources)}, *pl_go [] = {"go", "golang", "goproxy", NULL, setsrc_fn(pl_go_setsrc), NULL, sources_ptr(pl_go_sources)}, *pl_dotnet[] = {"nuget", "net", ".net", "dotnet", NULL, setsrc_fn(pl_dotnet_setsrc), NULL, sources_ptr(pl_dotnet_sources)}, *pl_java [] = {"java", "maven", "gradle", NULL, setsrc_fn(pl_java_setsrc), NULL, sources_ptr(pl_java_sources)}, *pl_php [] = {"php", "composer", NULL, setsrc_fn(pl_php_setsrc), NULL, sources_ptr(pl_php_sources)}, *pl_cran [] = {"r", "cran", NULL, setsrc_fn(pl_r_setsrc), NULL, sources_ptr(pl_r_sources)}, *pl_julia [] = {"julia", NULL, setsrc_fn(pl_julia_setsrc), NULL, sources_ptr(pl_julia_sources)}, **pl_packagers[] = { pl_ruby, pl_python, pl_nodejs, pl_perl, pl_rust, pl_go, pl_dotnet, pl_java, pl_php, pl_cran, pl_julia }, *os_ubuntu [] = {"ubuntu", NULL, setsrc_fn(os_ubuntu_setsrc)}, **os_systems[] = { os_ubuntu }, *wr_anaconda[] = {"conda", "anaconda", NULL, NULL}, *wr_tex [] = {"latex", "ctan", "tex", NULL, NULL}, *wr_emacs [] = {"emacs", NULL, NULL}, **wr_softwares[] = { wr_anaconda, wr_tex, wr_emacs }; #undef getsrc_fn #undef getsrc_fn #undef sources_ptr static const char const* usage[] = { "chsrc: Change Source \e[1;35m" Chsrc_Version "\e[0m by \e[4;31mRubyMetric\e[0m\n", "维护: https://gitee.com/RubyMetric/chsrc\n", "使用:chsrc [target]", "help # 打印此帮助,或 h, -h, --help", "list [target] # 查看可换源软件,或该软件可以换哪些源", "cesu # 对该软件所有源测速", "get # 查看当前软件的源使用情况", "set # 换源,自动测速后挑选最快源", "set -1 # 1,2,3的1。换源,不测速,挑选经维护者测速排序的第一源", "set -v # 换源,并打印换源所执行的具体操作\n" }; void call_cmd (void* cmdptr, const char* arg) { void (*cmd_func)(const char*) = cmdptr; if (NULL==arg) { xy_info("chsrc: 将使用默认镜像"); } cmd_func(arg); } void print_available_mirrors () { xy_info ("chsrc: 支持以下镜像站,荣耀与感恩均归于这些站点,以及它们的开发/维护者们"); for (int i=0; iabbr, mir->site); puts(mir->name); } } void print_supported_targets_ (const char const*** array, size_t size) { for (int i=0; i