diff options
Diffstat (limited to 'internal/processing')
| -rw-r--r-- | internal/processing/markers/get.go | 54 | ||||
| -rw-r--r-- | internal/processing/markers/markers.go | 35 | ||||
| -rw-r--r-- | internal/processing/markers/update.go | 48 | ||||
| -rw-r--r-- | internal/processing/processor.go | 7 | 
4 files changed, 144 insertions, 0 deletions
diff --git a/internal/processing/markers/get.go b/internal/processing/markers/get.go new file mode 100644 index 000000000..dd94376f1 --- /dev/null +++ b/internal/processing/markers/get.go @@ -0,0 +1,54 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program.  If not, see <http://www.gnu.org/licenses/>. + +package markers + +import ( +	"context" +	"errors" +	"fmt" + +	apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model" +	"github.com/superseriousbusiness/gotosocial/internal/db" +	"github.com/superseriousbusiness/gotosocial/internal/gtserror" +	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" +	"github.com/superseriousbusiness/gotosocial/internal/typeutils" +) + +// Get returns an API model for the markers of the requested timelines. +// If a timeline marker hasn't been set yet, it's not included in the response. +func (p *Processor) Get(ctx context.Context, account *gtsmodel.Account, names []apimodel.MarkerName) (*apimodel.Marker, gtserror.WithCode) { +	markers := make([]*gtsmodel.Marker, 0, len(names)) +	for _, name := range names { +		marker, err := p.state.DB.GetMarker(ctx, account.ID, typeutils.APIMarkerNameToMarkerName(name)) +		if err != nil { +			if errors.Is(err, db.ErrNoEntries) { +				continue +			} +			// Real database error. +			return nil, gtserror.NewErrorInternalError(err) +		} +		markers = append(markers, marker) +	} + +	apiMarker, err := p.tc.MarkersToAPIMarker(ctx, markers) +	if err != nil { +		return nil, gtserror.NewErrorInternalError(fmt.Errorf("error converting marker to api: %w", err)) +	} + +	return apiMarker, nil +} diff --git a/internal/processing/markers/markers.go b/internal/processing/markers/markers.go new file mode 100644 index 000000000..7a1215241 --- /dev/null +++ b/internal/processing/markers/markers.go @@ -0,0 +1,35 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program.  If not, see <http://www.gnu.org/licenses/>. + +package markers + +import ( +	"github.com/superseriousbusiness/gotosocial/internal/state" +	"github.com/superseriousbusiness/gotosocial/internal/typeutils" +) + +type Processor struct { +	state *state.State +	tc    typeutils.TypeConverter +} + +func New(state *state.State, tc typeutils.TypeConverter) Processor { +	return Processor{ +		state: state, +		tc:    tc, +	} +} diff --git a/internal/processing/markers/update.go b/internal/processing/markers/update.go new file mode 100644 index 000000000..96eb17833 --- /dev/null +++ b/internal/processing/markers/update.go @@ -0,0 +1,48 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program.  If not, see <http://www.gnu.org/licenses/>. + +package markers + +import ( +	"context" +	"errors" +	"fmt" + +	apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model" +	"github.com/superseriousbusiness/gotosocial/internal/db" +	"github.com/superseriousbusiness/gotosocial/internal/gtserror" +	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" +) + +// Update updates the given markers and returns an API model for them. +func (p *Processor) Update(ctx context.Context, markers []*gtsmodel.Marker) (*apimodel.Marker, gtserror.WithCode) { +	for _, marker := range markers { +		if err := p.state.DB.UpdateMarker(ctx, marker); err != nil { +			if errors.Is(err, db.ErrAlreadyExists) { +				return nil, gtserror.NewErrorConflict(err, "marker updated by another client") +			} +			return nil, gtserror.NewErrorInternalError(err) +		} +	} + +	apiMarker, err := p.tc.MarkersToAPIMarker(ctx, markers) +	if err != nil { +		return nil, gtserror.NewErrorInternalError(fmt.Errorf("error converting marker to api: %w", err)) +	} + +	return apiMarker, nil +} diff --git a/internal/processing/processor.go b/internal/processing/processor.go index 377f176e5..2f1f43826 100644 --- a/internal/processing/processor.go +++ b/internal/processing/processor.go @@ -30,6 +30,7 @@ import (  	"github.com/superseriousbusiness/gotosocial/internal/processing/admin"  	"github.com/superseriousbusiness/gotosocial/internal/processing/fedi"  	"github.com/superseriousbusiness/gotosocial/internal/processing/list" +	"github.com/superseriousbusiness/gotosocial/internal/processing/markers"  	"github.com/superseriousbusiness/gotosocial/internal/processing/media"  	"github.com/superseriousbusiness/gotosocial/internal/processing/report"  	"github.com/superseriousbusiness/gotosocial/internal/processing/search" @@ -59,6 +60,7 @@ type Processor struct {  	admin    admin.Processor  	fedi     fedi.Processor  	list     list.Processor +	markers  markers.Processor  	media    media.Processor  	report   report.Processor  	search   search.Processor @@ -84,6 +86,10 @@ func (p *Processor) List() *list.Processor {  	return &p.list  } +func (p *Processor) Markers() *markers.Processor { +	return &p.markers +} +  func (p *Processor) Media() *media.Processor {  	return &p.media  } @@ -140,6 +146,7 @@ func NewProcessor(  	processor.admin = admin.New(state, tc, mediaManager, federator.TransportController(), emailSender)  	processor.fedi = fedi.New(state, tc, federator, filter)  	processor.list = list.New(state, tc) +	processor.markers = markers.New(state, tc)  	processor.media = media.New(state, tc, mediaManager, federator.TransportController())  	processor.report = report.New(state, tc)  	processor.timeline = timeline.New(state, tc, filter)  | 
