structs.go

package main

import (
	"fmt"
	"gopkg.in/irc.v3"
	"strconv"
	"time"
)

///

type User struct {
	ID   int64 `gorm:"primaryKey"`
	Name string
}

type Channel struct {
	ID   int64 `gorm:"primaryKey"`
	Name string
}

type Message struct {
	ID   string    `gorm:"primaryKey"`
	Body string    `gorm:"not null"`
	Mod  bool      `gorm:"not null"`
	Date time.Time `gorm:"not null"`

	// optional:
	Deleted        *time.Time `gorm:"default:null"`
	ReplyMessageID *string    `gorm:"default:null"`

	ChannelID int64 `gorm:"not null"`
	AuthorID  int64 `gorm:"not null"`

	// Belongings
	Author  *User
	Channel *Channel
}

func (m *Message) String() string {
	var mod string
	if m.Mod {
		mod = "👮 "
	}
	return fmt.Sprintf("(%s) %s[%s]: %s", m.Channel.Name, mod, m.Author.Name, m.Body)
}

type ListeningChannel struct {
	ChannelName string `gorm:"primaryKey"`
	Active      bool   `gorm:"default:true"`
}

///

type Config struct {
	PostgresDSN string `koanf:"postgres_dsn"`
	//
	TwitchNick string `koanf:"twitch_nick"`
	TwitchUser string `koanf:"twitch_user"`
	TwitchName string `koanf:"twitch_name"`
	TwitchPass string `koanf:"twitch_pass"`
	//
	WebBind string `koanf:"web_bind"`
}

///

// ParseIRCMessage expects {msg} to have {Command == PRIVMSG}
func ParseIRCMessage(msg *irc.Message) (*Message, error) {
	// meta
	var (
		userName       = msg.Name
		channelName    = msg.Param(0)
		body           = msg.Param(1)
		messageID      string
		userID         int64
		channelID      int64
		isMod          bool
		replyMessageID string
	)
	if len(channelName) > 0 {
		channelName = channelName[1:]
	}

	// auxiliary
	var (
		userIDStr    string
		channelIDStr string
		modStr       string
		ok           bool
		err          error
	)

	if userIDStr, ok = msg.GetTag("user-id"); !ok {
		return nil, fmt.Errorf("userid was empty for user %v", userName)
	}
	if userID, err = strconv.ParseInt(userIDStr, 10, 64); err != nil {
		return nil, fmt.Errorf("userid %v could not be converted to int64", userIDStr)
	}

	if channelIDStr, ok = msg.GetTag("room-id"); !ok {
		return nil, fmt.Errorf("roomid was empty for channel %v", channelName)
	}
	if channelID, err = strconv.ParseInt(channelIDStr, 10, 64); err != nil {
		return nil, fmt.Errorf("roomid %v could not be converted to int64", channelIDStr)
	}

	if messageID, ok = msg.GetTag("id"); !ok {
		return nil, fmt.Errorf("messageid was empty for body '%v' by %v", body, userName)
	}

	if modStr, ok = msg.GetTag("mod"); !ok {
		return nil, fmt.Errorf("mod was empty for body '%v' by %v", body, userName)
	}
	isMod = modStr == "1"

	// reply
	replyMessageID, _ = msg.GetTag("reply-parent-msg-id")

	return &Message{
		ID:             messageID,
		Body:           body,
		Mod:            isMod,
		Date:           time.Now(),
		Deleted:        nil,
		ReplyMessageID: strOrNil(replyMessageID),
		ChannelID:      channelID,
		AuthorID:       userID,
		Author: &User{
			ID:   userID,
			Name: userName,
		},
		Channel: &Channel{
			ID:   channelID,
			Name: channelName,
		},
	}, nil
}