DynaNote/dynanote.py

#!/usr/bin/env python3
"""
DynaNote Shell - 使用 SM-2 算法的间隔重复闪卡系统
"""

import os
import sys
import cmd
import toml
import tempfile
import subprocess
from typing import Dict, List, Optional, Tuple
from tabulate import tabulate

# 导入现有模块
import hasher
import timer
import sm2


class MemoryUnit:
    """表示单个记忆单元（闪卡）"""

    def __init__(self, name: str, unit_id: Optional[str] = None):
        self.name = name
        self.unit_id = unit_id or hasher.get_md5(name)
        self.attachments = set()  # 元组集合 (字符串, 创建时间戳)
        self.created_time = timer.get_daystamp()
        self.algodata = {
            sm2.SM2Algorithm.algo_name: {
                'efactor': 2.5,
                'real_rept': 0,
                'rept': 0,
                'interval': 0,
                'last_date': 0,
                'next_date': 0,
                'is_activated': 0,
                'last_modify': timer.get_timestamp()
            }
        }

    def to_dict(self) -> Dict:
        """将记忆单元转换为字典以便 TOML 序列化"""
        return {
            'name': self.name,
            'attachments': list(self.attachments),
            'created_time': self.created_time,
            'algodata': self.algodata
        }

    @classmethod
    def from_dict(cls, unit_id: str, data: Dict) -> 'MemoryUnit':
        """从字典创建记忆单元"""
        unit = cls(data['name'], unit_id)
        unit.attachments = set(tuple(att) for att in data['attachments'])
        unit.created_time = data['created_time']
        unit.algodata = data['algodata']
        return unit


