init: 🎉 first commit

Author: spike014

.github/workflows/send_task.yml

@@ -0,0 +1,25 @@
+name: Distribute tasks on a CRON Schedule
+
+on:
+  schedule:
+    - cron: '50 * * * *'
+
+env:
+  GO_VERSION: 1.19
+
+jobs:
+  distribute:
+    name: Trigger task distributed
+    runs-on: ubuntu-latest
+    steps:
+      - name: Setup Go Environment
+        uses: actions/setup-go@v3
+        with:
+          go-version: ${{ env.GO_VERSION }}
+        id: go
+      - name: Checkout Source Code
+        uses: actions/checkout@v3
+      - name: Go Get dependencies
+        run: go get -v -t -d ./...
+      - name: Distribute task
+        run: go run cmd/main.go --config daily.toml --server ${{ secrets.EMAIL_SERVER }} --user ${{ secrets.EMAIL_USER }} --pass ${{ secrets.EMAIL_PASS }} --from ${{ secrets.EMAIL_FROM }} --to ${{ secrets.EMAIL_TO }}

README.md

@@ -1,2 +1,4 @@
-# daily_bot
-A robot that sents you a daily task and motivates you to cultivate a daily habit.
+# task_bot
+A robot that sends you a daily task and motivates you to cultivate a daily habit, power by `GitHub Actions`.
+
+

cmd/main.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"flag"
+
+	"code-in-gym/task_bot/config"
+	"code-in-gym/task_bot/notify/email"
+	"code-in-gym/task_bot/task"
+)
+
+func main() {
+	var (
+		configPath string
+		server     string
+		user       string
+		pass       string
+		to         string
+		from       string
+	)
+	flag.StringVar(&configPath, "config", "daily.toml", "daily config file path")
+	flag.StringVar(&server, "server", "", "email server")
+	flag.StringVar(&user, "user", "", "email user")
+	flag.StringVar(&pass, "pass", "", "email password")
+	flag.StringVar(&to, "to", "", "email to")
+	flag.StringVar(&from, "from", "", "email from")
+	flag.Parse()
+
+	// randomly choose some tasks
+	config := config.NewConfig[any]()
+	config.Load(configPath)
+
+	// notify
+	// Only support email now
+	emailConfig := email.Config{}
+	emailConfig.NewConfig(
+		server, user, pass, to, from,
+	)
+	config.Notify.Email = append(config.Notify.Email, emailConfig)
+	for _, e := range config.Notify.Email {
+		for k, v := range config.Task {
+			t := task.NewDailyTask(k, v)
+			index := t.Random()
+			e.NotifyFunc(k, t.Detail(index))
+		}
+	}
+}

config/config.go

@@ -0,0 +1,36 @@
+package config
+
+import (
+	"fmt"
+	"log"
+	"os"
+
+	"code-in-gym/task_bot/task"
+	"code-in-gym/task_bot/notify"
+
+	"github.com/BurntSushi/toml"
+)
+
+type Config[T any] struct {
+	Task   map[string]task.Task[T]
+	Notify notify.Config
+}
+
+func NewConfig[T any]() *Config[T] {
+	return &Config[T]{}
+}
+
+// Load load config from file,
+// DO NOT put any sensitive key or secret in it,
+// use GitHub Actions secrets.
+func (c *Config[T]) Load(file string) {
+	if _, err := os.Stat(file); err != nil {
+		log.Fatalf("Config file `%s` DOES NOT exist.", file)
+	}
+
+	_, err := toml.DecodeFile(file, &c)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}

daily.toml

@@ -0,0 +1,26 @@
+# daily task
+[task.leetcode]
+describe = "今天的 LeetCode 题目是 #%v, 试着在20分钟内解决它吧！ You can do it!"
+tasks = [
+  704, 27, 977, 209, 59, # 数组
+  203, 707, 206, 24, 19, 142, # 链表
+  242, 1002, 349, 202, 1, 454, 383, 15, 18, # 哈希表
+  344, 541, 151, 459, # 字符串
+  27, 344, 151, 206, 19, 142, 15, 18, # 双指针法  
+  232, 225, 20, 1047, 150, 239, 347, # 栈与队列 
+  226, 101, 104, 111, 222, 110, 257, 404, 513, 112, 106, 654, 617, 700, 98, 530, 501, 236, 235, 701, 450, 669, 108, 538,# 二叉树 
+  77, 216, 17, 39, 40, 131, 93, 78, 90, 491, 46, 47, 332, 51, 37,# 回溯
+  455, 376, 53, 122, 55, 45, 1005, 134, 135, 860, 406, 406, 452, 435, 763, 56, 738, 968,# 贪心算法 
+  509, 70, 746, 62, 63, 343, 96, 416, 1049, 494, 474, 518, 377, 70, 322, 279, 139, 198, 213, 337, 121, 122, 123, 188, 309, 714, 300, 674, 718, 1143, 1035, 53, 392, 115, 583, 72, 647, 516, # 动态规划
+  739, 496, 503, 42, 84, # 单调栈
+]
+cycle = true
+
+[task.reading]
+describe = "今天的回顾旧书是 %v, 享受阅读的乐趣吧！"
+tasks = [
+    "《这些人那些事》",
+    "《The Minimalist Entrepreneur》",
+    "《侠隐》",
+]
+cycle = false

