chsrc/chsrc.c
2023-09-04 09:08:03 +08:00

1344 lines
37 KiB
C
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/* --------------------------------------------------------------
* File : chsrc.c
* Authors : Aoran Zeng <ccmywish@qq.com>
* Created on : <2023-08-28>
* Last modified : <2023-09-03>
*
* 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;
}
}
/**
* 用于 _setsrc 函数检测用户输入的镜像站code是否存在于该target可用源中
*
* 如果用户输入 default 或者 def则选择第一个源
*/
int
does_the_mirror_exist (source_info* sources, size_t size, char* target, char* input)
{
int idx = 0;
source_info source = sources[0];
if (xy_streql("default", input) || xy_streql("def", input)) {
xy_info ("chsrc: 默认使用由维护团队排序第一的镜像(维护者测速第一)");
return idx;
}
bool exist_b = false;
for (int i=1; i<size; i++)
{
if (xy_streql(source.mirror->code, input)) {
idx = i;
exist_b = true;
break;
}
source = sources[i];
}
if (!exist_b) {
xy_error (xy_strjoin(3, "chsrc: 镜像站 ", input, " 不存在"));
xy_error (xy_2strjoin("chsrc: 查看可使用源,请使用 chsrc list ", target));
exit(1);
}
return idx;
}
char*
to_human_readable_speed (double speed)
{
char* scale[] = {"Byte/s", "KByte/s", "MByte/s", "GByte/s", "TByte/s"};
int i = 0;
while (speed > 1024.0)
{
i += 1;
speed /= 1024.0;
}
char* buf = xy_malloc0(64);
sprintf(buf, "%.2f %s", speed, scale[i]);
return buf;
}
/**
* 测速代码参考自 https://github.com/mirrorz-org/oh-my-mirrorz/blob/master/oh-my-mirrorz.py
* 修改为C语言一切功劳属于原作者
*
* @return 返回测得的速度,若出错,返回-1
*/
double
test_speed (char* url)
{
// 我们用 —L因为Ruby China源会跳转到其他地方
char* curl_cmd = xy_strjoin(4, "curl -qsL -o ", xy_os_devnull, " -w \"%{http_code} %{speed_download}\" -m8 -A chsrc/" Chsrc_Version
" ", url);
xy_info (xy_2strjoin("chsrc: 测速 ", url));
FILE* fp = popen(curl_cmd, "r");
char buf[64] = {0};
while(NULL!=fgets(buf, 64, fp));
// puts(buf);
// 如果尾部有换行,删除
char* last_lf = strrchr(buf, '\n');
if (last_lf) *last_lf = '\0';
char* split = strchr(buf, ' ');
if (split) *split = '\0';
// puts(buf); puts(split+1);
int http_code = atoi(buf);
double speed = atof(split+1);
char* speedstr = to_human_readable_speed(speed);
if (200!=http_code) {
xy_warn (xy_strjoin(4, "chsrc: 速度 ", speedstr, " | HTTP码 " , http_code));
} else {
xy_info (xy_2strjoin("chsrc: 速度 ", speedstr));
}
return speed;
}
/**
* @param[in] place 所有源的url后所接的部分url
*/
int
common_cesu (source_info* sources, size_t size, char* place)
{
double speeds[size];
for (int i=0;i<size;i++)
{
source_info src = sources[i];
const char* baseurl = src.url;
char* testurl = xy_2strjoin(baseurl, place);
double speed = test_speed (testurl);
speeds[i] = speed;
}
int fastidx = dblary_maxidx (speeds, size);
xy_success (xy_2strjoin("最快镜像站为: ", sources[fastidx].mirror->name));
return fastidx;
}
/***************************************** 换源 *********************************************/
/* Ruby源 @maintainer ccmywish */
/**
* 我们测 https://mirrors.bfsu.edu.cn/rubygems/gems/nokogiri-1.15.0-java.gem 大小为9.9MB
*
* @return 返回最快源索引
*/
int
pl_ruby_cesu (char* option)
{
size_t size = pl_ruby_sources_n;
source_info* sources = pl_ruby_sources;
return common_cesu (sources, size, "gems/nokogiri-1.15.0-java.gem");
}
void
pl_ruby_getsrc (char* option)
{
char* cmd = "gem sources";
xy_info (xy_2strjoin("chsrc: 运行 ", cmd));
system(cmd);
cmd = "bundle config get mirror.https://rubygems.org";
xy_info (xy_2strjoin("chsrc: 运行 ", cmd));
system(cmd);
}
/**
* Ruby换源参考https://gitee.com/RubyKids/rbenv-cn
*/
void
pl_ruby_setsrc (char* option)
{
int index = 0;
if (NULL!=option) {
index = does_the_mirror_exist (pl_ruby_sources, pl_ruby_sources_n, "ruby", option);
} else {
index = pl_ruby_cesu ("");
}
char* check_cmd = NULL;
if (xy_on_windows) check_cmd = "gem -v >nul 2>nul";
else check_cmd = "gem -v 1>/dev/null 2>&1";
bool exist_b = does_the_program_exist (check_cmd, "gem");
if (!exist_b) {
xy_error ("chsrc: 未找到 gem 相关命令,请检查是否存在");
return;
}
source_info source = pl_ruby_sources[index];
const char* source_name = source.mirror->name;
const char* source_url = source.url;
say_for_setsrc (&source);
xy_info("chsrc: 为 gem 命令换源");
system("gem source -r https://rubygems.org/");
char* cmd = xy_2strjoin("gem source -a ", source_url);
system(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);
xy_success(xy_2strjoin("chsrc: 感谢镜像提供方:", source_name));
}
/**
* TODO: 选择 Python 中的大文件
*/
int
pl_python_cesu (char* option)
{
return common_cesu (pl_python_sources, pl_python_sources_n,
"gems/nokogiri-1.15.0-java.gem");
}
/**
* @param[out] prog 返回 Python 的可用名,如果不可用,则返回 NULL
*/
void
_pl_python_check_cmd (char** prog)
{
char* check_cmd = NULL; *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 相关命令,请检查是否存在");
exit(1);
}
}
void
pl_python_getsrc (char* option)
{
char* prog = NULL;
_pl_python_check_cmd (&prog);
char* cmd = xy_2strjoin(prog, " -m pip config get global.index-url");
system(cmd);
}
/**
* 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* prog = NULL;
_pl_python_check_cmd (&prog);
selected = pl_python_cesu ("");
const char* source_name = pl_python_sources[selected].mirror->name;
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));
}
/**
* TODO: 寻找合适的文件
*/
int
pl_nodejs_cesu (char* option)
{
return common_cesu (pl_nodejs_sources, pl_nodejs_sources_n,
"gems/nokogiri-1.15.0-java.gem");
}
void
_pl_nodejs_check_cmd ()
{
char* check_cmd = 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 命令,请检查是否存在");
exit(1);
}
}
void
pl_nodejs_getsrc (char* option)
{
_pl_nodejs_check_cmd ();
char* cmd = "npm config get registry";
system(cmd);
}
/**
* NodeJS换源参考https://npmmirror.com/
*/
void
pl_nodejs_setsrc (char* option)
{
_pl_nodejs_check_cmd();
int selected = pl_nodejs_cesu ("");
const char* source_name = pl_nodejs_sources[selected].mirror->name;
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));
}
/**
* TODO: 寻找合适的文件
*/
int
pl_perl_cesu (char* option)
{
return common_cesu (pl_perl_sources, pl_perl_sources_n,
"gems/nokogiri-1.15.0-java.gem");
}
void
_pl_perl_check_cmd ()
{
char* check_cmd = 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 命令,请检查是否存在");
exit(1);
}
}
/* TODO: 暂未实现 */
void
pl_perl_getsrc (char* option)
{
_pl_perl_check_cmd ();
// char* cmd = "npm config get registry";
// system(cmd);
}
/**
* Perl换源参考https://help.mirrors.cernet.edu.cn/CPAN/
*/
void
pl_perl_setsrc (char* option)
{
int selected = 0; char* check_cmd, *prog = NULL;
selected = pl_perl_cesu ("");
const char* source_name = pl_perl_sources[selected].mirror->name;
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));
}
/**
* TODO: 寻找合适的文件
*/
int
pl_php_cesu (char* option)
{
return common_cesu (pl_php_sources, pl_php_sources_n,
"gems/nokogiri-1.15.0-java.gem");
}
void
_pl_php_check_cmd()
{
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 命令,请检查是否存在");
exit(1);
}
}
/* TODO: 待PHP用户确认 */
void
pl_php_getsrc (char* option)
{
_pl_php_check_cmd ();
char* cmd = "composer config repo.packagist composer";
system(cmd);
}
/**
* PHP 换源参考https://developer.aliyun.com/composer
*/
void
pl_php_setsrc (char* option)
{
_pl_php_check_cmd();
int selected = pl_php_cesu ("");
const char* source_name = pl_php_sources[selected].mirror->name;
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));
}
/**
* 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;i<sizeof(pl_go_sources);i++) {
// 循环测速
}
const char* source_name = pl_go_sources[selected].mirror->name;
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;i<sizeof(pl_rust_sources);i++) {
// 循环测速
}
const char* source_name = pl_rust_sources[selected].mirror->name;
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");
}
/**
* 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;i<sizeof(pl_java_sources);i++) {
// 循环测速
}
const char* source_name = pl_java_sources[selected].mirror->name;
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,
"<mirror>\n"
" <id>aliyunmaven</id>\n"
" <mirrorOf>*</mirrorOf>\n"
" <name>阿里云公共仓库</name>\n"
" <url>", source_url, "</url>\n"
"</mirror>");
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;i<sizeof(pl_r_sources);i++) {
// 循环测速
}
const char* source_name = pl_r_sources[selected].mirror->name;
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;i<sizeof(pl_r_sources);i++) {
// 循环测速
}
const char* source_name = pl_r_sources[selected].mirror->name;
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));
}
/**
* ubuntu不同架构下的换源是不一样的,这个针对x86架构
*/
void
os_ubuntu_setsrc (char* option)
{
int selected = 0;
for (int i=0;i<sizeof(os_ubuntu_sources);i++) {
// 循环测速
}
const char* source_name = os_ubuntu_sources[selected].mirror->name;
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");
char* cmd = xy_strjoin(3, "sed -E \'s@(^[^#]* .*)http[:|\\.|\\/|a-z|A-Z]*\\/ubuntu\\/@\\1",
source_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);
xy_info ("chsrc: 为 ubuntu 命令换源");
xy_success (xy_2strjoin("chsrc: 感谢镜像提供方:", source_name));
}
/**
* Debian Buster 以上版本默认支持 HTTPS 源。如果遇到无法拉取 HTTPS 源的情况,请先使用 HTTP 源并安装
* sudo apt install apt-transport-https ca-certificates
*/
void
os_debian_setsrc (char* option)
{
int selected = 0;
for (int i=0;i<sizeof(os_ubuntu_sources);i++) {
// 循环测速
}
const char* source_name = os_ubuntu_sources[selected].mirror->name;
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");
char* cmd = xy_strjoin(3, "sed -E \'s@(^[^#]* .*)http[:|\\.|\\/|a-z|A-Z]*\\/debian\\/@\\1",
source_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);
xy_info ("chsrc: 为 debian 命令换源");
xy_success (xy_2strjoin("chsrc: 感谢镜像提供方:", source_name));
}
/**
* fedora29版本及以下暂不支持
*/
void
os_fedora_setsrc (char* option)
{
int selected = 0;
for (int i=0;i<sizeof(os_fedora_sources);i++) {
// 循环测速
}
const char* source_name = os_fedora_sources[selected].mirror->name;
const char* source_abbr = os_fedora_sources[selected].mirror->abbr;
const char* source_url = os_fedora_sources[selected].url;
char* backup = "cp -rf /etc/yum.repos.d/fedora.repo /etc/yum.repos.d/fedora.repo.bak";
system(backup);
backup = "cp -rf /etc/yum.repos.d/fedora-updates.repo /etc/yum.repos.d/fedora-updates.repo.bak";
system(backup);
xy_info ("chsrc: 备份文件名:1. /etc/yum.repos.d/fedora.repo.bak");
xy_info ("chsrc: 备份文件名:2. /etc/yum.repos.d/fedora-updates.repo.bak");
char* cmd = xy_strjoin(9, "sed -e 's|^metalink=|#metalink=|g' ",
"-e 's|^#baseurl=http://download.example/pub/fedora/linux/|baseurl=",
source_url,
"|g' ",
"-i.bak ",
"/etc/yum.repos.d/fedora.repo ",
"/etc/yum.repos.d/fedora-modular.repo ",
"/etc/yum.repos.d/fedora-updates.repo ",
"/etc/yum.repos.d/fedora-updates-modular.repo");
system(cmd);
free(cmd);
xy_info("替换文件:/etc/yum.repos.d/fedora.repo");
xy_info("新增文件:/etc/yum.repos.d/fedora-modular.repo");
xy_info("替换文件:/etc/yum.repos.d/fedora-updates.repo");
xy_info("新增文件:/etc/yum.repos.d/fedora-updates-modular.repo");
// char* rm = "rm -rf /etc/yum.repos.d/fedora.repo.bak";
// system(rm);
// char* rm = "rm -rf /etc/yum.repos.d/fedora-updates.repo.bak";
// system(rm);
xy_info ("chsrc: 为 fedora 命令换源");
xy_success (xy_2strjoin("chsrc: 感谢镜像提供方:", source_name));
}
void
os_kali_setsrc(char* option)
{
int selected = 0;
for (int i=0;i<sizeof(os_kali_sources);i++) {
// 循环测速
}
const char* source_name = os_kali_sources[selected].mirror->name;
const char* source_abbr = os_kali_sources[selected].mirror->abbr;
const char* source_url = os_kali_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");
char* cmd = xy_strjoin(3, "sed -i \'s@(^[^#]* .*)http[:|\\.|\\/|a-z|A-Z]*\\/kali\\/@\\1",
source_url,
"@g\' /etc/apt/sources.list");
system(cmd);
free(cmd);
// char* rm = "rm -rf /etc/apt/source.list.bak";
// system(rm);
xy_info ("chsrc: 为 kali 命令换源");
xy_success (xy_2strjoin("chsrc: 感谢镜像提供方:", source_name));
}
void
os_openbsd_setsrc(char* option)
{
int selected = 0;
for (int i=0;i<sizeof(os_openbsd_sources);i++) {
// 循环测速
}
const char* source_name = os_openbsd_sources[selected].mirror->name;
const char* source_abbr = os_openbsd_sources[selected].mirror->abbr;
const char* source_url = os_openbsd_sources[selected].url;
char* backup = "cp -rf /etc/installurl /etc/installurl.bak";
system(backup);
xy_info ("chsrc: 备份文件名: /etc/installurl.bak");
char* cmd = xy_strjoin(3,"echo ",source_url," > /etc/installurl");
system(cmd);
free(cmd);
// char* rm = "rm -rf /etc/installurl.bak";
// system(rm);
xy_info ("chsrc: 为 openbsd 命令换源");
xy_success (xy_2strjoin("chsrc: 感谢镜像提供方:", source_name));
}
void
os_mysys2_setsrc(char* option)
{
int selected = 0;
for (int i=0;i<sizeof(os_mysys2_sources);i++) {
// 循环测速
}
const char* source_name = os_mysys2_sources[selected].mirror->name;
const char* source_abbr = os_mysys2_sources[selected].mirror->abbr;
const char* source_url = os_mysys2_sources[selected].url;
char* backup = "cp -rf /etc/pacman.d/mirrorlist.mingw32 /etc/pacman.d/mirrorlist.mingw32.bak";
system(backup);
backup = "cp -rf /etc/pacman.d/mirrorlist.mingw64 /etc/pacman.d/mirrorlist.mingw64.bak";
system(backup);
backup = "cp -rf /etc/pacman.d/mirrorlist.msys /etc/pacman.d/mirrorlist.msys.bak";
system(backup);
xy_info ("chsrc: 备份文件名:1. /etc/pacman.d/mirrorlist.mingw32.bak");
xy_info ("chsrc: 备份文件名:2. /etc/pacman.d/mirrorlist.mingw64.bak");
xy_info ("chsrc: 备份文件名:3. /etc/pacman.d/mirrorlist.msys.bak");
char* cmd = xy_strjoin(3,"sed -i \"s#https\?://mirror.msys2.org/#",source_url,"#g\" /etc/pacman.d/mirrorlist* ");
system(cmd);
free(cmd);
// char* rm = "rm -rf /etc/pacman.d/mirrorlist.mingw32.bak";
// system(rm);
// rm = "rm -rf /etc/pacman.d/mirrorlist.mingw64.bak";
// system(rm);
// rm = "rm -rf /etc/pacman.d/mirrorlist.msys.bak";
// system(rm);
xy_info ("chsrc: 为 mysys2 命令换源");
xy_success (xy_2strjoin("chsrc: 感谢镜像提供方:", source_name));
}
/************************************** Begin Target Matrix ****************************************/
def_target_info(pl_ruby);
target_info
pl_python_target = {pl_python_setsrc, NULL, NULL, pl_python_sources, 5},
pl_nodejs_target = {pl_nodejs_setsrc, NULL, NULL, pl_nodejs_sources, 2},
pl_perl_target = {pl_perl_setsrc, NULL, NULL, pl_perl_sources, 5},
pl_rust_target = {pl_rust_setsrc, NULL, NULL, pl_rust_sources, 5},
pl_go_target = {pl_go_setsrc, NULL, NULL, pl_go_sources, 3},
pl_dotnet_target = {pl_dotnet_setsrc, NULL, NULL, pl_dotnet_sources, 1},
pl_java_target = {pl_java_setsrc, NULL, NULL, pl_java_sources, 1},
pl_php_target = {pl_php_setsrc, NULL, NULL, pl_php_sources, 1},
pl_r_target = {pl_r_setsrc, NULL, NULL, pl_r_sources, 5},
pl_julia_target = {pl_julia_setsrc, NULL, NULL, pl_julia_sources, 3};
#define targetinfo(t) (const char const*)t
static const char const
*pl_ruby [] = {"gem", "ruby", "rb", "rubygems", NULL, targetinfo(&pl_ruby_target)},
*pl_python[] = {"pip", "python", "py", "pypi", NULL, targetinfo(&pl_python_target)},
*pl_nodejs[] = {"npm", "node", "js", "nodejs", NULL, targetinfo(&pl_nodejs_target)},
*pl_perl [] = {"perl", "cpan", NULL, targetinfo(&pl_perl_target)},
*pl_rust [] = {"rust", "cargo", "crate", "crates", NULL, targetinfo(&pl_rust_target)},
*pl_go [] = {"go", "golang", "goproxy", NULL, targetinfo(&pl_go_target)} ,
*pl_dotnet[] = {"nuget", "net", ".net", "dotnet", NULL, targetinfo(&pl_dotnet_target)},
*pl_java [] = {"java", "maven", "gradle", NULL, targetinfo(&pl_java_target)},
*pl_php [] = {"php", "composer", NULL, targetinfo(&pl_php_target)},
*pl_r [] = {"r", "cran", NULL, targetinfo(&pl_r_target)},
*pl_julia [] = {"julia", NULL, targetinfo(&pl_julia_target)},
**pl_packagers[] =
{
pl_ruby, pl_python, pl_nodejs, pl_perl,
pl_rust, pl_go, pl_dotnet, pl_java, pl_php,
pl_r, pl_julia
};
target_info
os_ubuntu_target = {os_ubuntu_setsrc, NULL, NULL, os_ubuntu_sources, 7},
os_debian_target = {os_debian_setsrc, NULL, NULL, os_debian_sources, 7},
os_fedora_target = {os_ubuntu_setsrc, NULL, NULL, os_ubuntu_sources, 7},
os_kali_target = {os_ubuntu_setsrc, NULL, NULL, os_ubuntu_sources, 7},
os_openbsd_target = {os_ubuntu_setsrc, NULL, NULL, os_ubuntu_sources, 7},
os_mysys2_target = {os_ubuntu_setsrc, NULL, NULL, os_ubuntu_sources, 7};
static const char const
*os_ubuntu [] = {"ubuntu", NULL, targetinfo(&os_ubuntu_target)},
*os_debian [] = {"debian", NULL, targetinfo(&os_debian_target)},
*os_fedora [] = {"debian", NULL, targetinfo(&os_debian_target)},
*os_kali [] = {"debian", NULL, targetinfo(&os_debian_target)},
*os_openbsd [] = {"debian", NULL, targetinfo(&os_debian_target)},
*os_mysys2 [] = {"debian", NULL, targetinfo(&os_debian_target)},
**os_systems[] =
{
os_ubuntu, os_debian
};
target_info
wr_anaconda_target = {NULL, NULL, NULL, NULL, 0},
wr_emacs_target = {NULL, NULL, NULL, NULL, 0},
wr_tex_target = {NULL, NULL, NULL, NULL, 0};
static const char const
*wr_anaconda[] = {"conda", "anaconda", NULL, targetinfo(&wr_anaconda_target)},
*wr_emacs [] = {"emacs", NULL, targetinfo(&wr_emacs_target)},
*wr_tex [] = {"latex", "ctan", "tex", NULL, targetinfo(&wr_tex_target) },
**wr_softwares[] =
{
wr_anaconda, wr_emacs, wr_tex
};
#undef targetinfo
/************************************** End Target Matrix ****************************************/
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 <command> [target]",
"help # 打印此帮助,或 h, -h, --help",
"list (或 ls, 或 l) # 查看可用镜像源,和可换源软件",
"list mirror(s) # 查看可用镜像源",
"list target(s) # 查看可换源软件",
"list <target> # 查看该软件可以使用哪些源",
"cesu <target> # 对该软件所有源测速",
"get <target> # 查看当前软件的源使用情况",
"set <target> # 换源,自动测速后挑选最快源",
"set <target> -1 # 1,2,3的1。换源不测速挑选经维护者测速排序的第一源",
"set <target> -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: 支持以下镜像站,荣耀均归属于这些站点,以及它们的开发/维护者们");
xy_warn ("chsrc: 下方code列可用于指定使用某镜像站请使用 chsrc set <target> <code>");
printf ("%-14s%-30s%-41s ", "code", "服务商缩写", "服务商URL"); puts("服务商名称");
puts ("-------------------------------------------------------------------------------------------------");
for (int i=0; i<xy_arylen(available_mirrors); i++)
{
mirror_info* mir = available_mirrors[i];
printf ("%-14s%-18s%-41s ", mir->code, mir->abbr, mir->site); puts(mir->name);
}
}
void
print_supported_targets_ (const char const*** array, size_t size)
{
for (int i=0; i<size; i++)
{
const char ** target = array[i];
const char* alias = target[0];
for (int k=1; alias!=NULL; k++)
{
printf ("%s\t", alias);
alias = target[k];
}
puts("");
}
puts("");
}
void
print_supported_targets ()
{
xy_info ("chsrc: 支持对以下目标换源 (同一行表示这几个命令兼容)");
xy_warn ("编程语言开发");
print_supported_targets_ (pl_packagers, xy_arylen(pl_packagers));
xy_warn ("操作系统");
print_supported_targets_ (os_systems, xy_arylen(os_systems));
xy_warn ("软件");
print_supported_targets_ (wr_softwares, xy_arylen(wr_softwares));
}
/**
* 用于 chsrc list <target>
*/
void
print_supported_sources_for_target (source_info sources[])
{
for (int i=0;i<4;i++)
{
source_info src = sources[i];
const mirror_info* mir = src.mirror;
printf ("%-14s%-18s%-50s ",mir->code, mir->abbr, src.url);
puts(mir->name);
}
}
int
print_help ()
{
for (int i=0; i<xy_arylen(usage); i++) {
puts (usage[i]);
}
}
/**
* 遍历我们内置的targets列表查询用户输入`input`是否与我们支持的某个target匹配
*
* @param[out] target_info 如果匹配到则返回内置targets列表中最后的target_info信息
*
* @return 匹配到则返回true未匹配到则返回false
*/
bool
iterate_targets_(const char const*** array, size_t size, const char* input, const char const*** target_info)
{
int matched = 0;
const char const** target = NULL;
int k = 0;
const char* alias = NULL;
for (int i=0; i<size; i++) {
target = array[i];
alias = target[k];
while (NULL!=alias) {
if (xy_streql(input, alias)) {
matched = 1; break;
}
k++;
alias = target[k];
}
if (!matched) k = 0;
if (matched) break;
}
if(!matched) {
*target_info = NULL;
return false;
}
do {
k++;
alias = target[k];
} while (NULL!=alias);
*target_info = target + k + 1;
return true;
}
#define iterate_targets(ary, input, target) iterate_targets_(ary, xy_arylen(ary), input, target)
#define Target_Set_Source 1
#define Target_Get_Source 2
#define Target_Cesu_Source 3
#define Target_List_Source 4
/**
* 寻找target并根据`code`执行相应的操作
*
* @param[in] input 用户输入的目标
* @param[in] code 对target要执行的操作
* @param[in] option 额外的指示可为NULL
*
* @return 找到目标返回true未找到返回false
*/
bool
get_target (const char* input, int code, char* option)
{
const char const** target_tmp = NULL;
bool matched = iterate_targets(pl_packagers, input, &target_tmp);
if (!matched) matched = iterate_targets(os_systems, input, &target_tmp);
if (!matched) matched = iterate_targets(wr_softwares, input, &target_tmp);
if (!matched) {
return false;
}
target_info* target = (target_info*) *target_tmp;
if (Target_Set_Source==code) {
if (target->setfn) target->setfn(option);
else xy_error (xy_strjoin(3, "chsrc: 暂未对", input, "实现set功能欢迎贡献"));
}
else if (Target_Get_Source==code) {
if (target->getfn) target->getfn("");
else xy_error (xy_strjoin(3, "chsrc: 暂未对", input, "实现get功能欢迎贡献"));
}
else if (Target_List_Source==code) {
xy_info (xy_strjoin(3,"chsrc: 对", input ,"支持以下镜像站,荣耀均归属于这些站点,以及它们的开发/维护者们"));
xy_warn (xy_strjoin(3, "chsrc: 下方code列可用于指定使用某源请使用 chsrc set ", input, " <code>"));
printf ("%-14s%-35s%-45s ", "code", "服务商缩写", "服务源URL"); puts("服务商名称");
puts ("--------------------------------------------------------------------------------------------------------");
print_supported_sources_for_target (target->sources);
}
else if (Target_Cesu_Source==code) {
if (!target->cesufn)
xy_error (xy_strjoin(3, "chsrc: 暂未对", input, "实现cesu功能欢迎贡献"));
else {
char* check_cmd = xy_str_to_quietcmd("curl --version");
bool exist_b = does_the_program_exist (check_cmd, "curl");
if (!exist_b) xy_error ("chsrc: 没有curl命令无法测速");
else target->cesufn("");
return true;
}
}
return true;
}
int
main (int argc, char const *argv[])
{
xy_useutf8(); argc -= 1;
if (argc==0) {
print_help(); return 0;
}
const char* command = argv[1];
const char* target = NULL;
bool matched = false;
/* chsrc help */
if (xy_streql(command, "h") ||
xy_streql(command, "-h") ||
xy_streql(command, "help") ||
xy_streql(command, "--help"))
{
print_help();
return 0;
}
/* chsrc list */
else if (xy_streql(command, "list") ||
xy_streql(command, "l") ||
xy_streql(command, "ls"))
{
if (argc < 2) {
print_available_mirrors();
puts("");
print_supported_targets();
} else {
if (xy_streql(argv[2],"mirrors")) {
print_available_mirrors(); return 0;
}
if (xy_streql(argv[2],"mirror")) {
print_available_mirrors(); return 0;
}
if (xy_streql(argv[2],"targets")) {
print_supported_targets(); return 0;
}
if (xy_streql(argv[2],"target")) {
print_supported_targets(); return 0;
}
matched = get_target(argv[2], Target_List_Source, NULL);
if (!matched) goto not_matched;
}
return 0;
}
/* chsrc cesu */
else if (xy_streql(command, "cesu") ||
xy_streql(command, "ce") ||
xy_streql(command, "c"))
{
if (argc < 2) {
xy_error ("chsrc: 请您提供想要测速源的软件名; 使用 chsrc list targets 查看所有支持的软件");
return 1;
}
matched = get_target(argv[2], Target_Cesu_Source, NULL);
if (!matched) goto not_matched;
return 0;
}
/* chsrc get */
else if (xy_streql(command, "get") ||
xy_streql(command, "g"))
{
if (argc < 2) {
xy_error ("chsrc: 请您提供想要查看源的软件名; 使用 chsrc list targets 查看所有支持的软件");
return 1;
}
matched = get_target(argv[2], Target_Get_Source, NULL);
if (!matched) goto not_matched;
return 0;
}
/* chsrc set */
else if (xy_streql(command, "set") ||
xy_streql(command, "s"))
{
if (argc < 2) {
xy_error ("chsrc: 请您提供想要设置源的软件名; 使用 chsrc list targets 查看所有支持的软件");
return 1;
}
char* option = NULL;
if (argc >= 3) {
option = (char*) argv[3]; // 暂时我们只接受最多三个参数
}
matched = get_target(argv[2], Target_Set_Source, option);
if (!matched) goto not_matched;
return 0;
}
/* 不支持的命令 */
else
{
xy_error ("chsrc: 不支持的命令,请使用 chsrc help 查看使用方式");
return 1;
}
not_matched:
if (!matched) {
xy_info("chsrc: 暂不支持的换源目标,请使用 chsrc list targets 查看可换源软件");
return 1;
}
}