first steps in posting connected. can make reply with no quote or quote OP

do not recommend working on this branch for the time being since
things are being moved around a lot

2 files changed, 102 insertions, 0 deletions

diff --git a/util/blacklist.go b/util/blacklist.go
new file mode 100644
index 0000000..5368037
--- /dev/null
+++ b/util/blacklist.go
@@ -0,0 +1,76 @@
+package util
+
+import (
+	"regexp"
+
+	"github.com/FChannel0/FChannel-Server/config"
+)
+
+type PostBlacklist struct {
+	Id    int
+	Regex string
+}
+
+func DeleteRegexBlacklistDB(id int) error {
+	query := `delete from postblacklist where id=$1`
+
+	_, err := config.DB.Exec(query, id)
+	return err
+}
+
+func GetRegexBlacklistDB() ([]PostBlacklist, error) {
+	var list []PostBlacklist
+
+	query := `select id, regex from postblacklist`
+
+	rows, err := config.DB.Query(query)
+	if err != nil {
+		return list, err
+	}
+
+	defer rows.Close()
+	for rows.Next() {
+		var temp PostBlacklist
+		rows.Scan(&temp.Id, &temp.Regex)
+
+		list = append(list, temp)
+	}
+
+	return list, nil
+}
+
+func IsPostBlacklist(comment string) (bool, error) {
+	postblacklist, err := GetRegexBlacklistDB()
+
+	if err != nil {
+		return false, err
+	}
+
+	for _, e := range postblacklist {
+		re := regexp.MustCompile(e.Regex)
+
+		if re.MatchString(comment) {
+			return true, nil
+		}
+	}
+
+	return false, nil
+}
+
+func WriteRegexBlacklistDB(regex string) error {
+	var re string
+
+	query := `select from postblacklist where regex=$1`
+	if err := config.DB.QueryRow(query, regex).Scan(&re); err != nil {
+		return err
+	}
+
+	if re != "" {
+		return nil
+	}
+
+	query = `insert into postblacklist (regex) values ($1)`
+
+	_, err := config.DB.Exec(query, regex)
+	return err
+}
diff --git a/util/util.go b/util/util.go
index 381db94..ade5eae 100644
--- a/util/util.go
+++ b/util/util.go
@@ -4,6 +4,8 @@ import (
 	"crypto/sha256"
 	"encoding/hex"
 	"fmt"
+	"mime/multipart"
+	"net/http"
 	"os"
 	"regexp"
 	"strings"
@@ -225,3 +227,27 @@ func CreateUniqueID(actor string) (string, error) {
 
 	return newID, nil
 }
+
+func GetFileContentType(out multipart.File) (string, error) {
+	buffer := make([]byte, 512)
+
+	_, err := out.Read(buffer)
+	if err != nil {
+		return "", err
+	}
+
+	out.Seek(0, 0)
+
+	contentType := http.DetectContentType(buffer)
+
+	return contentType, nil
+}
+
+func GetContentType(location string) string {
+	elements := strings.Split(location, ";")
+	if len(elements) > 0 {
+		return elements[0]
+	} else {
+		return location
+	}
+}