SMTP sender

smtp.go

@@ -0,0 +1,183 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"html"
+	"io/ioutil"
+	"mime/quotedprintable"
+	"net/smtp"
+	"os"
+	"strings"
+	"time"
+
+	"golang-fave/engine/mysqlpool"
+	"golang-fave/engine/sqlw"
+	"golang-fave/engine/wrapper/config"
+	"golang-fave/utils"
+)
+
+func smtp_send(host, port, user, pass, subject, msg string, receivers []string) error {
+	header := make(map[string]string)
+	header["From"] = user
+	header["To"] = strings.Join(receivers, ", ")
+	header["Subject"] = subject
+	header["MIME-Version"] = "1.0"
+	header["Content-Type"] = fmt.Sprintf("%s; charset=\"utf-8\"", "text/html")
+	header["Content-Transfer-Encoding"] = "quoted-printable"
+	header["Content-Disposition"] = "inline"
+
+	message := ""
+	for key, value := range header {
+		message += fmt.Sprintf("%s: %s\r\n", key, value)
+	}
+
+	var encodedMessage bytes.Buffer
+	finalMessage := quotedprintable.NewWriter(&encodedMessage)
+	finalMessage.Write([]byte(msg))
+	finalMessage.Close()
+	message += "\r\n" + encodedMessage.String()
+
+	return smtp.SendMail(
+		host+":"+port,
+		smtp.PlainAuth("", user, pass, host),
+		user,
+		receivers,
+		[]byte(message),
+	)
+}
+
+func smtp_prepare(db *sqlw.DB, conf *config.Config) {
+	rows, err := db.Query(
+		`SELECT
+			notify_mail.id,
+			notify_mail.email,
+			notify_mail.subject,
+			notify_mail.message
+		FROM
+			notify_mail
+		WHERE
+			notify_mail.status = 2
+		ORDER BY
+			notify_mail.id ASC
+		;`,
+	)
+	if err == nil {
+		defer rows.Close()
+		values := make([]string, 4)
+		scan := make([]interface{}, len(values))
+		for i := range values {
+			scan[i] = &values[i]
+		}
+		for rows.Next() {
+			err = rows.Scan(scan...)
+			if err == nil {
+				if _, err := db.Exec(
+					`UPDATE notify_mail SET status = 3 WHERE id = ?;`,
+					utils.StrToInt(string(values[0])),
+				); err != nil {
+					go func(db *sqlw.DB, conf *config.Config, id int, subject, msg string, receivers []string) {
+						if err := smtp_send(
+							(*conf).SMTP.Host,
+							utils.IntToStr((*conf).SMTP.Port),
+							(*conf).SMTP.Login,
+							(*conf).SMTP.Password,
+							subject,
+							msg,
+							receivers,
+						); err == nil {
+							if _, err := db.Exec(
+								`UPDATE notify_mail SET status = 1 WHERE id = ?;`,
+								id,
+							); err != nil {
+								fmt.Printf("Smtp send error (sql, success): %v\n", err)
+							}
+						} else {
+							if _, err := db.Exec(
+								`UPDATE notify_mail SET error = ?, status = 0 WHERE id = ?;`,
+								err.Error(),
+								id,
+							); err != nil {
+								fmt.Printf("Smtp send error (sql, error): %v\n", err)
+							}
+						}
+					}(
+						db,
+						conf,
+						utils.StrToInt(string(values[0])),
+						html.EscapeString(string(values[2])),
+						html.EscapeString(string(values[3])),
+						[]string{html.EscapeString(string(values[1]))},
+					)
+				}
+			}
+		}
+	}
+}
+
+func smtp_process(dir, host string, mp *mysqlpool.MySqlPool) {
+	db := mp.Get(host)
+	if db != nil {
+		conf := config.ConfigNew()
+		if err := conf.ConfigRead(strings.Join([]string{dir, "config", "config.json"}, string(os.PathSeparator))); err == nil {
+			if (*conf).SMTP.Host != "" && (*conf).SMTP.Login != "" && (*conf).SMTP.Password != "" {
+				if err := db.Ping(); err == nil {
+					smtp_prepare(db, conf)
+				}
+			}
+		} else {
+			fmt.Printf("Smtp error (config): %v\n", err)
+		}
+	}
+}
+
+func smtp_loop(www_dir string, stop chan bool, mp *mysqlpool.MySqlPool) {
+	dirs, err := ioutil.ReadDir(www_dir)
+	if err == nil {
+		for _, dir := range dirs {
+			select {
+			case <-stop:
+				break
+			default:
+				if mp != nil {
+					target_dir := strings.Join([]string{www_dir, dir.Name()}, string(os.PathSeparator))
+					if utils.IsDirExists(target_dir) {
+						smtp_process(target_dir, dir.Name(), mp)
+					}
+				}
+			}
+		}
+	}
+}
+
+func smtp_start(www_dir string, mp *mysqlpool.MySqlPool) (chan bool, chan bool) {
+	ch := make(chan bool)
+	stop := make(chan bool)
+	go func() {
+		for {
+			select {
+			case <-time.After(5 * time.Second):
+				// Run every 5 seconds
+				smtp_loop(www_dir, stop, mp)
+			case <-ch:
+				ch <- true
+				return
+			}
+		}
+	}()
+	return ch, stop
+}
+
+func smtp_stop(ch, stop chan bool) {
+	for {
+		select {
+		case stop <- true:
+		case ch <- true:
+			<-ch
+			return
+		case <-time.After(3 * time.Second):
+			fmt.Println("Smtp error: force exit by timeout after 3 seconds")
+			return
+		}
+	}
+}