package main

import "fmt"

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

// Originator 发起者，拥有需要保存的状态
type Originator struct {
	state string
}

// SetState 设置当前状态
func (o *Originator) SetState(state string) {
	o.state = state
	fmt.Printf("状态设置为: %s\n", state)
}

// S*eToMemento 保存当前状态到备忘录
func (o *Originator) S*eToMemento() *Memento {
	return &Memento{state: o.state}
}

// RestoreFromMemento 从备忘录恢复状态
func (o *Originator) RestoreFromMemento(m *Memento) {
	o.state = m.state
	fmt.Printf("状态恢复为: %s\n", o.state)
}

// Caretaker 管理者，管理多个备忘录（例如历史记录）
type Caretaker struct {
	mementos []*Memento
}

// Add 添加一个备忘录
func (c *Caretaker) Add(m *Memento) {
	c.mementos = append(c.mementos, m)
}

// Get 获取指定索引的备忘录
func (c *Caretaker) Get(index int) *Memento {
	if index < 0 || index >= len(c.mementos) {
		return nil
	}
	return c.mementos[index]
}

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

	originator.SetState("状态1")
	caretaker.Add(originator.S*eToMemento())

	originator.SetState("状态2")
	caretaker.Add(originator.S*eToMemento())

	originator.SetState("状态3")

	// 撤销到上一个状态
	lastMemento := caretaker.Get(1)
	if lastMemento != nil {
		originator.RestoreFromMemento(lastMemento)
	}
}