source: code/trunk/msgstore.go@ 771

package soju

import (
        "bytes"
        "context"
        "encoding/base64"
        "fmt"
        "time"

        "git.sr.ht/~sircmpwn/go-bare"
        "gopkg.in/irc.v3"
)

// messageStore is a per-user store for IRC messages.
type messageStore interface {
        Close() error
        // LastMsgID queries the last message ID for the given network, entity and
        // date. The message ID returned may not refer to a valid message, but can be
        // used in history queries.
        LastMsgID(network *Network, entity string, t time.Time) (string, error)
        // LoadLatestID queries the latest non-event messages for the given network,
        // entity and date, up to a count of limit messages, sorted from oldest to newest.
        LoadLatestID(ctx context.Context, network *Network, entity, id string, limit int) ([]*irc.Message, error)
        Append(network *Network, entity string, msg *irc.Message) (id string, err error)
}

type chatHistoryTarget struct {
        Name          string
        LatestMessage time.Time
}

// chatHistoryMessageStore is a message store that supports chat history
// operations.
type chatHistoryMessageStore interface {
        messageStore

        // ListTargets lists channels and nicknames by time of the latest message.
        // It returns up to limit targets, starting from start and ending on end,
        // both excluded. end may be before or after start.
        // If events is false, only PRIVMSG/NOTICE messages are considered.
        ListTargets(ctx context.Context, network *Network, start, end time.Time, limit int, events bool) ([]chatHistoryTarget, error)
        // LoadBeforeTime loads up to limit messages before start down to end. The
        // returned messages must be between and excluding the provided bounds.
        // end is before start.
        // If events is false, only PRIVMSG/NOTICE messages are considered.
        LoadBeforeTime(ctx context.Context, network *Network, entity string, start, end time.Time, limit int, events bool) ([]*irc.Message, error)
        // LoadBeforeTime loads up to limit messages after start up to end. The
        // returned messages must be between and excluding the provided bounds.
        // end is after start.
        // If events is false, only PRIVMSG/NOTICE messages are considered.
        LoadAfterTime(ctx context.Context, network *Network, entity string, start, end time.Time, limit int, events bool) ([]*irc.Message, error)
}

type msgIDType uint

const (
        msgIDNone msgIDType = iota
        msgIDMemory
        msgIDFS
)

const msgIDVersion uint = 0

type msgIDHeader struct {
        Version uint
        Network bare.Int
        Target  string
        Type    msgIDType
}

type msgIDBody interface {
        msgIDType() msgIDType
}

func formatMsgID(netID int64, target string, body msgIDBody) string {
        var buf bytes.Buffer
        w := bare.NewWriter(&buf)

        header := msgIDHeader{
                Version: msgIDVersion,
                Network: bare.Int(netID),
                Target:  target,
                Type:    body.msgIDType(),
        }
        if err := bare.MarshalWriter(w, &header); err != nil {
                panic(err)
        }
        if err := bare.MarshalWriter(w, body); err != nil {
                panic(err)
        }
        return base64.RawURLEncoding.EncodeToString(buf.Bytes())
}

func parseMsgID(s string, body msgIDBody) (netID int64, target string, err error) {
        b, err := base64.RawURLEncoding.DecodeString(s)
        if err != nil {
                return 0, "", fmt.Errorf("invalid internal message ID: %v", err)
        }

        r := bare.NewReader(bytes.NewReader(b))

        var header msgIDHeader
        if err := bare.UnmarshalBareReader(r, &header); err != nil {
                return 0, "", fmt.Errorf("invalid internal message ID: %v", err)
        }

        if header.Version != msgIDVersion {
                return 0, "", fmt.Errorf("invalid internal message ID: got version %v, want %v", header.Version, msgIDVersion)
        }

        if body != nil {
                typ := body.msgIDType()
                if header.Type != typ {
                        return 0, "", fmt.Errorf("invalid internal message ID: got type %v, want %v", header.Type, typ)
                }

                if err := bare.UnmarshalBareReader(r, body); err != nil {
                        return 0, "", fmt.Errorf("invalid internal message ID: %v", err)
                }
        }

        return int64(header.Network), header.Target, nil
}