package main

import (
    "fmt"
    "os"
)

// 储存祝福信息
var Mgr wishMessages

//定义保存祝福信息的结构体
type wishMessages struct {
    wish map[uint]*wish
}

//定义祝福话语的结构体
type wish struct {
    id      uint   //祝福信息id
    lucky   string //被祝福的幸运儿
    wisher  string //祝福者
    message string //祝福的话语
}

//查询所有祝福信息
func (w wishMessages) showAllWishs() {
    println("----所有祝福信息如下----\n\n")
    println("ID", "幸运儿", "祝福者", "祝福的话")
    for i, v := range w.wish {
        println(i, v.lucky, v.wisher, v.message)
    }
    println("--------------------------")
}

//增加祝福信息
func (w wishMessages) addWish() {
    println("请输入新祝福id:")
    var (
        wishID      uint
        wishLucky   string
        wishWisher  string
        wishMessage string
    )
    fmt.Scanln(&wishID)
    _, ok := w.wish[wishID]
    if ok {
        println("此祝福话语存在!")
        return
    }
    println("请输入幸运儿:")
    fmt.Scanln(&wishLucky)
    println("请输入祝福者:")
    fmt.Scanln(&wishWisher)
    println("请输入新的祝福的话语:")
    fmt.Scanln(&wishMessage)
    w.wish[wishID] = &wish{wishID, wishLucky, wishWisher, wishMessage}
}

//修改祝福的信息
func (w wishMessages) updateWish() {
    println("请输入要修改祝福话的id:")
    var (
        wishID      uint
        wishLucky   string
        wishWisher  string
        wishMessage string
        num         int
    )
    fmt.Scanln(&wishID)
    _, ok := w.wish[wishID]
    if !ok {
        println("查无此祝福话!")
        return
    }
    println("请输入要修改祝福的哪项信息:\n1.幸运儿;2.祝福者,3.祝福的话语,4.所有信息\n请输入序号:")
    fmt.Scanln(&num)
    switch num {
    case 1:
        println("请输入要修改幸运儿:")
        fmt.Scanln(&wishLucky)
        w.wish[wishID].lucky = wishLucky
    case 2:
        println("请输入要修改祝福者:")
        fmt.Scanln(&wishWisher)
        w.wish[wishID].wisher = wishWisher
    case 3:
        println("请输入要修改祝福的话语:")
        fmt.Scanln(&wishMessage)
        w.wish[wishID].message = wishMessage
    case 4:
        println("请输入要修改幸运儿:")
        fmt.Scanln(&wishLucky)
        println("请输入要修改祝福者:")
        fmt.Scanln(&wishWisher)
        println("请输入要修改祝福的话语:")
        fmt.Scanln(&wishMessage)
        w.wish[wishID] = &wish{
            lucky:   wishLucky,
            wisher:  wishWisher,
            message: wishMessage,
        }
    }
    println("----修改成功----")
}

//删除祝福的信息
func (w wishMessages) delWish() {
    println("请输入要修改祝福话语的id:")
    var wishID uint
    fmt.Scanln(&wishID)
    delete(w.wish, wishID)
    println("----删除成功----")
}

Mgr = wishMessages{
    wish: make(map[uint]*wish, 100),
}
println("-------------欢迎使用祝福信息管理系统-------------")
for {
    println(`
        1.查看所有祝福信息
        2.增加祝福信息
        3.修改祝福信息
        4.删除祝福信息
        5.退出程序
------------------------------------------
    `)
    println("请输入上面对应的序号来选择对应的操作:")
    var wishID uint
    fmt.Scanln(&wishID)
    switch wishID {
    case 1:
        Mgr.showAllWishs()
    case 2:
        Mgr.addWish()
    case 3:
        Mgr.updateWish()
    case 4:
        Mgr.delWish()
    case 5:
        os.Exit(1)
    default:
        println("输入错误，请重新输入!")
    }
}