import "github.com/TeaMeow/Mego"

func main() {
	// 初始化一個基本的 Mego 引擎。
	e := mego.Default()
	// 定義一個 GetUser 方法供客戶端呼叫。
	e.Register("GetUser", func(c *mego.Context) {
		// 回應 `{"username": "YamiOdymel"}` 資料。
		c.Respond(mego.H{
			"username": "YamiOdymel",
		})
	})
	// 執行引擎在埠口 5000 上。
	e.Run()
}

// 初始化一個帶有自動回復、紀錄中介軟體的基本引擎。
e := mego.Default()

// 初始化一個無預設中介軟體的引擎。
e := mego.New()

func main() {
	e := mego.Default()

	// 廣播一個 `UpdateApp` 事件給所有監聽 `Channel1` 頻道的客戶端，觸發其事件監聽函式。
	e.Emit("UpdateApp", "Channel1", nil)

	// 廣播一個 `UpdateApp` 事件給 `Channel1` 頻道並帶有指定的資料供客戶端讀取。
	e.Emit("UpdateApp", "Channel1", mego.H{
		"version": "1.0.0",
	})

	e.Run()
}

func main() {
	e := mego.Default()

	// 更改預設的事件訂閱處理函式。
	e.HandleSubscribe(func(evt string, ch string, ctx *mego.Context) bool {
		fmt.Printf("客戶端訂閱了 %s 事件的 %s 頻道！", evt, ch)
		return true
	})

	e.Run()
}

