import subprocess
import readline
import os
import shlex

# 自动补全功能的实现
def completer(text, state):
    # 获取系统中可执行命令的列表
    commands = os.listdir('/bin') + os.listdir('/usr/bin') + os.listdir('/usr/local/bin')
    # 根据用户输入的前缀匹配命令
    matches = [cmd for cmd in commands if cmd.startswith(text)]
    # 返回匹配的命令
    return matches[state] if state < len(matches) else None

# 模拟一个简单的 zsh 终端
def pseudo_zsh():
    # 配置 readline 的自动补全功能
    readline.parse_and_bind("tab: complete")
    readline.set_completer(completer)
    
    while True:
        try:
            # 显示提示符并获取用户输入
            cmd = input("20240915786@\u9648\u5764\u9633 ~ % ")
            
            # 如果输入特定命令 "hexianglong"，退出程序
            if cmd.strip() == "hexianglong":
                print("Exiting secret mode...")
                break
            
            # 使用 shlex 分割用户输入为命令和参数
            args = shlex.split(cmd)
            if not args:  # 如果输入为空，跳过本次循环
                continue
            
            # 实现简单的 cd 命令
            if args[0] == 'cd':
                try:
                    os.chdir(args[1])  # 切换到指定目录
                except IndexError:
                    print("cd: missing argument")  # 缺少参数
                except FileNotFoundError:
                    print(f"cd: no such file or directory: {args[1]}")  # 目录不存在
                continue
            
            # 执行其他命令
            try:
                subprocess.run(args)  # 调用系统命令
            except FileNotFoundError:
                print(f"zsh: command not found: {args[0]}")  # 命令未找到
        except KeyboardInterrupt:
            # 捕获 Ctrl+C，忽略中断
            pass
        except EOFError:
            # 捕获 Ctrl+D，忽略结束输入
            pass

# 程序入口
if __name__ == "__main__":
    pseudo_zsh()