main.go

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"
	"strings"
	"time"

	"github.com/line/line-bot-sdk-go/v7/linebot"
	gpt3 "github.com/sashabaranov/go-gpt3"
)

var bot *linebot.Client
var client *gpt3.Client
var summaryQueue GroupStorage

func main() {
	var err error
	summaryQueue = make(GroupStorage)
	bot, err = linebot.New(os.Getenv("ChannelSecret"), os.Getenv("ChannelAccessToken"))
	log.Println("Bot:", bot, " err:", err)

	port := os.Getenv("PORT")
	apiKey := os.Getenv("ChatGptToken")

	if apiKey != "" {
		client = gpt3.NewClient(apiKey)
	}

	http.HandleFunc("/callback", callbackHandler)
	addr := fmt.Sprintf(":%s", port)
	http.ListenAndServe(addr, nil)
}

func callbackHandler(w http.ResponseWriter, r *http.Request) {
	events, err := bot.ParseRequest(r)

	if err != nil {
		if err == linebot.ErrInvalidSignature {
			w.WriteHeader(400)
		} else {
			w.WriteHeader(500)
		}
		return
	}

	for _, event := range events {
		if event.Type == linebot.EventTypeMessage {
			switch message := event.Message.(type) {
			// Handle only on text message
			case *linebot.TextMessage:
				// 預設訊息
				reply := "msg ID:" + message.ID + ":" + "Get:" + message.Text + " , \n OK!"

				// 如果聊天機器人在群組中，開始儲存訊息。
				if event.Source.GroupID != "" {

					// 先取得使用者 Display Name (也就是顯示的名稱)
					userName := event.Source.UserID
					userProfile, err := bot.GetProfile(event.Source.UserID).Do()
					if err == nil {
						userName = userProfile.DisplayName
					}

					// event.Source.GroupID 就是聊天群組的 ID，並且透過聊天群組的 ID 來放入 Map 之中。
					q := summaryQueue[event.Source.GroupID]
					m := MsgDetail{
						MsgText:  message.Text,
						UserName: userName,
						Time:     time.Now(),
					}
					log.Println("Save msg:", m)
					summaryQueue[event.Source.GroupID] = append(q, m)
					log.Println("All msg:", q)
				}

				// Directly to ChatGPT
				if strings.Contains(message.Text, ":gpt") {
					reply = CompleteContext(message.Text)

					if _, err = bot.ReplyMessage(event.ReplyToken, linebot.NewTextMessage(reply)).Do(); err != nil {
						log.Print(err)
					}
				} else if strings.EqualFold(message.Text, ":list_all") {
					q := summaryQueue[event.Source.GroupID]
					for _, m := range q {
						reply = reply + fmt.Sprintf("[%s]: %s . %s\n", m.UserName, m.MsgText, m.Time.Local().UTC().Format("2006-01-02 15:04:05"))
					}

					if _, err = bot.ReplyMessage(event.ReplyToken, linebot.NewTextMessage(reply)).Do(); err != nil {
						log.Print(err)
					}
				} else if strings.EqualFold(message.Text, ":sum_all") {
					// 把聊天群組裡面的訊息都捲出來（依照先後順序）
					oriContext := ""
					q := summaryQueue[event.Source.GroupID]
					for _, m := range q {
						// [xxx]: 他講了什麼... 時間
						oriContext = oriContext + fmt.Sprintf("[%s]: %s . %s\n", m.UserName, m.MsgText, m.Time.Local().UTC().Format("2006-01-02 15:04:05"))
					}

					// 取得使用者暱稱
					userName := event.Source.UserID
					userProfile, err := bot.GetProfile(event.Source.UserID).Do()
					if err == nil {
						userName = userProfile.DisplayName
					}

					// 訊息內先回，再來總結。
					if _, err = bot.ReplyMessage(event.ReplyToken, linebot.NewTextMessage("好的，總結文字已經發給您了"+userName)).Do(); err != nil {
						log.Print(err)
					}

					// 就是請 ChatGPT 幫你總結
					oriContext = fmt.Sprintf("幫我總結 `%s`", oriContext)
					reply = CompleteContext(oriContext)

					// 因為 ChatGPT 可能會很慢，所以這邊後來用 SendMsg 來發送私訊給使用者。
					if _, err = bot.PushMessage(event.Source.UserID, linebot.NewTextMessage(reply)).Do(); err != nil {
						log.Print(err)
					}
				}

			// Handle only on Sticker message
			case *linebot.StickerMessage:
				var kw string
				for _, k := range message.Keywords {
					kw = kw + "," + k
				}

				outStickerResult := fmt.Sprintf("收到貼圖訊息: %s, pkg: %s kw: %s  text: %s", message.StickerID, message.PackageID, kw, message.Text)
				if _, err = bot.ReplyMessage(event.ReplyToken, linebot.NewTextMessage(outStickerResult)).Do(); err != nil {
					log.Print(err)
				}
			}
		}
	}
}