mirror of
https://github.com/RubyMetric/chsrc.git
synced 2024-11-23 13:39:47 +08:00
974 lines
26 KiB
C
974 lines
26 KiB
C
/* --------------------------------------------------------------
|
||
* File : chsrc.c
|
||
* Authors : Aoran Zeng <ccmywish@qq.com>
|
||
* 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;
|
||
}
|
||
}
|
||
|
||
|
||
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]);
|
||
}
|
||
|
||
|
||
/**
|
||
* 测速代码参考自 https://github.com/mirrorz-org/oh-my-mirrorz/blob/master/oh-my-mirrorz.py
|
||
* 修改为C语言,一切功劳属于原作者
|
||
*
|
||
* @return 返回测得的速度,若无速度或出错,返回0
|
||
*/
|
||
double
|
||
test_speed (char* url)
|
||
{
|
||
char* curl_cmd, *devnull = NULL;
|
||
if (xy_on_windows)
|
||
devnull = "nul";
|
||
else
|
||
devnull = "/dev/null";
|
||
|
||
curl_cmd = xy_strjoin(4, "curl -qs -o ", devnull, "-w '%{http_code} %{speed_download}' -m8 -A chsrc/" Chsrc_Version
|
||
"(+https://gitee.com/RubyMetric/chsrc)", url);
|
||
|
||
FILE* fp = popen(curl_cmd, "r");
|
||
char buf[64] = {0};
|
||
fgets(buf, 64, fp);
|
||
fclose(fp);
|
||
char* split = strchr(buf, ' ');
|
||
*split = '\0';
|
||
int http_code = atoi(buf);
|
||
double speed = atof(split+1);
|
||
|
||
char* speedstr = to_human_readable_speed(speed);
|
||
|
||
puts(url);
|
||
|
||
if (200!=http_code) {
|
||
xy_warn (xy_2strjoin("HTTP码 = ", buf));
|
||
}
|
||
puts(xy_2strjoin("速度 = ", speedstr));
|
||
}
|
||
|
||
|
||
/***************************************** 换源 *********************************************/
|
||
|
||
/**
|
||
* 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;i<sizeof(pl_perl_sources);i++) {
|
||
// 循环测速
|
||
}
|
||
|
||
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));
|
||
}
|
||
|
||
|
||
|
||
/**
|
||
* 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;i<sizeof(pl_nodejs_sources);i++) {
|
||
// 循环测速
|
||
}
|
||
|
||
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));
|
||
}
|
||
|
||
|
||
/**
|
||
* 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;i<sizeof(pl_python_sources);i++) {
|
||
// 循环测速
|
||
}
|
||
|
||
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));
|
||
}
|
||
|
||
|
||
/**
|
||
* Ruby换源
|
||
*
|
||
* 参考:https://gitee.com/RubyKids/rbenv-cn
|
||
*/
|
||
void
|
||
pl_ruby_setsrc (char* option)
|
||
{
|
||
int selected = 0; char* check_cmd = NULL;
|
||
for (int i=0;i<sizeof(pl_ruby_sources);i++) {
|
||
// 循环测速
|
||
}
|
||
|
||
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;
|
||
}
|
||
|
||
const char* source_name = pl_ruby_sources[selected].mirror->name;
|
||
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));
|
||
}
|
||
|
||
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);
|
||
}
|
||
|
||
|
||
|
||
/**
|
||
* 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");
|
||
}
|
||
|
||
|
||
|
||
|
||
/**
|
||
* 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;i<sizeof(pl_php_sources);i++) {
|
||
// 循环测速
|
||
}
|
||
|
||
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));
|
||
}
|
||
|
||
|
||
|
||
/**
|
||
* 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));
|
||
}
|
||
|
||
|
||
|
||
|
||
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));
|
||
}
|
||
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: 为 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), getsrc_fn(pl_ruby_getsrc), 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),NULL,sources_ptr(os_ubuntu_sources)},
|
||
*os_debian [] = {"debian", NULL, setsrc_fn(os_debian_setsrc),NULL,sources_ptr(os_debian_sources)},
|
||
**os_systems[] =
|
||
{
|
||
os_ubuntu,os_debian
|
||
},
|
||
|
||
|
||
*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 <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: 支持以下镜像站,荣耀与感恩均归于这些站点,以及它们的开发/维护者们");
|
||
for (int i=0; i<xy_arylen(available_mirrors); i++)
|
||
{
|
||
mirror_info* mir = available_mirrors[i];
|
||
printf ("%-18s%-41s ", 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 ("%-18s%-50s ", 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_func 如果匹配到,则返回内置targets列表中NULL的位置
|
||
*
|
||
* @return 匹配到则返回true,未匹配到则返回false
|
||
*/
|
||
bool
|
||
iterate_targets_(const char const*** array, size_t size, const char* input, const char const*** target_func)
|
||
{
|
||
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_func = NULL;
|
||
return false;
|
||
}
|
||
|
||
do {
|
||
k++;
|
||
alias = target[k];
|
||
} while (NULL!=alias);
|
||
*target_func = target + k;
|
||
return true;
|
||
}
|
||
|
||
#define iterate_targets(ary, input, target_func) iterate_targets_(ary, xy_arylen(ary), input, target_func)
|
||
|
||
#define Target_Set_Source 1
|
||
#define Target_Get_Source 2
|
||
#define Target_List_Sources 3
|
||
|
||
/**
|
||
* 寻找target,并根据`code`执行相应的操作
|
||
*
|
||
* @param[in] input 用户输入的目标
|
||
* @param[in] code 对target要执行的操作
|
||
*
|
||
* @return 找到目标返回true,未找到返回false
|
||
*/
|
||
bool
|
||
get_target (const char* input, int code)
|
||
{
|
||
const char const** target_func = NULL;
|
||
|
||
bool matched = iterate_targets(pl_packagers, input, &target_func);
|
||
if (!matched) matched = iterate_targets(os_systems, input, &target_func);
|
||
if (!matched) matched = iterate_targets(wr_softwares, input, &target_func);
|
||
|
||
if (!matched) {
|
||
return false;
|
||
}
|
||
|
||
if (Target_Set_Source==code) {
|
||
puts("chsrc: 对该软件换源");
|
||
call_cmd ((void*) *(target_func+code), NULL);
|
||
}
|
||
else if (Target_Get_Source==code) {
|
||
puts("chsrc: 查看该软件的换源情况");
|
||
call_cmd ((void*) *(target_func+code), NULL);
|
||
}
|
||
else if (Target_List_Sources==code) {
|
||
source_info* sources = (source_info*) * (target_func + Target_List_Sources);
|
||
print_supported_sources_for_target (sources);
|
||
}
|
||
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_Sources);
|
||
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;
|
||
}
|
||
// TODO:
|
||
// matched = get_target(argv[2],);
|
||
// if (!matched) goto not_matched;
|
||
puts("chsrc: 测试提供该软件源的镜像站点速度");
|
||
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);
|
||
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;
|
||
}
|
||
matched = get_target(argv[2], Target_Set_Source);
|
||
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;
|
||
}
|
||
}
|