#!/bin/bash

# debian 初始化

# 设置颜色
RED='\033[0;31m'
GREEN='\033[0;32m'
NC='\033[0m'

# 显示帮助信息
show_help() {
    echo "用法: $0 [选项]"
    echo "选项:"
    echo "  rootLogin    配置root登录"
    echo "  soft         安装常用软件"
    echo "  all          执行所有操作"
    echo "  help         显示此帮助信息"
}

# 配置root登录
config_root_login() {
    echo -e "${GREEN}开始配置root登录...${NC}"
    
    # 检查是否以root权限运行
    if [ "$EUID" -ne 0 ]; then 
        echo -e "${RED}请使用root权限运行此脚本${NC}"
        exit 1
    fi

    # 修改SSH配置
    echo -e "${GREEN}修改SSH配置...${NC}"
    if grep -q "^#*PermitRootLogin" /etc/ssh/sshd_config; then
        sed -i 's/^#*PermitRootLogin.*/PermitRootLogin yes/' /etc/ssh/sshd_config
    else
        echo "PermitRootLogin yes" >> /etc/ssh/sshd_config
    fi

    # 重启SSH服务
    echo -e "${GREEN}重启SSH服务...${NC}"
    service ssh restart

    echo -e "${GREEN}root登录配置完成！${NC}"
}

# 安装软件
install_software() {
    echo -e "${GREEN}开始安装常用软件...${NC}"
    
    # 安装curl
    echo -e "${GREEN}安装curl、wget和sudo...${NC}"
    apt-get install -y wget curl sudo unzip

    # 安装vim
    echo -e "${GREEN}安装vim...${NC}"
    apt-get install vim -y
    
    echo -e "${GREEN}配置vim...${NC}"
    tee ~/.vimrc <<-'EOF'
if has('mouse')
    set mouse-=a  " 关闭所有模式的鼠标支持
endif
EOF

    # 安装git
    echo -e "${GREEN}安装git...${NC}"
    apt-get install git -y

    # 配置ll别名
    echo -e "${GREEN}配置ll别名...${NC}"
    echo "alias ll='ls -l'" >> ~/.bashrc
    source ~/.bashrc

    echo -e "${GREEN}软件安装完成！${NC}"
}

# 主程序
case "$1" in
    "rootLogin")
        config_root_login
        ;;
    "soft")
        install_software
        ;;
    "all")
        config_root_login
        install_software
        ;;
    "help"|"")
        show_help
        ;;
    *)
        echo -e "${RED}无效的参数: $1${NC}"
        show_help
        exit 1
        ;;
esac