118 lines
2.9 KiB
Go
118 lines
2.9 KiB
Go
package toot
|
|
|
|
import "fmt"
|
|
import "encoding/json"
|
|
import "errors"
|
|
import "strings"
|
|
import "git.eeqj.de/sneak/feta/jsonapis"
|
|
|
|
//import "github.com/davecgh/go-spew/spew"
|
|
import "github.com/rs/zerolog/log"
|
|
|
|
//import "encoding/hex"
|
|
import mh "github.com/multiformats/go-multihash"
|
|
import mhopts "github.com/multiformats/go-multihash/opts"
|
|
|
|
// Hash is a type for storing a string-based base58 multihash of a
|
|
// toot's identity
|
|
type Hash string
|
|
|
|
// Toot is an object we use internally for storing a discovered toot
|
|
type Toot struct {
|
|
Original []byte
|
|
Parsed *jsonapis.APISerializedToot
|
|
Hash Hash
|
|
FromHost string
|
|
}
|
|
|
|
// NewTootCollectionFromMastodonAPIResponse takes a byte array from a masto
|
|
// api response and provides you with a nice array of pointers to parsed
|
|
// toots
|
|
func NewTootCollectionFromMastodonAPIResponse(in []byte, hostname string) ([]*Toot, error) {
|
|
var rt []json.RawMessage
|
|
err := json.Unmarshal(in, &rt)
|
|
if err != nil {
|
|
return nil, errors.New("unable to parse api response")
|
|
}
|
|
|
|
var tc []*Toot
|
|
|
|
// iterate over rawtoots from api
|
|
for _, item := range rt {
|
|
parsed := new(jsonapis.APISerializedToot)
|
|
err := json.Unmarshal(item, parsed)
|
|
if err != nil {
|
|
log.Error().Msg("unable to parse toot, skipping")
|
|
continue
|
|
}
|
|
t := new(Toot)
|
|
t.Parsed = parsed
|
|
o, err := item.MarshalJSON()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
t.Original = o
|
|
t.FromHost = hostname
|
|
t.calcHash()
|
|
tc = append(tc, t)
|
|
}
|
|
return tc, nil
|
|
}
|
|
|
|
func (t *Toot) String() string {
|
|
return fmt.Sprintf("%#v", t)
|
|
}
|
|
|
|
func (t *Toot) multiHash(in []byte) string {
|
|
opts := new(mhopts.Options)
|
|
opts.Algorithm = "sha2-256"
|
|
opts.Encoding = "base58"
|
|
var found bool
|
|
opts.AlgorithmCode, found = mh.Names[opts.Algorithm]
|
|
if !found {
|
|
panic("oops")
|
|
}
|
|
opts.Length = mh.DefaultLengths[opts.AlgorithmCode]
|
|
r := strings.NewReader(string(in))
|
|
h, err := opts.Multihash(r)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return h.B58String()
|
|
}
|
|
|
|
// DiskStoragePath is a helper function on a Toot that allows it to provide
|
|
// a storage path on disk. This should probably be moved into the FSStorage
|
|
// backend instead. FIXME
|
|
// It's here because it's a pure function that just formats its own toot attributes
|
|
// into a string.
|
|
func (t *Toot) DiskStoragePath() string {
|
|
// FIXME make this error if fields are missing
|
|
// '/YYYYMMDD/example.com/username/YYYY-MM-DD.HHMMSS.username@fromHost.multihash.json'
|
|
return fmt.Sprintf("%s/%s/%s/%s.%s@%s.%s.json",
|
|
t.Parsed.CreatedAt.Format("20060102"),
|
|
strings.ToLower(t.FromHost),
|
|
t.Parsed.Account.Acct,
|
|
t.Parsed.CreatedAt.Format("2006-01-02.150405"),
|
|
t.Parsed.Account.Acct,
|
|
strings.ToLower(t.FromHost),
|
|
t.Hash,
|
|
)
|
|
}
|
|
|
|
func (t *Toot) identityHashInput() string {
|
|
return fmt.Sprintf(
|
|
"%s.%s.%s.%s.%s",
|
|
t.Parsed.Account.URL,
|
|
t.Parsed.CreatedAt,
|
|
t.Parsed.ID,
|
|
t.Parsed.Content,
|
|
strings.ToLower(t.FromHost),
|
|
)
|
|
}
|
|
|
|
func (t *Toot) calcHash() {
|
|
hi := t.identityHashInput()
|
|
t.Hash = Hash(t.multiHash([]byte(hi)))
|
|
}
|