M main.go → main.go

@@ -43,6 +43,8 @@ for _, line := range split {
 			c.Write(line)
 		}
 	} else if err != nil {
+		msg.Params = []string{target, response}
+		c.WriteMessage(&msg)
 		log.Printf("error: %v", err)
 	} else {
 		for _, line := range split {

A plugins/profile.go

@@ -0,0 +1,83 @@ 
+// User profile plugins: desktop, homescreen, selfie, battlestation, etc.
+package plugins
+
+import (
+	"fmt"
+	"strings"
+
+	"git.icyphox.sh/paprika/database"
+	"gopkg.in/irc.v3"
+)
+
+func init() {
+	Register(Profile{})
+}
+
+type Profile struct{}
+
+func (Profile) Triggers() []string {
+	return []string{
+		".selfie",
+		".desktop", ".dt",
+		".hs", ".homescreen", ".home",
+		".bs", ".battlestation", ".keyb",
+	}
+}
+
+func (Profile) Execute(m *irc.Message) (string, error) {
+	parts := strings.SplitN(m.Trailing(), " ", 2)
+	trigger := parts[0]
+
+	var key string
+
+	switch trigger {
+	case ".desktop", ".dt":
+		key = "desktop"
+	case ".hs", ".homescreen", ".home":
+		key = "homescreen"
+	case ".bs", ".battlestation", ".keyb":
+		key = "battlestation"
+	default:
+		// Strip the '.'
+		key = trigger[1:]
+	}
+
+	if len(parts) == 1 {
+		val, err := database.DB.Get([]byte(fmt.Sprintf(
+			"%s/%s",
+			key,
+			strings.ToLower(m.Prefix.Name),
+		)))
+		if err != nil {
+			return fmt.Sprintf(
+				"Error fetching %s. Use '%s <link>' to set it.", key, trigger,
+			), err
+		}
+		return fmt.Sprintf("\x02%s\x02: %s", m.Prefix.Name, string(val)), nil
+	} else if len(parts) == 2 {
+		// Querying @nick's thing.
+		if strings.HasPrefix(parts[1], "@") {
+			val, err := database.DB.Get([]byte(fmt.Sprintf(
+				"%s/%s",
+				key,
+				parts[1][1:],
+			)))
+			if err != nil {
+				return fmt.Sprintf("Error fetching %s", key), err
+			}
+			return fmt.Sprintf("\x02%s\x02: %s", parts[1][1:], string(val)), nil
+		}
+		// User wants to set the thing.
+		value := parts[1]
+
+		err := database.DB.Set(
+			[]byte(fmt.Sprintf("%s/%s", key, strings.ToLower(m.Prefix.Name))),
+			[]byte(value),
+		)
+		if err != nil {
+			return fmt.Sprintf("Error saving %s", key), err
+		}
+		return fmt.Sprintf("Saved your %s successfully", key), nil
+	}
+	return "", nil
+}