修正支持列表

.gitignore

@@ -0,0 +1,5 @@
+################################################################################
+# 此 .gitignore 文件已由 Microsoft(R) Visual Studio 自动创建。
+################################################################################
+
+/.vs

README.md

@@ -1,34 +1,22 @@
-# 一键切换Ubuntu系统到简体中文
-## 两个版本
-1.C++  
-2.纯Shell  
-## 不同版本区别
-这两个版本的唯一区别就是运行方式不一样，功能和结构没有任何区别  
-不信你可以去看源代码，C++版本我只是用了system函数执行shell语句，效果不能说天差地别吧，至少也能说是一模一样
-***
+# 一键切换Ubuntu(和Debian)系统到简体中文
+## 测试通过的系统
+|系统名称|版本|
+|-|-|
+|Debian|11.6.0|
+|Ubuntu|22.04 LTS|
+
 ## 使用方法
-# **注意！程序运行完之后会自动重启目标电脑/服务器，请确保没有未保存的工作！否则后果自负！**
 确保已经安装wget或者有其他方法可以传输文件  
 ```shell
 apt update && apt install -y wget
 ```
-***
-下面两个版本选一个就行
-***
-下载程序(Shell)
-```shell
-wget https://raw.githubusercontent.com/TheWhiteDog9487/ubuntu-language-onekey-zh_cn/main/language.sh
-```
-下载程序(C++)
+
+下载程序
 ```shell
 wget https://raw.githubusercontent.com/TheWhiteDog9487/ubuntu-language-onekey-zh_cn/main/language
 ```
-运行程序(Shell)
-```shell
-sudo bash language.sh
-```
-运行程序(C++)
+运行程序
 ```shell
 sudo chmod o+x language
 sudo ./language
-```
+```

language.cpp

@@ -1,9 +1,95 @@
-#include<cstdlib>
+#include <cstdlib>
+#include <iostream>
+#include <string>
+#include <fstream>
 using namespace std;
-int main(){
+
+string get_linux_distro_name() {
+    // ChatGPT
+    std::string distro_name = "";
+    std::ifstream release_file("/etc/os-release");
+    if (release_file.is_open()) {
+        std::string line = "";
+        while (getline(release_file, line)) {
+            if (line.find("PRETTY_NAME=") != std::string::npos) {
+                distro_name = line.substr(line.find("=") + 1);
+                distro_name.erase(0, 1); // remove opening quote
+                distro_name.erase(distro_name.size() - 1); // remove closing quote
+                break;}}
+        release_file.close();}
+    return distro_name;}
+
+auto check() {
+    cout << "请输入您的选项（输入y或者yes继续）";
+    string userin = "";
+    cin >> userin;
+    if (userin == "y" or userin == "yes")
+        return;
+    else{
+        cout << "输入不正确欸，如果想停止本程序的话请按下Ctrl+C" << endl;
+        check();}}
+
+auto Ubuntu(){
+    cout << "提示：本程序需要root权限，如果您不是以root运行的本程序，请Ctrl+C终止并重新提权运行" << endl;
+    cout << "如果确认权限没有问题，请输入y或者yes";
+    check();
+    cout << "即将开始安装并配置简体中文语言包，请输入y或者yes继续，或者按Ctrl+C终止本程序";
+    check();
     system("apt install -y language-pack-zh-hans");
     system("locale-gen zh_CN.UTF-8");
     system("mv /etc/default/locale /etc/default/locale.backup");
-    system("touch /etc/default/locale");
-    system("echo LANG=\"zh_CN.UTF-8\" > /etc/default/locale");
-    system("reboot");}
+    system("wget https://raw.githubusercontent.com/TheWhiteDog9487/ubuntu-language-onekey-zh_cn/main/locale");
+    system("mv locale /etc/default/locale");
+    system("chmod 644 /etc/default/locale");
+    cout << "原/etc/default/locale已经备份为/etc/default/locale.backup" << endl;
+    cout << "即将重启以应用语言设置，请保存您未完成的工作以防数据丢失" << endl;
+    cout << "输入y或者yes重启或者按Ctrl+C终止本程序" << endl;
+    check();
+    system("reboot");}
+
+auto Debian(){
+    cout << "提示：本程序需要root权限，如果您不是以root运行的本程序，请Ctrl+C终止并重新提权运行" << endl;
+    cout << "如果确认权限没有问题，请输入y或者yes";
+    check();
+    system("mv /etc/locale.gen /etc/locale.gen.backup");
+    system("echo zh_CN.UTF-8 UTF-8 > /etc/locale.gen");
+    system("locale-gen");
+    system("chmod 644 /etc/locale.gen");
+    system("mv /etc/default/locale /etc/default/locale.backup");
+    system("wget https://raw.githubusercontent.com/TheWhiteDog9487/ubuntu-language-onekey-zh_cn/main/locale");
+    system("mv locale /etc/default/locale");
+    system("chmod 644 /etc/default/locale");
+    cout << "原/etc/locale.gen已经备份为/etc/locale.gen.backup" << endl;
+    cout << "原/etc/default/locale已经备份为/etc/default/locale.backup" << endl;
+    cout << "即将重启以应用语言设置，请保存您未完成的工作以防数据丢失" << endl;
+    cout << "输入y或者yes重启或者按Ctrl+C终止本程序" << endl;
+    check();
+    system("reboot");}
+
+int main() {
+    auto OS = [](){
+        #pragma warning(suppress : 4996)
+        if (getenv("OS") != NULL){
+            return "Windows";}
+        #pragma warning(suppress : 4996)
+        else if (getenv("HOME") != NULL){
+            return "Linux";}
+        else{
+            return "未知";}}();
+
+	if (OS == "Windows"){
+		cout << "Windows请在“设置”应用中调整语言(❁´◡`❁)。" << endl;
+        exit(-1);}
+    else if (OS == "Linux") {
+        if (get_linux_distro_name().find("Ubuntu") != string::npos){
+            Ubuntu();}
+        if (get_linux_distro_name().find("Debian") != string::npos){
+            Debian();}
+        else { exit(-1); }}
+	else if (OS == "未知"){
+        cout << "未知操作系统，为确保数据安全将会强制退出。"
+            << endl 
+            << "如果是误报，请联系作者修复问题。"
+            << endl;
+        exit(-1);}
+}

language.sh

@@ -1,7 +0,0 @@
-apt update
-apt install -y language-pack-zh-hans
-locale-gen zh_CN.UTF-8
-mv /etc/default/locale /etc/default/locale.backup
-touch /etc/default/locale
-echo LANG="zh_CN.UTF-8" > /etc/default/locale
-reboot

locale

@@ -0,0 +1,12 @@
+LANG="zh_CN.UTF-8"
+LANGUAGE="zh_CN:zh:en_US:en"
+LC_NUMERIC="zh_CN.UTF-8"
+LC_TIME="zh_CN.UTF-8"
+LC_MONETARY="zh_CN.UTF-8"
+LC_PAPER="zh_CN.UTF-8"
+LC_IDENTIFICATION="zh_CN.UTF-8"
+LC_NAME="zh_CN.UTF-8"
+LC_ADDRESS="zh_CN.UTF-8"
+LC_TELEPHONE="zh_CN.UTF-8"
+LC_MEASUREMENT="zh_CN.UTF-8"
+LC_ALL=zh_CN.UTF-8