summaryrefslogtreecommitdiff
path: root/internal/federation/federatingdb/update.go
blob: e4a4920c805e3ec83d2468aae00ad83b488b79ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
package federatingdb

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"

	"github.com/go-fed/activity/streams"
	"github.com/go-fed/activity/streams/vocab"
	"github.com/sirupsen/logrus"
	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
	"github.com/superseriousbusiness/gotosocial/internal/typeutils"
	"github.com/superseriousbusiness/gotosocial/internal/util"
)

// Update sets an existing entry to the database based on the value's
// id.
//
// Note that Activity values received from federated peers may also be
// updated in the database this way if the Federating Protocol is
// enabled. The client may freely decide to store only the id instead of
// the entire value.
//
// The library makes this call only after acquiring a lock first.
func (f *federatingDB) Update(ctx context.Context, asType vocab.Type) error {
	l := f.log.WithFields(
		logrus.Fields{
			"func":   "Update",
			"asType": asType.GetTypeName(),
		},
	)
	m, err := streams.Serialize(asType)
	if err != nil {
		return err
	}
	b, err := json.Marshal(m)
	if err != nil {
		return err
	}

	l.Debugf("received UPDATE asType %s", string(b))

	targetAcctI := ctx.Value(util.APAccount)
	if targetAcctI == nil {
		// If the target account wasn't set on the context, that means this request didn't pass through the
		// API, but came from inside GtS as the result of another activity on this instance. That being so,
		// we can safely just ignore this activity, since we know we've already processed it elsewhere.
		return nil
	}
	targetAcct, ok := targetAcctI.(*gtsmodel.Account)
	if !ok {
		l.Error("UPDATE: target account was set on context but couldn't be parsed")
	}

	requestingAcctI := ctx.Value(util.APRequestingAccount)
	if targetAcctI == nil {
		l.Error("UPDATE: requesting account wasn't set on context")
	}
	requestingAcct, ok := requestingAcctI.(*gtsmodel.Account)
	if !ok {
		l.Error("UPDATE: requesting account was set on context but couldn't be parsed")
	}

	fromFederatorChanI := ctx.Value(util.APFromFederatorChanKey)
	if fromFederatorChanI == nil {
		l.Error("UPDATE: from federator channel wasn't set on context")
	}
	fromFederatorChan, ok := fromFederatorChanI.(chan gtsmodel.FromFederator)
	if !ok {
		l.Error("UPDATE: from federator channel was set on context but couldn't be parsed")
	}

	typeName := asType.GetTypeName()
	if typeName == gtsmodel.ActivityStreamsApplication ||
		typeName == gtsmodel.ActivityStreamsGroup ||
		typeName == gtsmodel.ActivityStreamsOrganization ||
		typeName == gtsmodel.ActivityStreamsPerson ||
		typeName == gtsmodel.ActivityStreamsService {
		// it's an UPDATE to some kind of account
		var accountable typeutils.Accountable

		switch asType.GetTypeName() {
		case gtsmodel.ActivityStreamsApplication:
			l.Debug("got update for APPLICATION")
			i, ok := asType.(vocab.ActivityStreamsApplication)
			if !ok {
				return errors.New("UPDATE: could not convert type to application")
			}
			accountable = i
		case gtsmodel.ActivityStreamsGroup:
			l.Debug("got update for GROUP")
			i, ok := asType.(vocab.ActivityStreamsGroup)
			if !ok {
				return errors.New("UPDATE: could not convert type to group")
			}
			accountable = i
		case gtsmodel.ActivityStreamsOrganization:
			l.Debug("got update for ORGANIZATION")
			i, ok := asType.(vocab.ActivityStreamsOrganization)
			if !ok {
				return errors.New("UPDATE: could not convert type to organization")
			}
			accountable = i
		case gtsmodel.ActivityStreamsPerson:
			l.Debug("got update for PERSON")
			i, ok := asType.(vocab.ActivityStreamsPerson)
			if !ok {
				return errors.New("UPDATE: could not convert type to person")
			}
			accountable = i
		case gtsmodel.ActivityStreamsService:
			l.Debug("got update for SERVICE")
			i, ok := asType.(vocab.ActivityStreamsService)
			if !ok {
				return errors.New("UPDATE: could not convert type to service")
			}
			accountable = i
		}

		updatedAcct, err := f.typeConverter.ASRepresentationToAccount(accountable, true)
		if err != nil {
			return fmt.Errorf("UPDATE: error converting to account: %s", err)
		}

		if updatedAcct.Domain == f.config.Host {
			// no need to update local accounts
			// in fact, if we do this will break the shit out of things so do NOT
			return nil
		}

		if requestingAcct.URI != updatedAcct.URI {
			return fmt.Errorf("UPDATE: update for account %s was requested by account %s, this is not valid", updatedAcct.URI, requestingAcct.URI)
		}

		updatedAcct.ID = requestingAcct.ID // set this here so the db will update properly instead of trying to PUT this and getting constraint issues
		if err := f.db.UpdateByID(requestingAcct.ID, updatedAcct); err != nil {
			return fmt.Errorf("UPDATE: database error inserting updated account: %s", err)
		}

		fromFederatorChan <- gtsmodel.FromFederator{
			APObjectType:     gtsmodel.ActivityStreamsProfile,
			APActivityType:   gtsmodel.ActivityStreamsUpdate,
			GTSModel:         updatedAcct,
			ReceivingAccount: targetAcct,
		}

	}

	return nil
}