class DynaNoteShell(cmd.Cmd):
    """DynaNote 间隔重复系统的交互式 shell"""

    intro = '欢迎使用 DynaNote Shell. 输入 help 或 ? 查看命令列表. \n'
    prompt = '(dynanote) > '

    def __init__(self):
        super().__init__()
        self.data_file = './data.toml'
        self.memory_units: Dict[str, MemoryUnit] = {}
        self.selected_unit: Optional[MemoryUnit] = None
        self.load_data()

    def load_data(self) -> None:
        """从 data.toml 文件加载记忆单元"""
        if os.path.exists(self.data_file):
            try:
                with open(self.data_file, 'r', encoding='utf-8') as f:
                    data = toml.load(f)

                for unit_id, unit_data in data.items():
                    self.memory_units[unit_id] = MemoryUnit.from_dict(unit_id, unit_data)

                print(f"已加载 {len(self.memory_units)} 个记忆单元")
            except Exception as e:
                print(f"加载数据时出错: {e}")
        else:
            print("未找到数据文件. 从空数据库开始. ")

    def save_data(self) -> None:
        """将记忆单元保存到 data.toml 文件"""
        try:
            data = {}
            for unit_id, unit in self.memory_units.items():
                data[unit_id] = unit.to_dict()

            with open(self.data_file, 'w', encoding='utf-8') as f:
                toml.dump(data, f)

            print("数据保存成功")
        except Exception as e:
            print(f"保存数据时出错: {e}")

    def find_unit_by_prefix(self, prefix: str) -> Optional[str]:
        """通过前缀查找单元 ID（类似 docker rm）"""
        matches = [unit_id for unit_id in self.memory_units.keys()
                  if unit_id.startswith(prefix)]

        if len(matches) == 1:
            return matches[0]
        elif len(matches) > 1:
            print(f"多个单元匹配前缀 '{prefix}': {', '.join(matches)}")
            return None
        else:
            # 尝试通过名称查找
            name_matches = [unit_id for unit_id, unit in self.memory_units.items()
                           if unit.name == prefix]
            if len(name_matches) == 1:
                return name_matches[0]
            elif len(name_matches) > 1:
                print(f"多个单元具有名称 '{prefix}'")
                return None

        return None

    def do_list(self, arg: str) -> None:
        """列出记忆单元
        用法: list [all|整数]
        - list all: 显示所有记忆单元
        - list 5: 显示复习次数<=5且今天到期的单元
        - list: 同 'list 5'
        """
        if not arg:
            arg = "5"

        if arg.lower() == "all":
            units_to_show = list(self.memory_units.values())
        else:
            try:
                max_reviews = int(arg)
                today = timer.get_daystamp()
                units_to_show = [
                    unit for unit in self.memory_units.values()
                    if unit.algodata[sm2.SM2Algorithm.algo_name]['real_rept'] <= max_reviews
                    and unit.algodata[sm2.SM2Algorithm.algo_name]['next_date'] <= today
                ]
            except ValueError:
                print("无效参数. 使用 'all' 或整数. ")
                return

        if not units_to_show:
            print("没有符合条件的记忆单元")
            return

        # 准备 tabulate 数据
        table_data = []
        for unit in units_to_show:
            algodata = unit.algodata[sm2.SM2Algorithm.algo_name]
            table_data.append([
                unit.unit_id[:8] + "...",
                unit.name,
                algodata['real_rept'],
                algodata['efactor'],
                algodata['next_date'],
                len(unit.attachments)
            ])

        headers = ["ID", "名称", "复习次数", "EFactor", "下次复习", "附件数"]
        print(tabulate(table_data, headers=headers, tablefmt="grid"))

    def do_select(self, arg: str) -> None:
        """通过 ID、前缀或名称选择记忆单元
        用法: select <单元ID|前缀|名称>
        """
        if not arg:
            print("请提供单元 ID、前缀或名称")
            return

        unit_id = self.find_unit_by_prefix(arg)
        if unit_id:
            self.selected_unit = self.memory_units[unit_id]
            print(f"已选择单元: {self.selected_unit.name} ({unit_id[:8]}...)")
        else:
            print(f"未找到唯一匹配 '{arg}' 的单元")

    def do_attach(self, arg: str) -> None:
        """将字符串附加到选定的记忆单元
        用法: attach <字符串>
        """
        if not self.selected_unit:
            print("未选择单元. 请先使用 'select'. ")
            return

        if not arg:
            print("请提供要附加的字符串")
            return

        attachment = (arg, int(timer.get_timestamp()))
        self.selected_unit.attachments.add(attachment)
        print(f"附件已添加到 {self.selected_unit.name}")

    def do_new(self, arg: str) -> None:
        """创建新的记忆单元
        用法: new <名称>
        """
        if not arg:
            print("请提供新单元的名称")
            return

        unit = MemoryUnit(arg)
        self.memory_units[unit.unit_id] = unit
        self.selected_unit = unit
        print(f"已创建新单元: {unit.name} ({unit.unit_id[:8]}...)")

    def do_del(self, arg: str) -> None:
        """删除记忆单元
        用法: del <单元ID|前缀|名称>
        """
        if not arg:
            print("请提供单元 ID、前缀或名称")
            return

        unit_id = self.find_unit_by_prefix(arg)
        if unit_id:
            unit_name = self.memory_units[unit_id].name
            del self.memory_units[unit_id]

            # 如果删除的是当前选中的单元，则清除选择
            if self.selected_unit and self.selected_unit.unit_id == unit_id:
                self.selected_unit = None

            print(f"已删除单元: {unit_name}")
        else:
            print(f"未找到唯一匹配 '{arg}' 的单元")

    def do_mark(self, arg: str) -> None:
        """使用评分（0-5）更新 SM-2 算法
        用法: mark <评分>
        """
        if not self.selected_unit:
            print("未选择单元. 请先使用 'select'. ")
            return

        try:
            rating = int(arg)
            if rating < 0 or rating > 5:
                print("评分必须在 0 到 5 之间")
                return
        except ValueError:
            print("请提供有效的整数评分（0-5）")
            return

        # 显示评分标准以确认
        print("\nSM-2 评分标准:")
        print("  5 - 完美回答")
        print("  4 - 犹豫后正确回答")
        print("  3 - 经过严重困难后回忆起正确答案")
        print("  2 - 错误回答; 但记得正确答案")
        print("  1 - 错误回答; 但正确答案似乎熟悉")
        print("  0 - 完全遗忘")

        # 请求确认
        response = input(f"\n确认对 '{self.selected_unit.name}' 评分 {rating}？(y/N): ")
        if response.lower() not in ['y', 'yes']:
            print("评分已取消")
            return

        sm2.SM2Algorithm.revisor(self.selected_unit.algodata, rating)
        print(f"已使用评分 {rating} 更新 {self.selected_unit.name} 的 SM-2 参数")

    def do_show(self, arg: str) -> None:
        """显示选定记忆单元的详细信息
        用法: show
        """
        if not self.selected_unit:
            print("未选择单元. 请先使用 'select'. ")
            return

        unit = self.selected_unit
        algodata = unit.algodata[sm2.SM2Algorithm.algo_name]

        print(f"\n记忆单元: {unit.name}")
        print(f"ID: {unit.unit_id}")
        print(f"创建时间: {unit.created_time}")
        print(f"\nSM-2 算法数据:")
        print(f"  EFactor: {algodata['efactor']}")
        print(f"  实际复习次数: {algodata['real_rept']}")
        print(f"  当前重复次数: {algodata['rept']}")
        print(f"  间隔: {algodata['interval']} 天")
        print(f"  上次复习: {algodata['last_date']}")
        print(f"  下次复习: {algodata['next_date']}")
        print(f"  是否激活: {algodata['is_activated']}")

        if unit.attachments:
            print(f"\n附件 ({len(unit.attachments)}):")
            for i, (text, timestamp) in enumerate(unit.attachments, 1):
                print(f"  {i}. {text[:50]}{'...' if len(text) > 50 else ''} (于 {timestamp})")
        else:
            print("\n无附件")

    def do_edit(self, arg: str) -> None:
        """使用 nano 编辑器编辑选定的记忆单元
        用法: edit
        """
        if not self.selected_unit:
            print("未选择单元. 请先使用 'select'. ")
            return

        # 创建包含单元数据的临时文件
        with tempfile.NamedTemporaryFile(mode='w', suffix='.toml', delete=False) as f:
            temp_file = f.name
            toml.dump({self.selected_unit.unit_id: self.selected_unit.to_dict()}, f)

        try:
            # 启动 nano 编辑器
            subprocess.run(['nano', temp_file], check=True)

            # 读取编辑后的文件
            with open(temp_file, 'r', encoding='utf-8') as f:
                edited_data = toml.load(f)

            # 更新记忆单元
            if self.selected_unit.unit_id in edited_data:
                updated_data = edited_data[self.selected_unit.unit_id]
                self.memory_units[self.selected_unit.unit_id] = MemoryUnit.from_dict(
                    self.selected_unit.unit_id, updated_data
                )
                self.selected_unit = self.memory_units[self.selected_unit.unit_id]
                print("单元更新成功")
            else:
                print("错误: 在编辑后的文件中未找到单元 ID")

        except subprocess.CalledProcessError:
            print("编辑器已关闭但未保存")
        except Exception as e:
            print(f"编辑单元时出错: {e}")
        finally:
            # 清理临时文件
            if os.path.exists(temp_file):
                os.unlink(temp_file)

    def do_clear(self, arg: str) -> None:
        """清屏
        用法: clear
        """
        os.system('clear')

    def do_help(self, arg: str) -> None:
        """显示帮助和 SM-2 评分标准
        用法: help
        """
        print("\nDynaNote Shell 命令:")
        print("  list [all|整数] - 列出记忆单元")
        print("  select <id|前缀|名称> - 选择记忆单元")
        print("  attach <字符串> - 将字符串附加到选定单元")
        print("  new <名称> - 创建新记忆单元")
        print("  del <id|前缀|名称> - 删除记忆单元")
        print("  mark <评分> - 使用评分（0-5）更新 SM-2")
        print("  show - 显示选定单元详情")
        print("  edit - 使用 nano 编辑选定单元")
        print("  clear - 清屏")
        print("  help - 显示此帮助")
        print("  exit - 退出 shell")

        print("\nSM-2 评分标准:")
        print("  5 - 完美回答")
        print("  4 - 犹豫后正确回答")
        print("  3 - 经过严重困难后回忆起正确答案")
        print("  2 - 错误回答; 但记得正确答案")
        print("  1 - 错误回答; 但正确答案似乎熟悉")
        print("  0 - 完全遗忘")

    def do_exit(self, arg: str) -> bool:
        """退出 shell
        用法: exit
        """
        self.save_data()
        print("再见！")
        return True

    def do_quit(self, arg: str) -> bool:
        """退出 shell（exit 的别名）
        用法: quit
        """
        return self.do_exit(arg)

    # 别名
    def do_sel(self, arg: str) -> None:
        """select 命令的别名"""
        self.do_select(arg)

    def do_ls(self, arg: str) -> None:
        """list 命令的别名"""
        self.do_list(arg)


def main():
    """主入口点"""
    try:
        DynaNoteShell().cmdloop()
    except KeyboardInterrupt:
        print("\n用户中断")
    except Exception as e:
        print(f"错误: {e}")


if __name__ == '__main__':
    main()