Add keepalive to Client

[rbdr/mobius] / hotline / news.go

package hotline

import (
	"bytes"
	"crypto/rand"
	"encoding/binary"
	"log"
	"sort"
	"time"
)

type ThreadedNews struct {
	Categories map[string]NewsCategoryListData15 `yaml:"Categories"`
}

type NewsCategoryListData15 struct {
	Type     []byte                            `yaml:"Type"`     //Size 2 ; Bundle (2) or category (3)
	Name     string                            `yaml:"Name"`     //
	Articles map[uint32]*NewsArtData           `yaml:"Articles"` // Optional, if Type is Category
	SubCats  map[string]NewsCategoryListData15 `yaml:"SubCats"`
	Count    []byte                            // Article or SubCategory count Size 2
	AddSN    []byte                            // Size 4
	DeleteSN []byte                            // Size 4
	GUID     []byte                            // Size 16
}

func (newscat *NewsCategoryListData15) GetNewsArtListData() NewsArtListData {
	var newsArts []NewsArtList
	var newsArtsPayload []byte

	for i, art := range newscat.Articles {
		ID := make([]byte, 4)
		binary.BigEndian.PutUint32(ID, i)

		newArt := NewsArtList{
			ID:          ID,
			TimeStamp:   art.Date,
			ParentID:    art.ParentArt,
			Flags:       []byte{0, 0, 0, 0},
			FlavorCount: []byte{0, 0},
			Title:       []byte(art.Title),
			Poster:      []byte(art.Poster),
			ArticleSize: art.DataSize(),
		}
		newsArts = append(newsArts, newArt)
	}

	sort.Sort(byID(newsArts))

	for _, v := range newsArts {
		newsArtsPayload = append(newsArtsPayload, v.Payload()...)
	}

	nald := NewsArtListData{
		ID:          []byte{0, 0, 0, 0},
		Name:        []byte{},
		Description: []byte{},
		NewsArtList: newsArtsPayload,
	}

	return nald
}

// NewsArtData repsents a single news article
type NewsArtData struct {
	Title         string `yaml:"Title"`
	Poster        string `yaml:"Poster"`
	Date          []byte `yaml:"Date"`             //size 8
	PrevArt       []byte `yaml:"PrevArt"`          //size 4
	NextArt       []byte `yaml:"NextArt"`          //size 4
	ParentArt     []byte `yaml:"ParentArt"`        //size 4
	FirstChildArt []byte `yaml:"FirstChildArtArt"` //size 4
	DataFlav      []byte `yaml:"DataFlav"`         // "text/plain"
	Data          string `yaml:"Data"`
}

func (art *NewsArtData) DataSize() []byte {
	dataLen := make([]byte, 2)
	binary.BigEndian.PutUint16(dataLen, uint16(len(art.Data)))

	return dataLen
}

type NewsArtListData struct {
	ID          []byte `yaml:"ID"` // Size 4
	Name        []byte `yaml:"Name"`
	Description []byte `yaml:"Description"` // not used?
	NewsArtList []byte // List of articles			Optional (if article count > 0)
}

func (nald *NewsArtListData) Payload() []byte {
	count := make([]byte, 4)
	binary.BigEndian.PutUint32(count, uint32(len(nald.NewsArtList)))

	out := append(nald.ID, count...)
	out = append(out, []byte{uint8(len(nald.Name))}...)
	out = append(out, nald.Name...)
	out = append(out, []byte{uint8(len(nald.Description))}...)
	out = append(out, nald.Description...)
	out = append(out, nald.NewsArtList...)

	return out
}

// NewsArtList is a summarized ver sion of a NewArtData record for display in list view
type NewsArtList struct {
	ID          []byte // Size 4
	TimeStamp   []byte // Year (2 bytes), milliseconds (2 bytes) and seconds (4 bytes)
	ParentID    []byte // Size 4
	Flags       []byte // Size 4
	FlavorCount []byte // Size 2
	// Title size	1
	Title []byte // string
	// Poster size	1
	// Poster	Poster string
	Poster     []byte
	FlavorList []NewsFlavorList
	// Flavor list…			Optional (if flavor count > 0)
	ArticleSize []byte // Size 2
}

type byID []NewsArtList

func (s byID) Len() int {
	return len(s)
}
func (s byID) Swap(i, j int) {
	s[i], s[j] = s[j], s[i]
}
func (s byID) Less(i, j int) bool {
	return binary.BigEndian.Uint32(s[i].ID) < binary.BigEndian.Uint32(s[j].ID)
}