go.mod

@@ -0,0 +1,10 @@
+module code-in-gym/task_bot
+
+go 1.19
+
+require (
+	github.com/BurntSushi/toml v1.3.0
+	gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df
+)
+
+require gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc // indirect

go.sum

@@ -0,0 +1,6 @@
+github.com/BurntSushi/toml v1.3.0 h1:Ws8e5YmnrGEHzZEzg0YvK/7COGYtTC5PbaH9oSSbgfA=
+github.com/BurntSushi/toml v1.3.0/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=
+gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc h1:2gGKlE2+asNV9m7xrywl36YYNnBG5ZQ0r/BOOxqPpmk=
+gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc/go.mod h1:m7x9LTH6d71AHyAX77c9yqWCCa3UKHcVEj9y7hAtKDk=
+gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df h1:n7WqCuqOuCbNr617RXOY0AWRXxgwEyPp2z+p0+hgMuE=
+gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df/go.mod h1:LRQQ+SO6ZHR7tOkpBDuZnXENFzX8qRjMDMyPD6BRkCw=

notify/base/base.go

@@ -0,0 +1,6 @@
+package base
+
+type DefaultNotify struct {
+	Kind string
+	NotifyFunc func(string, string) error
+}

notify/email/email.go

@@ -0,0 +1,64 @@
+package email
+
+import (
+	"code-in-gym/task_bot/notify/base"
+	"net"
+	"strconv"
+	"strings"
+
+	"gopkg.in/gomail.v2"
+)
+
+type Config struct {
+	base.DefaultNotify
+	Server string
+	User   string
+	Pass   string
+	To     string
+	From   string
+}
+
+func (c *Config) NewConfig(server, user, pass, to, from string) error {
+	c.NotifyFunc = c.SendNotify
+	c.Kind = "email"
+	c.Server = server
+	c.Pass = pass
+	c.User = user
+	c.To = to
+	c.From = from
+	return nil
+}
+
+func (c *Config) SendNotify(subject, content string) error {
+	host, p, err := net.SplitHostPort(c.Server)
+	if err != nil {
+		return err
+	}
+
+	port, err := strconv.Atoi(p)
+	if err != nil {
+		return err
+	}
+
+	email := "Task" + "[" + c.User + "]"
+	if c.From != "" {
+		email = c.From
+	}
+
+	split := func(r rune) bool {
+		return r == ';' || r == ','
+	}
+
+	recipients := strings.FieldsFunc(c.To, split)
+
+	m := gomail.NewMessage()
+	m.SetHeader("From", email)
+	m.SetHeader("To", recipients...)
+	m.SetHeader("Subject", "[TaskBot] " + subject)
+	m.SetBody("text/html; charset=UTF-8", content)
+
+	d := gomail.NewDialer(host, port, c.User, c.Pass)
+	err = d.DialAndSend(m)
+
+	return err
+}

notify/notify.go

@@ -0,0 +1,11 @@
+package notify
+
+import "code-in-gym/task_bot/notify/email"
+
+type Notify interface {
+	SendNotify()
+}
+
+type Config struct {
+	Email []email.Config `toml:"email"`
+}

task/task.go

@@ -0,0 +1,43 @@
+package task
+
+import (
+	"fmt"
+	"math/rand"
+	"time"
+)
+
+type TaskHandle[T any] interface {
+	Random() int
+	Detail() string
+	Finished(int) // remove the finished task, then rewrite the task config file
+}
+
+type Task[T any] struct {
+	Describe string
+	Tasks    []T
+	Cycle    bool
+}
+
+type DailyTask[T any] struct {
+	Task[T]
+	TaskType string
+}
+
+func NewDailyTask[T any](taskType string, tasks Task[T]) *DailyTask[T] {
+	// TODO: remove finished task
+	// if cycle is true and all tasks were finished, 
+	// restart it from the beginning.
+	return &DailyTask[T]{
+		Task:     tasks,
+		TaskType: taskType,
+	}
+}
+
+func (d *DailyTask[T]) Detail(index int) string {
+	return fmt.Sprintf(d.Describe, d.Tasks[index])
+}
+
+func (d *DailyTask[T]) Random() int {
+	rand.Seed(time.Now().Unix())
+	return rand.Intn(len(d.Tasks))
+}