func main() {
	e := mego.Default()

	// 建立一個會協助客戶端訂閱事件的方法。
	e.Register("SubscribeMyEvent", func(c *mego.Context) {
		// ... 認證邏輯 ...

		// 手動讓這個客戶端訂閱 `MyEvent` 事件中的 `MyChannel` 頻道。
		c.Subscribe("MyEvent", "MyChannel")
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("UnsubscribeMyEvent", func(c *mego.Context) {
		// 這個客戶端將會取消訂閱 `MyEvent` 事件中的 `MyChannel` 頻道。
		c.Unsubscribe("MyEvent", "MyChannel")
	})

	e.Run()
}

func main() {
	e := mego.Default()

	// 對頻道 `Channel1` 內的 [0] 與 [1] 客戶端廣播 `UpdateApp` 事件，
	// 位於此頻道的其他客戶端不會接收到此事件。
	e.EmitMultiple("UpdateApp", "Channel1", nil, []*mego.Session{
		e.Sessions[0],
		e.Sessions[1],
	})

	e.Run()
}

func main() {
	e := mego.Default()

	// 過濾所有客戶端，僅有 ID 為 `0k3LbEjL` 且在 `Channel1` 內才能夠接收到 UpdateApp 事件。
	e.EmitFilter("UpdateApp", "Channel1", nil, func(s *mego.Session) bool {
		// 此函式會過濾、遍歷每個客戶端，如果此函式回傳 `true`，此客戶端則會接收到此事件。
		return s.ID == "0k3LbEjL"
	})

	e.Run()
}

// User 是一個使用者資料建構體。
type User struct {
	Username string
}

func main() {
	e := mego.Default()

	e.Register("CreateUser", func(c *mego.Context) {
		var u User

		// 將接收到的資料映射到本地的使用者資料建構體，這樣才能讀取其資料。
		if c.Bind(&u) == nil {
			fmt.Println(u.Username)
		}

		// 透過 `MustBind` 在映射錯誤時直接呼叫 `panic` 終止此請求繼續。
		c.MustBind(&u)
		fmt.Println(u.Username)
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("Sum", func(c *mego.Context) {
		// 以正整數資料型態取得傳入的兩個參數。
		a := c.Param(0).GetInt()
		b := c.Param(1).GetInt()

		// 將兩個轉換成正整數的參數進行相加取得結果。
		fmt.Printf("A + B = %d", a+b)
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("Sum", func(c *mego.Context) {
		// 取得客戶端初始化連線時所傳入的 `Token` 資料。
		c.Session.GetString("Token")

		// 在階段中保存名為 `Foo` 且值為 `Bar` 的內容，下次相同客戶端呼叫時可以進行存取。
		c.Session.Set("Foo", "Bar")
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("CreateUser", func(c *mego.Context) {
		// 針對此請求，回傳一個指定的狀態碼與資料。
		c.Respond(mego.H{
			"message": "成功建立使用者！",
		})

		// 針對此請求，回傳一個錯誤相關資料與狀態碼，還有錯誤本身。
		c.RespondWithError(mego.StatusError, mego.H{
			"empty": "username",
		}, errors.New("使用者名稱不可為空。"))
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("CreateUser", func(c *mego.Context) {
		// 廣播 `UpdateApp` 事件到此請求客戶端所監聽的 `Channel1` 中。
		c.Emit("UpdateApp", "Channel1", nil)

		// 廣播 `UpdateApp` 事件到此請求客戶端以外「其他所有人」所監聽的 `Channel1` 中。
		c.EmitOthers("UpdateApp", "Channel1", nil)
	})

	e.Run()
}

func main() {
	e := mego.Default()

	// logMw 是一個會紀錄請求編號的紀錄中介軟體。
	logMw := func(c *mego.Context) {
		fmt.Printf("%s 已連線。", c.ID)
		// ... 紀錄邏輯 ...
		c.Next()
	}

	// authMw 是一個驗證請求者是否有權限的身份檢查中介軟體。
	authMw := func(c *mego.Context) {
		if c.Session.ID != "jn3Dl74eX" {
			// ... 驗證邏輯 ...
		}
		c.Next()
	}

	// 在 Mego 引擎中使用上述兩個全域中介軟體。
	e.Use(logMw, authMw)

	e.Run()
}

func main() {
	e := mego.Default()

	mw := func(c *mego.Context) {
		// 從連線開始的時間。
		start := time.Now()
		// 呼叫下一個中介軟體或執行處理函式。
		c.Next()
		// 取得執行完畢後的時間。
		end := time.Now()

		// 將開始與結束的時間相減，取得此請求所耗費的時間。
		fmt.Printf("此請求共花費了 %v 時間", end.Sub(start))
	}

	// 在 Mego 引擎中使用上述全域中介軟體。
	e.Use(mw)

	e.Run()
}

func main() {
	e := mego.Default()

	mw := func(c *mego.Context) {
		if c.Session.ID != "jn3Dl74eX" {
			// 直接終止此請求的繼續，並停止呼叫接下來的中介軟體與處理函式。
			c.Abort()

			// 終止此請求並且回傳資料。客戶端並不會知道是錯誤發生，仍會依一般回應處理。
			c.AbortWithRespond(mego.H{
				"message": "嗨！雖然你不是 jn3Dl74eX 但我們還是很歡迎你！",
			})

			// 終止此請求並且回傳一個錯誤的資料與狀態碼和錯誤本體。
			c.AbortWithError(mego.StatusNoPermission, nil, errors.New("沒有相關權限可訪問。"))
		}
	}

	// 在 Mego 引擎中使用上述全域中介軟體。
	e.Use(mw)

	e.Run()
}

func main() {
	e := mego.Default()

	mw := func(c *mego.Context) {
		// 在上下文建構體中存入名為 `Foo` 的 `Bar` 值。
		c.Set("Foo", "Bar")
		// 呼叫下一個中介軟體或處理函式。
		c.Next()
	}

	mw2 := func(c *mego.Context) {
		// 檢查是否有 `Foo` 這個鍵名，若沒有則終止此請求繼續。
		if v, ok := c.Get("Foo"); !ok {
			c.Abort()
		}
		fmt.Println(v) // 輸出：Bar
	}

	// 在 Mego 引擎中使用上述兩個全域中介軟體。
	e.Use(mw, mw2)

	e.Run()
}

func main() {
	e := mego.Default()

	mw := func(c *mego.Context) {
		fmt.Println(c.GetString("Foo")) // 輸出：Bar

		// c.GetInt()
		// c.GetBool()
		// c.GetStringMap()
		// ...
	}

	e.Run()
}

func main() {
	e := mego.New()
	e.Use(mego.Recovery())
}

func main() {
	e := mego.New()
	e.Use(mego.Logger())
}

func main() {
	e := mego.Default()

	e.Register("CreateUser", mego.RateLimit(mego.RateLimitOption{
		Period: 1 * time.Hour,
		Limit: 100
	}), func(c *mego.Context) {
		// ...
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("CreateUser", mego.RequestLimit(mego.RequestLimitConfig{
		Limit: 1000
	}), func(c *mego.Context) {
		// ...
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("UploadPhoto", func(c *mego.Context) {
		// 透過 `GetFile` 取得客戶端上傳的 `Photo` 檔案。
		if file, err := c.GetFile("Photo"); err == nil {
			c.Respond(mego.H{
				"filename":  file.Name,      // 檔案原始名稱。
				"size":      file.Size,      // 檔案大小（位元組）。
				"extension": file.Extension, // 檔案副檔名（無符號）。
				"path":      file.Path,      // 檔案本地路徑。
			})
		}
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("UploadPhoto", func(c *mego.Context) {
		// 透過 `GetFiles` 取得客戶端上傳的多個檔案。
		if files, err := c.GetFiles("Photos"); err == nil {
			for _, file := range files {
				fmt.Println(file.Name) // 檔案原始名稱。
			}
		}
	})

	e.Run()
}

// 建立一個自訂的區塊處理函式。
myChunkHandler := func(c *mego.Context, raw *mego.RawFile, dest *mego.File) mego.ChunkStatus {
	// ... 區塊處理邏輯 ...
	// fmt.Println(raw.Binary)

	// 如果這個區塊是最後一塊的話。
	if raw.Last {
		// 將完成的資料保存到檔案建構體。
		dest.Name = "MyFile"
		dest.Size = 128691

		// 表示區塊處理完畢。
		return mego.ChunkDone
	}

	// 如果尚未完成則要求客戶端傳送下一塊。
	return mego.ChunkNext
}

func main() {
	e := mego.Default()

	// 更改全域的區塊處理函式。
	e.HandleChunk(myChunkHandler)

	// 註冊一個接受檔案的方法供客戶端上傳檔案至此。
	e.Register("UploadVideo", func(c *mego.Context) {
		var u User

		// 取得已從區塊組成的完整檔案。
		file := c.MustGetFile()
		fmt.Println(file.Name)

		// 你仍可以將檔案夾帶的資料映射到本地建構體。
		c.MustBind(&u)
		fmt.Println(u.Username)
	})

	// 你也能獨立更改 `UploadVideo` 方法的區塊處理函式。
	e.Method["UploadVideo"].ChunkHandler = myChunkHandler

	e.Run()
}

func main() {
	e := mego.Default()

	// 斷開客戶端 `HkBE9lebt` 與伺服器的連線。
	e.Sessions["HkBE9lebt"].Disconnect()

	// 或者你也不需要指名道姓⋯⋯。
	e.Register("CreateUser", func(c *mego.Context) {
		// 直接斷開發送此請求的客戶端連線。
		c.Session.Disconnect()
	})

	e.Run()
}

func main() {
	e := mego.Default()

	e.Register("CreateUser", func(c *mego.Context) {
		// 複製一份目前的上下文建構體。
		copied := c.Copy()
		// 然後在 Goroutine 中使用這個複製體。
		go func() {
			time.Sleep(5 * time.Second)
			// ... 在這裡使用 copied 而非 c ...
		}()
	})

	e.Run()
}

func main() {
	e := mego.Default()

	// 這是一個錯誤中介軟體，會不斷地在上下文建構體中塞入一堆錯誤。
	errMw := func(c *mego.Context) {
		// 透過 `Error` 能將錯誤訊息堆放置上下文建構體中，
		// 並在最終一次獲得所有錯誤資訊。
		c.Error(errors.New("這是超重要的錯誤訊息啊！"))
		c.Error(errors.New("世界要毀滅啦！"))

		c.Next()
	}

	// 在 Mego 引擎中使用上述全域中介軟體。
	e.Use(errMw)

	e.Register("CreateUser", func(c *mego.Context) {
		// 檢查上下文建構體裡是否有任何錯誤。若有則回傳錯誤訊息給客戶端。
		if len(c.Errors) != 0 {
			c.AbortWithError(mego.StatusError, nil, c.Errors[0].Err)
		}
	})

	e.Run()
}

e.Close()