func (nal *NewsArtList) Payload() []byte {
	out := append(nal.ID, nal.TimeStamp...)
	out = append(out, nal.ParentID...)
	out = append(out, nal.Flags...)

	out = append(out, []byte{0, 1}...)

	out = append(out, []byte{uint8(len(nal.Title))}...)
	out = append(out, nal.Title...)
	out = append(out, []byte{uint8(len(nal.Poster))}...)
	out = append(out, nal.Poster...)
	out = append(out, []byte{0x0a, 0x74, 0x65, 0x78, 0x74, 0x2f, 0x70, 0x6c, 0x61, 0x69, 0x6e}...) // TODO: wat?
	out = append(out, nal.ArticleSize...)

	return out
}

type NewsFlavorList struct {
	// Flavor size	1
	// Flavor text	size		MIME type string
	// Article size	2
}

func (newscat *NewsCategoryListData15) Payload() []byte {
	count := make([]byte, 2)
	binary.BigEndian.PutUint16(count, uint16(len(newscat.Articles)+len(newscat.SubCats)))

	out := append(newscat.Type, count...)

	if bytes.Equal(newscat.Type, []byte{0, 3}) {
		// Generate a random GUID
		b := make([]byte, 16)
		_, err := rand.Read(b)
		if err != nil {
			log.Fatal(err)
		}

		out = append(out, b...)                  // GUID
		out = append(out, []byte{0, 0, 0, 1}...) // Add SN (TODO: not sure what this is)
		out = append(out, []byte{0, 0, 0, 2}...) // Delete SN (TODO: not sure what this is)
	}

	out = append(out, newscat.nameLen()...)
	out = append(out, []byte(newscat.Name)...)

	return out
}

// ReadNewsCategoryListData parses a byte slice into a NewsCategoryListData15 struct
// For use on the client side
func ReadNewsCategoryListData(payload []byte) NewsCategoryListData15 {
	ncld := NewsCategoryListData15{
		Type:  payload[0:2],
		Count: payload[2:4],
	}

	if bytes.Equal(ncld.Type, []byte{0, 3}) {
		ncld.GUID = payload[4:20]
		ncld.AddSN = payload[20:24]
		ncld.AddSN = payload[24:28]
		ncld.Name = string(payload[29:])
	} else {
		ncld.Name = string(payload[5:])
	}

	return ncld
}

func (newscat *NewsCategoryListData15) nameLen() []byte {
	return []byte{uint8(len(newscat.Name))}
}

type NewsPath struct {
	Paths []string
}

func (np *NewsPath) Payload() []byte {
	var out []byte

	count := make([]byte, 2)
	binary.BigEndian.PutUint16(count, uint16(len(np.Paths)))

	out = append(out, count...)
	for _, p := range np.Paths {
		pLen := byte(len(p))
		out = append(out, []byte{0, 0}...) // what is this?
		out = append(out, pLen)
		out = append(out, []byte(p)...)
	}

	return out
}

func ReadNewsPath(newsPath []byte) []string {
	if len(newsPath) == 0 {
		return []string{}
	}
	pathCount := binary.BigEndian.Uint16(newsPath[0:2])

	pathData := newsPath[2:]
	var paths []string

	for i := uint16(0); i < pathCount; i++ {
		pathLen := pathData[2]
		paths = append(paths, string(pathData[3:3+pathLen]))

		pathData = pathData[pathLen+3:]
	}

	return paths
}

func (s *Server) GetNewsCatByPath(paths []string) map[string]NewsCategoryListData15 {
	cats := s.ThreadedNews.Categories
	for _, path := range paths {
		cats = cats[path].SubCats
	}
	return cats
}

// News article date field contains this structure:
// Year					2
// Milliseconds	2 (seriously?)
// Seconds			4
func NewsDate() []byte {
	t := time.Now()
	ms := []byte{0, 0}
	seconds := []byte{0, 0, 0, 0}

	year := []byte{0, 0}
	binary.BigEndian.PutUint16(year, uint16(t.Year()))

	yearStart := time.Date(t.Year(), time.January, 1, 0, 0, 0, 0, time.Local)

	binary.BigEndian.PutUint32(seconds, uint32(t.Sub(yearStart).Seconds()))

	date := append(year, ms...)
	date = append(date, seconds...)

	return date
}