package main

import "fmt"

// Memento 备忘录结构体，保存Originator的状态
type Memento struct {
    state string
}

// Originator 发起人，维护当前状态，可生成或恢复备忘录
type Originator struct {
    state string
}

func (o *Originator) SetState(state string) {
    o.state = state
}

func (o *Originator) GetState() string {
    return o.state
}

// 创建备忘录，保存当前状态
func (o *Originator) CreateMemento() *Memento {
    return &Memento{state: o.state}
}

// 从备忘录恢复状态
func (o *Originator) RestoreFromMemento(m *Memento) {
    o.state = m.state
}

// Caretaker 管理者，保存多个备忘录（例如用于撤销栈）
type Caretaker struct {
    history []*Memento
}

func (c *Caretaker) Add(m *Memento) {
    c.history = append(c.history, m)
}

func (c *Caretaker) Get(index int) *Memento {
    if index < 0 || index >= len(c.history) {
        return nil
    }
    return c.history[index]
}

func main() {
    originator := &Originator{}
    caretaker := &Caretaker{}

    originator.SetState("第一次编辑")
    caretaker.Add(originator.CreateMemento())

    originator.SetState("第二次编辑")
    caretaker.Add(originator.CreateMemento())

    originator.SetState("第三次编辑")
    fmt.Println("当前状态:", originator.GetState())

    // 恢复到第二次编辑的状态
    memento := caretaker.Get(1)
    if memento != nil {
        originator.RestoreFromMemento(memento)
    }
    fmt.Println("恢复后状态:", originator.GetState())
}