Create & Init Project...

app/job/main/archive/dao/email/BUILD

@@ -0,0 +1,56 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = [
+        "dao_test.go",
+        "email_test.go",
+    ],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/job/main/archive/conf:go_default_library",
+        "//app/job/main/archive/model/result:go_default_library",
+        "//app/service/main/archive/api:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "dao.go",
+        "email.go",
+    ],
+    importpath = "go-common/app/job/main/archive/dao/email",
+    tags = ["automanaged"],
+    deps = [
+        "//app/job/main/archive/conf:go_default_library",
+        "//app/job/main/archive/model/result:go_default_library",
+        "//app/service/main/archive/api:go_default_library",
+        "//app/service/main/archive/model/archive:go_default_library",
+        "//library/log:go_default_library",
+        "//vendor/gopkg.in/gomail.v2:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/job/main/archive/dao/email/dao.go

@@ -0,0 +1,22 @@
+package email
+
+import (
+	gomail "gopkg.in/gomail.v2"
+
+	"go-common/app/job/main/archive/conf"
+)
+
+// Dao is redis dao.
+type Dao struct {
+	c     *conf.Config
+	email *gomail.Dialer
+}
+
+// New is new redis dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		c:     c,
+		email: gomail.NewDialer(c.Mail.Host, c.Mail.Port, c.Mail.Username, c.Mail.Password),
+	}
+	return d
+}

app/job/main/archive/dao/email/dao_test.go

@@ -0,0 +1,19 @@
+package email
+
+import (
+	"flag"
+	"path/filepath"
+
+	"go-common/app/job/main/archive/conf"
+)
+
+var (
+	d *Dao
+)
+
+func init() {
+	dir, _ := filepath.Abs("../../cmd/archive-job-test.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	d = New(conf.Conf)
+}

app/job/main/archive/dao/email/email.go

@@ -0,0 +1,87 @@
+package email
+
+import (
+	"encoding/json"
+	"fmt"
+	"strconv"
+
+	"go-common/app/job/main/archive/model/result"
+	"go-common/app/service/main/archive/api"
+	"go-common/app/service/main/archive/model/archive"
+	"go-common/library/log"
+
+	gomail "gopkg.in/gomail.v2"
+)
+
+var (
+	_states = map[int]string{
+		0:  "开放浏览",
+		-1: "待审",
+		-2: "打回稿件回收站",
+		-3: "网警锁定删除",
+		-4: "锁定稿件",
+		// -5:   "锁定稿件开放浏览",
+		-6: "修复待审",
+		-7: "暂缓审核",
+		// -8:   "补档待审",
+		-9:  "等待转码",
+		-10: "延迟发布",
+		-11: "视频源待修",
+		// -12:  "上传失败",
+		-13: "允许评论待审",
+		// -14:  "临时回收站",
+		-15:  "分发中",
+		-16:  "转码失败",
+		-30:  "创建已提交",
+		-40:  "UP主定时发布",
+		-100: "UP主删除",
+	}
+)
+
+func stateDescribe(state int) string {
+	des, ok := _states[state]
+	if ok {
+		return des
+	}
+	return strconv.Itoa(state)
+}
+
+const (
+	_bangumiMailSub = "番剧稿件《%s》状态变更"
+	_movieMailSub   = "电影稿件《%s》状态变更"
+	_mailBody       = `
+稿件标题：%s
+稿件状态：%s => %s
+其他变更：%s => %s
+稿件地址：http://www.bilibili.com/video/av%d
+审核后台：http://manager.bilibili.co/#!/archive/modify/%d`
+)
+
+// PGCNotifyMail notify pgc mail
+func (d *Dao) PGCNotifyMail(a *api.Arc, nw *result.Archive, old *result.Archive) {
+	msg := gomail.NewMessage()
+	msg.SetHeader("From", d.c.Mail.Username)
+	switch {
+	case a.AttrVal(archive.AttrBitIsBangumi) == archive.AttrYes:
+		msg.SetHeader("To", d.c.Mail.Bangumi...)
+		msg.SetHeader("Subject", fmt.Sprintf(_bangumiMailSub, a.Title))
+	case a.AttrVal(archive.AttrBitIsMovie) == archive.AttrYes:
+		msg.SetHeader("To", d.c.Mail.Movie...)
+		msg.SetHeader("Subject", fmt.Sprintf(_movieMailSub, a.Title))
+	default:
+		return
+	}
+	obs, _ := json.Marshal(old)
+	nbs, _ := json.Marshal(nw)
+
+	if old.State != nw.State {
+		oldState, newState := stateDescribe(old.State), stateDescribe(nw.State)
+		msg.SetBody("text/plain", fmt.Sprintf(_mailBody, a.Title, oldState, newState, obs, nbs, a.Aid, a.Aid))
+	} else {
+		state := stateDescribe(int(a.State))
+		msg.SetBody("text/plain", fmt.Sprintf(_mailBody, a.Title, state, state, obs, nbs, a.Aid, a.Aid))
+	}
+	if err := d.email.DialAndSend(msg); err != nil {
+		log.Error("s.email.DialAndSend(aid: %d) error(%v)", a.Aid, err)
+	}
+}

app/job/main/archive/dao/email/email_test.go

@@ -0,0 +1,15 @@
+package email
+
+import (
+	"go-common/app/job/main/archive/model/result"
+	"go-common/app/service/main/archive/api"
+	"testing"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func Test_PGCNotifyMail(t *testing.T) {
+	Convey("PGCNotifyMail", t, func() {
+		d.PGCNotifyMail(&api.Arc{}, &result.Archive{}, &result.Archive{})
+	})
+}