Initial Commit

Author: patyhank

.gitignore

@@ -0,0 +1 @@
+test

.idea/.gitignore

@@ -0,0 +1,20 @@
+package common
+
+import "regexp"
+
+const LiveUpdateCommand = "\x1B[D\x1B[C\x1B[4~"
+
+type ViewState int32
+
+const (
+	ViewStateBacking ViewState = iota
+	ViewStateSearching
+	ViewStateReadyViewing
+	ViewStateViewing
+)
+
+// PostRegex [Message Cursor ID Status PushCount]
+var PostRegex = regexp.MustCompile("([> ])(\\d+) ([+Mm -!]) ?(X?\\d+|[爆 ]) ?(\\d{1,2})/ ?(\\d{1,2}) (\\w+)\\s+([\\S\\s]*?)\\s+\\n")
+
+// CommentRegex [Message StrType Author Content Time]
+var CommentRegex = regexp.MustCompile("(→ |→|推|噓) (\\w+): ([\\s\\S]+?)(?: +?| {0})(?:\\S+\\n|(\\d+/\\d+ \\d+:\\d+))")

.idea/pttLink-public.iml

@@ -0,0 +1,21 @@
+package common
+
+import (
+	"github.com/disgoorg/disgo/bot"
+	"github.com/patyhank/ptd/ent"
+	"sync"
+)
+
+type Instance struct {
+	discord      bot.Client
+	currentPost  *ent.PostInfo // 目前瀏覽的貼文
+	previousPost *ent.PostInfo // 上一個瀏覽的貼文，瀏覽下一篇時，於Discord新增已完成標籤
+	db           *ent.Client
+
+	viewState ViewState
+	sync.Mutex
+}
+
+func NewInstance() {
+
+}

common/consts.go

@@ -0,0 +1,47 @@
+package common
+
+import (
+	"github.com/patyhank/ptd/ent"
+	"strings"
+)
+
+// SplitMessageByContentLength 將訊息依照字數分割，避免訊息過長無法發送至Discord
+func SplitMessageByContentLength(collection ent.Messages, wordCount int) (chunks []ent.Messages) {
+	var chunk []*ent.Message
+	var count int
+	for _, data := range collection {
+		count += len(data.Content)
+		if count > wordCount {
+			chunks = append(chunks, chunk)
+			chunk = []*ent.Message{}
+			count = 0
+		}
+		chunk = append(chunk, data)
+	}
+	if len(chunk) > 0 {
+		chunks = append(chunks, chunk)
+	}
+	return
+}
+
+// SplitContentByLength
+// 將字串依照指定的字數分割，並提前於換行處分割
+// 這讓Discord訊息更好閱讀，並且不會截斷連結
+func SplitContentByLength(input string, wordCount int) (chunks []string) {
+	var chunk string
+	var count int
+	collection := strings.Split(input, "\n")
+	for _, data := range collection {
+		count += len([]rune(data))
+		if count >= wordCount {
+			chunks = append(chunks, chunk)
+			chunk = ""
+			count = 0
+		}
+		chunk += data + "\n"
+	}
+	if len(chunk) > 0 {
+		chunks = append(chunks, chunk)
+	}
+	return
+}

common/ptt.go

@@ -0,0 +1,22 @@
+package config
+
+type CommentType int32
+
+const (
+	_ CommentType = iota
+	CommentTypeUpVote
+	CommentTypeDownVote
+	CommentTypeReply
+)
+
+func (c CommentType) String() string {
+	switch c {
+	case CommentTypeUpVote:
+		return "推"
+	case CommentTypeDownVote:
+		return "噓"
+	case CommentTypeReply:
+		return "→"
+	}
+	return ""
+}

common/utils.go

@@ -0,0 +1,65 @@
+package config
+
+import (
+	"github.com/disgoorg/disgo/discord"
+	"github.com/disgoorg/snowflake/v2"
+	"strings"
+)
+
+// Config is the configuration for the bot
+type Config struct {
+	Discord Discord   `yaml:"discord"`
+	PTT     PTTConfig `yaml:"ptt"`
+
+	Searches []SearchConfig `yaml:"searches"`
+}
+
+type Discord struct {
+	Token string `yaml:"token"`
+}
+
+type EmojiConfig struct {
+	UpVote   discord.Emoji `yaml:"up_vote"`
+	DownVote discord.Emoji `yaml:"down_vote"`
+	Reply    discord.Emoji `yaml:"reply"`
+}
+
+type PTTConfig struct {
+	Connection ConnectionConfig `yaml:"connection"`
+}
+
+type ConnectionConfig struct {
+	Username   string `yaml:"username"`
+	Password   string `yaml:"password"`
+	Host       string `yaml:"host"`
+	HostOrigin string `yaml:"host_origin"`
+}
+type SearchConfig struct {
+	Board              string          `json:"board"`
+	SearchVariant      []SearchPattern `json:"search_variant"`
+	TitleSearchVariant []SearchPattern `json:"title_search_variant"`
+
+	PreFillSize int `yaml:"prefill_size"`
+
+	PostMatchRegex string `json:"post_match_regex"`
+	PostTitle      string `json:"post_title"`
+
+	TrackingSeconds int `yaml:"tracking_seconds"`
+
+	ForumChannel snowflake.ID `yaml:"forum_channel,omitempty"`
+	TextChannel  snowflake.ID `yaml:"channel,omitempty"`
+	Emoji        EmojiConfig  `yaml:"emoji"`
+}
+
+type SearchPattern string
+
+func (s SearchPattern) Keys() []string {
+	content := string(s)
+
+	content = strings.ReplaceAll(content, "\r\n", "\n")
+	content = strings.ReplaceAll(content, "\\r", "\r")
+
+	args := strings.Split(content, "|")
+
+	return args
+}

config/comment.go

@@ -0,0 +1,46 @@
+package config
+
+import (
+	"github.com/disgoorg/disgo/discord"
+)
+
+var DefaultConfig = Config{
+	Discord: Discord{
+		Token: "",
+	},
+	PTT: PTTConfig{
+		Connection: ConnectionConfig{
+			Username:   "user",
+			Password:   "pass",
+			Host:       "wss://ws.ptt.cc/bbs",
+			HostOrigin: "https://term.ptt.cc",
+		},
+	},
+	Searches: []SearchConfig{
+		{
+			Board:              "c_chat",
+			SearchVariant:      []SearchPattern{"/holo\\r/直播\\r", "/holo\\r/間直播\\r", "/holo\\r/直播單\\r"}, // 不建議超過3避免無法搜尋文章日期等
+			TitleSearchVariant: []SearchPattern{"/holo\\r/直播\\r", "/holo\\r/間直播\\r", "/holo\\r/直播單\\r"},
+			TrackingSeconds:    600,
+			PostMatchRegex:     "\\[\\S+?] \\[Vtub] {1,2}Hololive ([晚日])間直播單(?:（| \\()(\\d+)(?:）|\\))",
+			PostTitle:          "%[2]s-%[1]s間直播串",
+
+			ForumChannel: 0,
+			TextChannel:  0,
+			Emoji: EmojiConfig{
+				UpVote: discord.Emoji{
+					Name: "push",
+					ID:   1033734367937310720,
+				},
+				DownVote: discord.Emoji{
+					Name: "boo",
+					ID:   1033734366330892428,
+				},
+				Reply: discord.Emoji{
+					Name: "addon",
+					ID:   1033734364950953994,
+				},
+			},
+		},
+	},
+}