mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-10-31 22:40:01 +00:00
[feature] Allow users to export data via the settings panel (#3140)
* [feature] Allow users to export data via the settings panel * rename/move some stuff
This commit is contained in:
parent
43519324b3
commit
38f041cea1
32 changed files with 2102 additions and 7 deletions
|
@ -333,6 +333,56 @@ definitions:
|
|||
type: object
|
||||
x-go-name: Account
|
||||
x-go-package: github.com/superseriousbusiness/gotosocial/internal/api/model
|
||||
accountExportStats:
|
||||
description: |-
|
||||
AccountExportStats models an account's stats
|
||||
specifically for the purpose of informing about
|
||||
export sizes at the /api/v1/exports/stats endpoint.
|
||||
properties:
|
||||
blocks_count:
|
||||
description: Number of accounts blocked by this account.
|
||||
example: 15
|
||||
format: int64
|
||||
type: integer
|
||||
x-go-name: BlocksCount
|
||||
followers_count:
|
||||
description: Number of accounts following this account.
|
||||
example: 50
|
||||
format: int64
|
||||
type: integer
|
||||
x-go-name: FollowersCount
|
||||
following_count:
|
||||
description: Number of accounts followed by this account.
|
||||
example: 50
|
||||
format: int64
|
||||
type: integer
|
||||
x-go-name: FollowingCount
|
||||
lists_count:
|
||||
description: Number of lists created by this account.
|
||||
example: 10
|
||||
format: int64
|
||||
type: integer
|
||||
x-go-name: ListsCount
|
||||
media_storage:
|
||||
description: 'TODO: String representation of media storage size attributed to this account.'
|
||||
example: 500MB
|
||||
type: string
|
||||
x-go-name: MediaStorage
|
||||
mutes_count:
|
||||
description: Number of accounts muted by this account.
|
||||
example: 11
|
||||
format: int64
|
||||
type: integer
|
||||
x-go-name: MutesCount
|
||||
statuses_count:
|
||||
description: Number of statuses created by this account.
|
||||
example: 81986
|
||||
format: int64
|
||||
type: integer
|
||||
x-go-name: StatusesCount
|
||||
type: object
|
||||
x-go-name: AccountExportStats
|
||||
x-go-package: github.com/superseriousbusiness/gotosocial/internal/api/model
|
||||
accountRelationship:
|
||||
properties:
|
||||
blocked_by:
|
||||
|
@ -6364,6 +6414,128 @@ paths:
|
|||
summary: Get an array of custom emojis available on the instance.
|
||||
tags:
|
||||
- custom_emojis
|
||||
/api/v1/exports/blocks.csv:
|
||||
get:
|
||||
operationId: exportBlocks
|
||||
produces:
|
||||
- text/csv
|
||||
responses:
|
||||
"200":
|
||||
description: CSV file of accounts that you block.
|
||||
"401":
|
||||
description: unauthorized
|
||||
"406":
|
||||
description: not acceptable
|
||||
"500":
|
||||
description: internal server error
|
||||
security:
|
||||
- OAuth2 Bearer:
|
||||
- read:blocks
|
||||
summary: Export a CSV file of accounts that you block.
|
||||
tags:
|
||||
- import-export
|
||||
/api/v1/exports/followers.csv:
|
||||
get:
|
||||
operationId: exportFollowers
|
||||
produces:
|
||||
- text/csv
|
||||
responses:
|
||||
"200":
|
||||
description: CSV file of accounts that follow you.
|
||||
"401":
|
||||
description: unauthorized
|
||||
"406":
|
||||
description: not acceptable
|
||||
"500":
|
||||
description: internal server error
|
||||
security:
|
||||
- OAuth2 Bearer:
|
||||
- read:follows
|
||||
summary: Export a CSV file of accounts that follow you.
|
||||
tags:
|
||||
- import-export
|
||||
/api/v1/exports/following.csv:
|
||||
get:
|
||||
operationId: exportFollowing
|
||||
produces:
|
||||
- text/csv
|
||||
responses:
|
||||
"200":
|
||||
description: CSV file of accounts that you follow.
|
||||
"401":
|
||||
description: unauthorized
|
||||
"406":
|
||||
description: not acceptable
|
||||
"500":
|
||||
description: internal server error
|
||||
security:
|
||||
- OAuth2 Bearer:
|
||||
- read:follows
|
||||
summary: Export a CSV file of accounts that you follow.
|
||||
tags:
|
||||
- import-export
|
||||
/api/v1/exports/lists.csv:
|
||||
get:
|
||||
operationId: exportLists
|
||||
produces:
|
||||
- text/csv
|
||||
responses:
|
||||
"200":
|
||||
description: CSV file of lists.
|
||||
"401":
|
||||
description: unauthorized
|
||||
"406":
|
||||
description: not acceptable
|
||||
"500":
|
||||
description: internal server error
|
||||
security:
|
||||
- OAuth2 Bearer:
|
||||
- read:lists
|
||||
summary: Export a CSV file of lists created by you.
|
||||
tags:
|
||||
- import-export
|
||||
/api/v1/exports/mutes.csv:
|
||||
get:
|
||||
operationId: exportMutes
|
||||
produces:
|
||||
- text/csv
|
||||
responses:
|
||||
"200":
|
||||
description: CSV file of accounts that you mute.
|
||||
"401":
|
||||
description: unauthorized
|
||||
"406":
|
||||
description: not acceptable
|
||||
"500":
|
||||
description: internal server error
|
||||
security:
|
||||
- OAuth2 Bearer:
|
||||
- read:mutes
|
||||
summary: Export a CSV file of accounts that you mute.
|
||||
tags:
|
||||
- import-export
|
||||
/api/v1/exports/stats:
|
||||
get:
|
||||
operationId: exportStats
|
||||
produces:
|
||||
- application/json
|
||||
responses:
|
||||
"200":
|
||||
description: Export stats for the requesting account.
|
||||
schema:
|
||||
$ref: '#/definitions/accountExportStats'
|
||||
"401":
|
||||
description: unauthorized
|
||||
"406":
|
||||
description: not acceptable
|
||||
"500":
|
||||
description: internal server error
|
||||
security:
|
||||
- OAuth2 Bearer:
|
||||
- read:account
|
||||
summary: Returns informational stats on the number of items that can be exported for requesting account.
|
||||
tags:
|
||||
- import-export
|
||||
/api/v1/favourites:
|
||||
get:
|
||||
description: |-
|
||||
|
|
|
@ -204,3 +204,11 @@ For more information on the way GoToSocial manages passwords, please see the [Pa
|
|||
In the migration section you can manage settings related to aliasing and/or migrating your account to or from another account.
|
||||
|
||||
Please see the [migration document](./migration.md) for more information on moving your account.
|
||||
|
||||
## Export & Import
|
||||
|
||||
In the export & import section, you can export data from your GoToSocial account, or import data into it (TODO).
|
||||
|
||||
### Export
|
||||
|
||||
To export your following, followers, lists, account blocks, or account mutes, you can use the button on this page. All exports will be served in Mastodon-compatible CSV format, so you can import them later into Mastodon or another GoToSocial instance, if you like.
|
||||
|
|
|
@ -28,6 +28,7 @@
|
|||
"github.com/superseriousbusiness/gotosocial/internal/api/client/bookmarks"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/api/client/conversations"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/api/client/customemojis"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/api/client/exports"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/api/client/favourites"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/api/client/featuredtags"
|
||||
filtersV1 "github.com/superseriousbusiness/gotosocial/internal/api/client/filters/v1"
|
||||
|
@ -68,6 +69,7 @@ type Client struct {
|
|||
bookmarks *bookmarks.Module // api/v1/bookmarks
|
||||
conversations *conversations.Module // api/v1/conversations
|
||||
customEmojis *customemojis.Module // api/v1/custom_emojis
|
||||
exports *exports.Module // api/v1/exports
|
||||
favourites *favourites.Module // api/v1/favourites
|
||||
featuredTags *featuredtags.Module // api/v1/featured_tags
|
||||
filtersV1 *filtersV1.Module // api/v1/filters
|
||||
|
@ -116,6 +118,7 @@ func (c *Client) Route(r *router.Router, m ...gin.HandlerFunc) {
|
|||
c.bookmarks.Route(h)
|
||||
c.conversations.Route(h)
|
||||
c.customEmojis.Route(h)
|
||||
c.exports.Route(h)
|
||||
c.favourites.Route(h)
|
||||
c.featuredTags.Route(h)
|
||||
c.filtersV1.Route(h)
|
||||
|
@ -152,6 +155,7 @@ func NewClient(state *state.State, p *processing.Processor) *Client {
|
|||
bookmarks: bookmarks.New(p),
|
||||
conversations: conversations.New(p),
|
||||
customEmojis: customemojis.New(p),
|
||||
exports: exports.New(p),
|
||||
favourites: favourites.New(p),
|
||||
featuredTags: featuredtags.New(p),
|
||||
filtersV1: filtersV1.New(p),
|
||||
|
|
76
internal/api/client/exports/blocks.go
Normal file
76
internal/api/client/exports/blocks.go
Normal file
|
@ -0,0 +1,76 @@
|
|||
// 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 exports
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/oauth"
|
||||
)
|
||||
|
||||
// ExportBlocksGETHandler swagger:operation GET /api/v1/exports/blocks.csv exportBlocks
|
||||
//
|
||||
// Export a CSV file of accounts that you block.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - import-export
|
||||
//
|
||||
// produces:
|
||||
// - text/csv
|
||||
//
|
||||
// security:
|
||||
// - OAuth2 Bearer:
|
||||
// - read:blocks
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// name: accounts
|
||||
// description: CSV file of accounts that you block.
|
||||
// '401':
|
||||
// description: unauthorized
|
||||
// '406':
|
||||
// description: not acceptable
|
||||
// '500':
|
||||
// description: internal server error
|
||||
func (m *Module) ExportBlocksGETHandler(c *gin.Context) {
|
||||
authed, err := oauth.Authed(c, true, true, true, true)
|
||||
if err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorUnauthorized(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err := apiutil.NegotiateAccept(c, apiutil.CSVHeaders...); err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorNotAcceptable(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
records, errWithCode := m.processor.Account().ExportBlocks(
|
||||
c.Request.Context(),
|
||||
authed.Account,
|
||||
)
|
||||
if errWithCode != nil {
|
||||
apiutil.ErrorHandler(c, errWithCode, m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
apiutil.EncodeCSVResponse(c.Writer, c.Request, http.StatusOK, records)
|
||||
}
|
54
internal/api/client/exports/exports.go
Normal file
54
internal/api/client/exports/exports.go
Normal file
|
@ -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 exports
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/processing"
|
||||
)
|
||||
|
||||
const (
|
||||
BasePath = "/v1/exports"
|
||||
StatsPath = BasePath + "/stats"
|
||||
FollowingPath = BasePath + "/following.csv"
|
||||
FollowersPath = BasePath + "/followers.csv"
|
||||
ListsPath = BasePath + "/lists.csv"
|
||||
BlocksPath = BasePath + "/blocks.csv"
|
||||
MutesPath = BasePath + "/mutes.csv"
|
||||
)
|
||||
|
||||
type Module struct {
|
||||
processor *processing.Processor
|
||||
}
|
||||
|
||||
func New(processor *processing.Processor) *Module {
|
||||
return &Module{
|
||||
processor: processor,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Module) Route(attachHandler func(method string, path string, f ...gin.HandlerFunc) gin.IRoutes) {
|
||||
attachHandler(http.MethodGet, StatsPath, m.ExportStatsGETHandler)
|
||||
attachHandler(http.MethodGet, FollowingPath, m.ExportFollowingGETHandler)
|
||||
attachHandler(http.MethodGet, FollowersPath, m.ExportFollowersGETHandler)
|
||||
attachHandler(http.MethodGet, ListsPath, m.ExportListsGETHandler)
|
||||
attachHandler(http.MethodGet, BlocksPath, m.ExportBlocksGETHandler)
|
||||
attachHandler(http.MethodGet, MutesPath, m.ExportMutesGETHandler)
|
||||
}
|
275
internal/api/client/exports/exports_test.go
Normal file
275
internal/api/client/exports/exports_test.go
Normal file
|
@ -0,0 +1,275 @@
|
|||
// 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 exports_test
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/stretchr/testify/suite"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/api/client/exports"
|
||||
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/filter/visibility"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/oauth"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/state"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/typeutils"
|
||||
"github.com/superseriousbusiness/gotosocial/testrig"
|
||||
)
|
||||
|
||||
type ExportsTestSuite struct {
|
||||
// Suite interfaces
|
||||
suite.Suite
|
||||
state state.State
|
||||
|
||||
// standard suite models
|
||||
testTokens map[string]*gtsmodel.Token
|
||||
testClients map[string]*gtsmodel.Client
|
||||
testApplications map[string]*gtsmodel.Application
|
||||
testUsers map[string]*gtsmodel.User
|
||||
testAccounts map[string]*gtsmodel.Account
|
||||
|
||||
// module being tested
|
||||
exportsModule *exports.Module
|
||||
}
|
||||
|
||||
func (suite *ExportsTestSuite) SetupSuite() {
|
||||
suite.testTokens = testrig.NewTestTokens()
|
||||
suite.testClients = testrig.NewTestClients()
|
||||
suite.testApplications = testrig.NewTestApplications()
|
||||
suite.testUsers = testrig.NewTestUsers()
|
||||
suite.testAccounts = testrig.NewTestAccounts()
|
||||
}
|
||||
|
||||
func (suite *ExportsTestSuite) SetupTest() {
|
||||
suite.state.Caches.Init()
|
||||
testrig.StartNoopWorkers(&suite.state)
|
||||
|
||||
testrig.InitTestConfig()
|
||||
testrig.InitTestLog()
|
||||
|
||||
suite.state.DB = testrig.NewTestDB(&suite.state)
|
||||
suite.state.Storage = testrig.NewInMemoryStorage()
|
||||
|
||||
testrig.StartTimelines(
|
||||
&suite.state,
|
||||
visibility.NewFilter(&suite.state),
|
||||
typeutils.NewConverter(&suite.state),
|
||||
)
|
||||
|
||||
testrig.StandardDBSetup(suite.state.DB, nil)
|
||||
testrig.StandardStorageSetup(suite.state.Storage, "../../../../testrig/media")
|
||||
|
||||
mediaManager := testrig.NewTestMediaManager(&suite.state)
|
||||
|
||||
federator := testrig.NewTestFederator(
|
||||
&suite.state,
|
||||
testrig.NewTestTransportController(
|
||||
&suite.state,
|
||||
testrig.NewMockHTTPClient(nil, "../../../../testrig/media"),
|
||||
),
|
||||
mediaManager,
|
||||
)
|
||||
|
||||
processor := testrig.NewTestProcessor(
|
||||
&suite.state,
|
||||
federator,
|
||||
testrig.NewEmailSender("../../../../web/template/", nil),
|
||||
mediaManager,
|
||||
)
|
||||
|
||||
suite.exportsModule = exports.New(processor)
|
||||
}
|
||||
|
||||
func (suite *ExportsTestSuite) TriggerHandler(
|
||||
handler gin.HandlerFunc,
|
||||
path string,
|
||||
contentType string,
|
||||
application *gtsmodel.Application,
|
||||
token *gtsmodel.Token,
|
||||
user *gtsmodel.User,
|
||||
account *gtsmodel.Account,
|
||||
) *httptest.ResponseRecorder {
|
||||
// Set up request.
|
||||
recorder := httptest.NewRecorder()
|
||||
ctx, _ := testrig.CreateGinTestContext(recorder, nil)
|
||||
|
||||
// Authorize the request ctx as though it
|
||||
// had passed through API auth handlers.
|
||||
ctx.Set(oauth.SessionAuthorizedApplication, application)
|
||||
ctx.Set(oauth.SessionAuthorizedToken, oauth.DBTokenToToken(token))
|
||||
ctx.Set(oauth.SessionAuthorizedUser, user)
|
||||
ctx.Set(oauth.SessionAuthorizedAccount, account)
|
||||
|
||||
// Create test request.
|
||||
target := "http://localhost:8080/api" + path
|
||||
ctx.Request = httptest.NewRequest(http.MethodGet, target, nil)
|
||||
ctx.Request.Header.Set("Accept", contentType)
|
||||
|
||||
// Trigger handler.
|
||||
handler(ctx)
|
||||
|
||||
return recorder
|
||||
}
|
||||
|
||||
func (suite *ExportsTestSuite) TearDownTest() {
|
||||
testrig.StandardDBTeardown(suite.state.DB)
|
||||
testrig.StandardStorageTeardown(suite.state.Storage)
|
||||
testrig.StopWorkers(&suite.state)
|
||||
}
|
||||
|
||||
func (suite *ExportsTestSuite) TestExports() {
|
||||
type testCase struct {
|
||||
handler gin.HandlerFunc
|
||||
path string
|
||||
contentType string
|
||||
application *gtsmodel.Application
|
||||
token *gtsmodel.Token
|
||||
user *gtsmodel.User
|
||||
account *gtsmodel.Account
|
||||
expect string
|
||||
}
|
||||
|
||||
testCases := []testCase{
|
||||
// Export Following
|
||||
{
|
||||
handler: suite.exportsModule.ExportFollowingGETHandler,
|
||||
path: exports.FollowingPath,
|
||||
contentType: apiutil.TextCSV,
|
||||
application: suite.testApplications["application_1"],
|
||||
token: suite.testTokens["local_account_1"],
|
||||
user: suite.testUsers["local_account_1"],
|
||||
account: suite.testAccounts["local_account_1"],
|
||||
expect: `Account address,Show boosts
|
||||
admin@localhost:8080,true
|
||||
1happyturtle@localhost:8080,true
|
||||
`,
|
||||
},
|
||||
// Export Followers.
|
||||
{
|
||||
handler: suite.exportsModule.ExportFollowersGETHandler,
|
||||
path: exports.FollowingPath,
|
||||
contentType: apiutil.TextCSV,
|
||||
application: suite.testApplications["application_1"],
|
||||
token: suite.testTokens["local_account_1"],
|
||||
user: suite.testUsers["local_account_1"],
|
||||
account: suite.testAccounts["local_account_1"],
|
||||
expect: `Account address
|
||||
1happyturtle@localhost:8080
|
||||
admin@localhost:8080
|
||||
`,
|
||||
},
|
||||
// Export Lists.
|
||||
{
|
||||
handler: suite.exportsModule.ExportListsGETHandler,
|
||||
path: exports.ListsPath,
|
||||
contentType: apiutil.TextCSV,
|
||||
application: suite.testApplications["application_1"],
|
||||
token: suite.testTokens["local_account_1"],
|
||||
user: suite.testUsers["local_account_1"],
|
||||
account: suite.testAccounts["local_account_1"],
|
||||
expect: `Cool Ass Posters From This Instance,admin@localhost:8080
|
||||
Cool Ass Posters From This Instance,1happyturtle@localhost:8080
|
||||
`,
|
||||
},
|
||||
// Export Mutes.
|
||||
{
|
||||
handler: suite.exportsModule.ExportMutesGETHandler,
|
||||
path: exports.MutesPath,
|
||||
contentType: apiutil.TextCSV,
|
||||
application: suite.testApplications["application_1"],
|
||||
token: suite.testTokens["local_account_1"],
|
||||
user: suite.testUsers["local_account_1"],
|
||||
account: suite.testAccounts["local_account_1"],
|
||||
expect: `Account address,Hide notifications
|
||||
`,
|
||||
},
|
||||
// Export Blocks.
|
||||
{
|
||||
handler: suite.exportsModule.ExportBlocksGETHandler,
|
||||
path: exports.BlocksPath,
|
||||
contentType: apiutil.TextCSV,
|
||||
application: suite.testApplications["application_1"],
|
||||
token: suite.testTokens["local_account_2"],
|
||||
user: suite.testUsers["local_account_2"],
|
||||
account: suite.testAccounts["local_account_2"],
|
||||
expect: `foss_satan@fossbros-anonymous.io
|
||||
`,
|
||||
},
|
||||
// Export Stats.
|
||||
{
|
||||
handler: suite.exportsModule.ExportStatsGETHandler,
|
||||
path: exports.StatsPath,
|
||||
contentType: apiutil.AppJSON,
|
||||
application: suite.testApplications["application_1"],
|
||||
token: suite.testTokens["local_account_1"],
|
||||
user: suite.testUsers["local_account_1"],
|
||||
account: suite.testAccounts["local_account_1"],
|
||||
expect: `{
|
||||
"media_storage": "",
|
||||
"followers_count": 2,
|
||||
"following_count": 2,
|
||||
"statuses_count": 8,
|
||||
"lists_count": 1,
|
||||
"blocks_count": 0,
|
||||
"mutes_count": 0
|
||||
}`,
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testCases {
|
||||
recorder := suite.TriggerHandler(
|
||||
test.handler,
|
||||
test.path,
|
||||
test.contentType,
|
||||
test.application,
|
||||
test.token,
|
||||
test.user,
|
||||
test.account,
|
||||
)
|
||||
|
||||
// Check response code.
|
||||
suite.EqualValues(http.StatusOK, recorder.Code)
|
||||
|
||||
// Check response body.
|
||||
b, err := io.ReadAll(recorder.Body)
|
||||
if err != nil {
|
||||
suite.FailNow(err.Error())
|
||||
}
|
||||
|
||||
// If json response, indent it nicely.
|
||||
if recorder.Result().Header.Get("Content-Type") == "application/json" {
|
||||
dst := &bytes.Buffer{}
|
||||
if err := json.Indent(dst, b, "", " "); err != nil {
|
||||
suite.FailNow(err.Error())
|
||||
}
|
||||
b = dst.Bytes()
|
||||
}
|
||||
|
||||
suite.Equal(test.expect, string(b))
|
||||
}
|
||||
}
|
||||
|
||||
func TestExportsTestSuite(t *testing.T) {
|
||||
suite.Run(t, new(ExportsTestSuite))
|
||||
}
|
76
internal/api/client/exports/followers.go
Normal file
76
internal/api/client/exports/followers.go
Normal file
|
@ -0,0 +1,76 @@
|
|||
// 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 exports
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/oauth"
|
||||
)
|
||||
|
||||
// ExportFollowersGETHandler swagger:operation GET /api/v1/exports/followers.csv exportFollowers
|
||||
//
|
||||
// Export a CSV file of accounts that follow you.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - import-export
|
||||
//
|
||||
// produces:
|
||||
// - text/csv
|
||||
//
|
||||
// security:
|
||||
// - OAuth2 Bearer:
|
||||
// - read:follows
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// name: accounts
|
||||
// description: CSV file of accounts that follow you.
|
||||
// '401':
|
||||
// description: unauthorized
|
||||
// '406':
|
||||
// description: not acceptable
|
||||
// '500':
|
||||
// description: internal server error
|
||||
func (m *Module) ExportFollowersGETHandler(c *gin.Context) {
|
||||
authed, err := oauth.Authed(c, true, true, true, true)
|
||||
if err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorUnauthorized(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err := apiutil.NegotiateAccept(c, apiutil.CSVHeaders...); err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorNotAcceptable(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
records, errWithCode := m.processor.Account().ExportFollowers(
|
||||
c.Request.Context(),
|
||||
authed.Account,
|
||||
)
|
||||
if errWithCode != nil {
|
||||
apiutil.ErrorHandler(c, errWithCode, m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
apiutil.EncodeCSVResponse(c.Writer, c.Request, http.StatusOK, records)
|
||||
}
|
76
internal/api/client/exports/following.go
Normal file
76
internal/api/client/exports/following.go
Normal file
|
@ -0,0 +1,76 @@
|
|||
// 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 exports
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/oauth"
|
||||
)
|
||||
|
||||
// ExportFollowingGETHandler swagger:operation GET /api/v1/exports/following.csv exportFollowing
|
||||
//
|
||||
// Export a CSV file of accounts that you follow.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - import-export
|
||||
//
|
||||
// produces:
|
||||
// - text/csv
|
||||
//
|
||||
// security:
|
||||
// - OAuth2 Bearer:
|
||||
// - read:follows
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// name: accounts
|
||||
// description: CSV file of accounts that you follow.
|
||||
// '401':
|
||||
// description: unauthorized
|
||||
// '406':
|
||||
// description: not acceptable
|
||||
// '500':
|
||||
// description: internal server error
|
||||
func (m *Module) ExportFollowingGETHandler(c *gin.Context) {
|
||||
authed, err := oauth.Authed(c, true, true, true, true)
|
||||
if err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorUnauthorized(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err := apiutil.NegotiateAccept(c, apiutil.CSVHeaders...); err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorNotAcceptable(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
records, errWithCode := m.processor.Account().ExportFollowing(
|
||||
c.Request.Context(),
|
||||
authed.Account,
|
||||
)
|
||||
if errWithCode != nil {
|
||||
apiutil.ErrorHandler(c, errWithCode, m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
apiutil.EncodeCSVResponse(c.Writer, c.Request, http.StatusOK, records)
|
||||
}
|
76
internal/api/client/exports/lists.go
Normal file
76
internal/api/client/exports/lists.go
Normal file
|
@ -0,0 +1,76 @@
|
|||
// 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 exports
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/oauth"
|
||||
)
|
||||
|
||||
// ExportListsGETHandler swagger:operation GET /api/v1/exports/lists.csv exportLists
|
||||
//
|
||||
// Export a CSV file of lists created by you.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - import-export
|
||||
//
|
||||
// produces:
|
||||
// - text/csv
|
||||
//
|
||||
// security:
|
||||
// - OAuth2 Bearer:
|
||||
// - read:lists
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// name: accounts
|
||||
// description: CSV file of lists.
|
||||
// '401':
|
||||
// description: unauthorized
|
||||
// '406':
|
||||
// description: not acceptable
|
||||
// '500':
|
||||
// description: internal server error
|
||||
func (m *Module) ExportListsGETHandler(c *gin.Context) {
|
||||
authed, err := oauth.Authed(c, true, true, true, true)
|
||||
if err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorUnauthorized(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err := apiutil.NegotiateAccept(c, apiutil.CSVHeaders...); err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorNotAcceptable(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
records, errWithCode := m.processor.Account().ExportLists(
|
||||
c.Request.Context(),
|
||||
authed.Account,
|
||||
)
|
||||
if errWithCode != nil {
|
||||
apiutil.ErrorHandler(c, errWithCode, m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
apiutil.EncodeCSVResponse(c.Writer, c.Request, http.StatusOK, records)
|
||||
}
|
76
internal/api/client/exports/mutes.go
Normal file
76
internal/api/client/exports/mutes.go
Normal file
|
@ -0,0 +1,76 @@
|
|||
// 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 exports
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/oauth"
|
||||
)
|
||||
|
||||
// ExportMutesGETHandler swagger:operation GET /api/v1/exports/mutes.csv exportMutes
|
||||
//
|
||||
// Export a CSV file of accounts that you mute.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - import-export
|
||||
//
|
||||
// produces:
|
||||
// - text/csv
|
||||
//
|
||||
// security:
|
||||
// - OAuth2 Bearer:
|
||||
// - read:mutes
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// name: accounts
|
||||
// description: CSV file of accounts that you mute.
|
||||
// '401':
|
||||
// description: unauthorized
|
||||
// '406':
|
||||
// description: not acceptable
|
||||
// '500':
|
||||
// description: internal server error
|
||||
func (m *Module) ExportMutesGETHandler(c *gin.Context) {
|
||||
authed, err := oauth.Authed(c, true, true, true, true)
|
||||
if err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorUnauthorized(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err := apiutil.NegotiateAccept(c, apiutil.CSVHeaders...); err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorNotAcceptable(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
records, errWithCode := m.processor.Account().ExportMutes(
|
||||
c.Request.Context(),
|
||||
authed.Account,
|
||||
)
|
||||
if errWithCode != nil {
|
||||
apiutil.ErrorHandler(c, errWithCode, m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
apiutil.EncodeCSVResponse(c.Writer, c.Request, http.StatusOK, records)
|
||||
}
|
77
internal/api/client/exports/stats.go
Normal file
77
internal/api/client/exports/stats.go
Normal file
|
@ -0,0 +1,77 @@
|
|||
// 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 exports
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/oauth"
|
||||
)
|
||||
|
||||
// ExportStatsGETHandler swagger:operation GET /api/v1/exports/stats exportStats
|
||||
//
|
||||
// Returns informational stats on the number of items that can be exported for requesting account.
|
||||
//
|
||||
// ---
|
||||
// tags:
|
||||
// - import-export
|
||||
//
|
||||
// produces:
|
||||
// - application/json
|
||||
//
|
||||
// security:
|
||||
// - OAuth2 Bearer:
|
||||
// - read:account
|
||||
//
|
||||
// responses:
|
||||
// '200':
|
||||
// description: Export stats for the requesting account.
|
||||
// schema:
|
||||
// "$ref": "#/definitions/accountExportStats"
|
||||
// '401':
|
||||
// description: unauthorized
|
||||
// '406':
|
||||
// description: not acceptable
|
||||
// '500':
|
||||
// description: internal server error
|
||||
func (m *Module) ExportStatsGETHandler(c *gin.Context) {
|
||||
authed, err := oauth.Authed(c, true, true, true, true)
|
||||
if err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorUnauthorized(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err := apiutil.NegotiateAccept(c, apiutil.JSONAcceptHeaders...); err != nil {
|
||||
apiutil.ErrorHandler(c, gtserror.NewErrorNotAcceptable(err, err.Error()), m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
exportStats, errWithCode := m.processor.Account().ExportStats(
|
||||
c.Request.Context(),
|
||||
authed.Account,
|
||||
)
|
||||
if errWithCode != nil {
|
||||
apiutil.ErrorHandler(c, errWithCode, m.processor.InstanceGetV1)
|
||||
return
|
||||
}
|
||||
|
||||
apiutil.JSON(c, http.StatusOK, exportStats)
|
||||
}
|
60
internal/api/model/exportimport.go
Normal file
60
internal/api/model/exportimport.go
Normal file
|
@ -0,0 +1,60 @@
|
|||
// 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 model
|
||||
|
||||
// AccountExportStats models an account's stats
|
||||
// specifically for the purpose of informing about
|
||||
// export sizes at the /api/v1/exports/stats endpoint.
|
||||
//
|
||||
// swagger:model accountExportStats
|
||||
type AccountExportStats struct {
|
||||
// TODO: String representation of media storage size attributed to this account.
|
||||
//
|
||||
// example: 500MB
|
||||
MediaStorage string `json:"media_storage"`
|
||||
|
||||
// Number of accounts following this account.
|
||||
//
|
||||
// example: 50
|
||||
FollowersCount int `json:"followers_count"`
|
||||
|
||||
// Number of accounts followed by this account.
|
||||
//
|
||||
// example: 50
|
||||
FollowingCount int `json:"following_count"`
|
||||
|
||||
// Number of statuses created by this account.
|
||||
//
|
||||
// example: 81986
|
||||
StatusesCount int `json:"statuses_count"`
|
||||
|
||||
// Number of lists created by this account.
|
||||
//
|
||||
// example: 10
|
||||
ListsCount int `json:"lists_count"`
|
||||
|
||||
// Number of accounts blocked by this account.
|
||||
//
|
||||
// example: 15
|
||||
BlocksCount int `json:"blocks_count"`
|
||||
|
||||
// Number of accounts muted by this account.
|
||||
//
|
||||
// example: 11
|
||||
MutesCount int `json:"mutes_count"`
|
||||
}
|
|
@ -35,6 +35,7 @@
|
|||
TextXML = `text/xml`
|
||||
TextHTML = `text/html`
|
||||
TextCSS = `text/css`
|
||||
TextCSV = `text/csv`
|
||||
)
|
||||
|
||||
// JSONContentType returns whether is application/json(;charset=utf-8)? content-type.
|
||||
|
|
|
@ -88,6 +88,12 @@
|
|||
AppXML,
|
||||
}
|
||||
|
||||
// CSVHeaders just contains the text/csv
|
||||
// MIME type, used for import/export.
|
||||
var CSVHeaders = []string{
|
||||
TextCSV,
|
||||
}
|
||||
|
||||
// NegotiateAccept takes the *gin.Context from an incoming request, and a
|
||||
// slice of Offers, and performs content negotiation for the given request
|
||||
// with the given content-type offers. It will return a string representation
|
||||
|
|
|
@ -18,6 +18,7 @@
|
|||
package util
|
||||
|
||||
import (
|
||||
"encoding/csv"
|
||||
"encoding/json"
|
||||
"encoding/xml"
|
||||
"io"
|
||||
|
@ -213,6 +214,47 @@ func EncodeXMLResponse(
|
|||
putBuf(buf)
|
||||
}
|
||||
|
||||
// EncodeCSVResponse encodes 'records' as CSV HTTP response
|
||||
// to ResponseWriter with given status code, using CSV content-type.
|
||||
func EncodeCSVResponse(
|
||||
rw http.ResponseWriter,
|
||||
r *http.Request,
|
||||
statusCode int,
|
||||
records [][]string,
|
||||
) {
|
||||
// Acquire buffer.
|
||||
buf := getBuf()
|
||||
|
||||
// Wrap buffer in CSV writer.
|
||||
csvWriter := csv.NewWriter(buf)
|
||||
|
||||
// Write all the records to the buffer.
|
||||
if err := csvWriter.WriteAll(records); err == nil {
|
||||
// Respond with the now-known
|
||||
// size byte slice within buf.
|
||||
WriteResponseBytes(rw, r,
|
||||
statusCode,
|
||||
TextCSV,
|
||||
buf.B,
|
||||
)
|
||||
} else {
|
||||
// This will always be an csv error, we
|
||||
// can't really add any more useful context.
|
||||
log.Error(r.Context(), err)
|
||||
|
||||
// Any error returned here is unrecoverable,
|
||||
// set Internal Server Error JSON response.
|
||||
WriteResponseBytes(rw, r,
|
||||
http.StatusInternalServerError,
|
||||
AppJSON,
|
||||
StatusInternalServerErrorJSON,
|
||||
)
|
||||
}
|
||||
|
||||
// Release.
|
||||
putBuf(buf)
|
||||
}
|
||||
|
||||
// writeResponseUnknownLength handles reading data of unknown legnth
|
||||
// efficiently into memory, and passing on to WriteResponseBytes().
|
||||
func writeResponseUnknownLength(
|
||||
|
|
|
@ -106,6 +106,14 @@ func (l *listDB) GetListsForAccountID(ctx context.Context, accountID string) ([]
|
|||
return l.GetListsByIDs(ctx, listIDs)
|
||||
}
|
||||
|
||||
func (l *listDB) CountListsForAccountID(ctx context.Context, accountID string) (int, error) {
|
||||
return l.db.
|
||||
NewSelect().
|
||||
Table("lists").
|
||||
Where("? = ?", bun.Ident("account_id"), accountID).
|
||||
Count(ctx)
|
||||
}
|
||||
|
||||
func (l *listDB) PopulateList(ctx context.Context, list *gtsmodel.List) error {
|
||||
var (
|
||||
err error
|
||||
|
|
|
@ -178,6 +178,11 @@ func (r *relationshipDB) GetAccountBlocks(ctx context.Context, accountID string,
|
|||
return r.GetBlocksByIDs(ctx, blockIDs)
|
||||
}
|
||||
|
||||
func (r *relationshipDB) CountAccountBlocks(ctx context.Context, accountID string) (int, error) {
|
||||
blockIDs, err := r.GetAccountBlockIDs(ctx, accountID, nil)
|
||||
return len(blockIDs), err
|
||||
}
|
||||
|
||||
func (r *relationshipDB) GetAccountFollowIDs(ctx context.Context, accountID string, page *paging.Page) ([]string, error) {
|
||||
return loadPagedIDs(&r.state.Caches.DB.FollowIDs, ">"+accountID, page, func() ([]string, error) {
|
||||
var followIDs []string
|
||||
|
|
|
@ -77,6 +77,11 @@ func(mute *gtsmodel.UserMute) error {
|
|||
)
|
||||
}
|
||||
|
||||
func (r *relationshipDB) CountAccountMutes(ctx context.Context, accountID string) (int, error) {
|
||||
muteIDs, err := r.getAccountMuteIDs(ctx, accountID, nil)
|
||||
return len(muteIDs), err
|
||||
}
|
||||
|
||||
func (r *relationshipDB) getMutesByIDs(ctx context.Context, ids []string) ([]*gtsmodel.UserMute, error) {
|
||||
// Load all mutes IDs via cache loader callbacks.
|
||||
mutes, err := r.state.Caches.DB.UserMute.LoadIDs("ID",
|
||||
|
|
|
@ -33,6 +33,9 @@ type List interface {
|
|||
// GetListsForAccountID gets all lists owned by the given accountID.
|
||||
GetListsForAccountID(ctx context.Context, accountID string) ([]*gtsmodel.List, error)
|
||||
|
||||
// CountListsForAccountID counts the number of lists owned by the given accountID.
|
||||
CountListsForAccountID(ctx context.Context, accountID string) (int, error)
|
||||
|
||||
// PopulateList ensures that the list's struct fields are populated.
|
||||
PopulateList(ctx context.Context, list *gtsmodel.List) error
|
||||
|
||||
|
|
|
@ -179,6 +179,9 @@ type Relationship interface {
|
|||
// GetAccountBlockIDs is like GetAccountBlocks, but returns just IDs.
|
||||
GetAccountBlockIDs(ctx context.Context, accountID string, page *paging.Page) ([]string, error)
|
||||
|
||||
// CountAccountBlocks counts the number of blocks owned by the given account.
|
||||
CountAccountBlocks(ctx context.Context, accountID string) (int, error)
|
||||
|
||||
// GetNote gets a private note from a source account on a target account, if it exists.
|
||||
GetNote(ctx context.Context, sourceAccountID string, targetAccountID string) (*gtsmodel.AccountNote, error)
|
||||
|
||||
|
@ -197,6 +200,9 @@ type Relationship interface {
|
|||
// GetMute returns the mute from account1 targeting account2, if it exists, or an error if it doesn't.
|
||||
GetMute(ctx context.Context, account1 string, account2 string) (*gtsmodel.UserMute, error)
|
||||
|
||||
// CountAccountMutes counts the number of mutes owned by the given account.
|
||||
CountAccountMutes(ctx context.Context, accountID string) (int, error)
|
||||
|
||||
// PutMute attempts to insert or update the given account mute in the database.
|
||||
PutMute(ctx context.Context, mute *gtsmodel.UserMute) error
|
||||
|
||||
|
|
159
internal/processing/account/export.go
Normal file
159
internal/processing/account/export.go
Normal file
|
@ -0,0 +1,159 @@
|
|||
// 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 account
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
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"
|
||||
)
|
||||
|
||||
// ExportStats returns the requester's export stats,
|
||||
// ie., the counts of items that can be exported.
|
||||
func (p *Processor) ExportStats(
|
||||
ctx context.Context,
|
||||
requester *gtsmodel.Account,
|
||||
) (*apimodel.AccountExportStats, gtserror.WithCode) {
|
||||
exportStats, err := p.converter.AccountToExportStats(ctx, requester)
|
||||
if err != nil {
|
||||
err = gtserror.Newf("db error getting export stats: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
return exportStats, nil
|
||||
}
|
||||
|
||||
// ExportFollowing returns a CSV file of
|
||||
// accounts that the requester follows.
|
||||
func (p *Processor) ExportFollowing(
|
||||
ctx context.Context,
|
||||
requester *gtsmodel.Account,
|
||||
) ([][]string, gtserror.WithCode) {
|
||||
// Fetch accounts followed by requester,
|
||||
// using a nil page to get everything.
|
||||
following, err := p.state.DB.GetAccountFollows(ctx, requester.ID, nil)
|
||||
if err != nil && !errors.Is(err, db.ErrNoEntries) {
|
||||
err = gtserror.Newf("db error getting follows: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
// Convert accounts to CSV-compatible
|
||||
// records, with appropriate column headers.
|
||||
records, err := p.converter.FollowingToCSV(ctx, following)
|
||||
if err != nil {
|
||||
err = gtserror.Newf("error converting follows to records: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// ExportFollowers returns a CSV file of
|
||||
// accounts that follow the requester.
|
||||
func (p *Processor) ExportFollowers(
|
||||
ctx context.Context,
|
||||
requester *gtsmodel.Account,
|
||||
) ([][]string, gtserror.WithCode) {
|
||||
// Fetch accounts following requester,
|
||||
// using a nil page to get everything.
|
||||
followers, err := p.state.DB.GetAccountFollowers(ctx, requester.ID, nil)
|
||||
if err != nil && !errors.Is(err, db.ErrNoEntries) {
|
||||
err = gtserror.Newf("db error getting followers: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
// Convert accounts to CSV-compatible
|
||||
// records, with appropriate column headers.
|
||||
records, err := p.converter.FollowersToCSV(ctx, followers)
|
||||
if err != nil {
|
||||
err = gtserror.Newf("error converting followers to records: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// ExportLists returns a CSV file of
|
||||
// lists created by the requester.
|
||||
func (p *Processor) ExportLists(
|
||||
ctx context.Context,
|
||||
requester *gtsmodel.Account,
|
||||
) ([][]string, gtserror.WithCode) {
|
||||
lists, err := p.state.DB.GetListsForAccountID(ctx, requester.ID)
|
||||
if err != nil && !errors.Is(err, db.ErrNoEntries) {
|
||||
err = gtserror.Newf("db error getting lists: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
// Convert lists to CSV-compatible records.
|
||||
records, err := p.converter.ListsToCSV(ctx, lists)
|
||||
if err != nil {
|
||||
err = gtserror.Newf("error converting lists to records: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// ExportBlocks returns a CSV file of
|
||||
// account blocks created by the requester.
|
||||
func (p *Processor) ExportBlocks(
|
||||
ctx context.Context,
|
||||
requester *gtsmodel.Account,
|
||||
) ([][]string, gtserror.WithCode) {
|
||||
blocks, err := p.state.DB.GetAccountBlocks(ctx, requester.ID, nil)
|
||||
if err != nil && !errors.Is(err, db.ErrNoEntries) {
|
||||
err = gtserror.Newf("db error getting blocks: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
// Convert blocks to CSV-compatible records.
|
||||
records, err := p.converter.BlocksToCSV(ctx, blocks)
|
||||
if err != nil {
|
||||
err = gtserror.Newf("error converting blocks to records: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// ExportMutes returns a CSV file of
|
||||
// account mutes created by the requester.
|
||||
func (p *Processor) ExportMutes(
|
||||
ctx context.Context,
|
||||
requester *gtsmodel.Account,
|
||||
) ([][]string, gtserror.WithCode) {
|
||||
mutes, err := p.state.DB.GetAccountMutes(ctx, requester.ID, nil)
|
||||
if err != nil && !errors.Is(err, db.ErrNoEntries) {
|
||||
err = gtserror.Newf("db error getting mutes: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
// Convert mutes to CSV-compatible records.
|
||||
records, err := p.converter.MutesToCSV(ctx, mutes)
|
||||
if err != nil {
|
||||
err = gtserror.Newf("error converting mutes to records: %w", err)
|
||||
return nil, gtserror.NewErrorInternalError(err)
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
|
@ -102,7 +102,7 @@ func (e *exporter) exportDomainBlocks(ctx context.Context, file *os.File) ([]*tr
|
|||
return domainBlocks, nil
|
||||
}
|
||||
|
||||
func (e *exporter) exportFollows(ctx context.Context, accounts []*transmodel.Account, file *os.File) ([]*transmodel.Follow, error) {
|
||||
func (e *exporter) exportFollowing(ctx context.Context, accounts []*transmodel.Account, file *os.File) ([]*transmodel.Follow, error) {
|
||||
followsUnique := make(map[string]*transmodel.Follow)
|
||||
|
||||
// for each account we want to export both where it's following and where it's followed
|
||||
|
@ -111,12 +111,12 @@ func (e *exporter) exportFollows(ctx context.Context, accounts []*transmodel.Acc
|
|||
whereFollowing := []db.Where{{Key: "account_id", Value: a.ID}}
|
||||
following := []*transmodel.Follow{}
|
||||
if err := e.db.GetWhere(ctx, whereFollowing, &following); err != nil {
|
||||
return nil, fmt.Errorf("exportFollows: error selecting follows owned by account %s: %s", a.ID, err)
|
||||
return nil, fmt.Errorf("exportFollowing: error selecting follows owned by account %s: %s", a.ID, err)
|
||||
}
|
||||
for _, follow := range following {
|
||||
follow.Type = transmodel.TransFollow
|
||||
if err := e.simpleEncode(ctx, file, follow, follow.ID); err != nil {
|
||||
return nil, fmt.Errorf("exportFollows: error encoding follow owned by account %s: %s", a.ID, err)
|
||||
return nil, fmt.Errorf("exportFollowing: error encoding follow owned by account %s: %s", a.ID, err)
|
||||
}
|
||||
followsUnique[follow.ID] = follow
|
||||
}
|
||||
|
@ -125,12 +125,12 @@ func (e *exporter) exportFollows(ctx context.Context, accounts []*transmodel.Acc
|
|||
whereFollowed := []db.Where{{Key: "target_account_id", Value: a.ID}}
|
||||
followed := []*transmodel.Follow{}
|
||||
if err := e.db.GetWhere(ctx, whereFollowed, &followed); err != nil {
|
||||
return nil, fmt.Errorf("exportFollows: error selecting follows targeting account %s: %s", a.ID, err)
|
||||
return nil, fmt.Errorf("exportFollowing: error selecting follows targeting account %s: %s", a.ID, err)
|
||||
}
|
||||
for _, follow := range followed {
|
||||
follow.Type = transmodel.TransFollow
|
||||
if err := e.simpleEncode(ctx, file, follow, follow.ID); err != nil {
|
||||
return nil, fmt.Errorf("exportFollows: error encoding follow targeting account %s: %s", a.ID, err)
|
||||
return nil, fmt.Errorf("exportFollowing: error encoding follow targeting account %s: %s", a.ID, err)
|
||||
}
|
||||
followsUnique[follow.ID] = follow
|
||||
}
|
||||
|
|
|
@ -69,7 +69,7 @@ func (e *exporter) ExportMinimal(ctx context.Context, path string) error {
|
|||
}
|
||||
|
||||
// export all follows that relate to local accounts
|
||||
follows, err := e.exportFollows(ctx, localAccounts, file)
|
||||
follows, err := e.exportFollowing(ctx, localAccounts, file)
|
||||
if err != nil {
|
||||
return fmt.Errorf("ExportMinimal: error exporting follows: %s", err)
|
||||
}
|
||||
|
|
385
internal/typeutils/csv.go
Normal file
385
internal/typeutils/csv.go
Normal file
|
@ -0,0 +1,385 @@
|
|||
// 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 typeutils
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strconv"
|
||||
|
||||
apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/config"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtscontext"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
||||
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
||||
)
|
||||
|
||||
func (c *Converter) AccountToExportStats(
|
||||
ctx context.Context,
|
||||
a *gtsmodel.Account,
|
||||
) (*apimodel.AccountExportStats, error) {
|
||||
// Ensure account stats populated.
|
||||
if a.Stats == nil {
|
||||
if err := c.state.DB.PopulateAccountStats(ctx, a); err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"error getting stats for account %s: %w",
|
||||
a.ID, err,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
listsCount, err := c.state.DB.CountListsForAccountID(ctx, a.ID)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"error counting lists for account %s: %w",
|
||||
a.ID, err,
|
||||
)
|
||||
}
|
||||
|
||||
blockingCount, err := c.state.DB.CountAccountBlocks(ctx, a.ID)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"error counting lists for account %s: %w",
|
||||
a.ID, err,
|
||||
)
|
||||
}
|
||||
|
||||
mutingCount, err := c.state.DB.CountAccountMutes(ctx, a.ID)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"error counting lists for account %s: %w",
|
||||
a.ID, err,
|
||||
)
|
||||
}
|
||||
|
||||
return &apimodel.AccountExportStats{
|
||||
FollowersCount: *a.Stats.FollowersCount,
|
||||
FollowingCount: *a.Stats.FollowingCount,
|
||||
StatusesCount: *a.Stats.StatusesCount,
|
||||
ListsCount: listsCount,
|
||||
BlocksCount: blockingCount,
|
||||
MutesCount: mutingCount,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// FollowingToCSV converts a slice of follows into
|
||||
// a slice of CSV-compatible Following records.
|
||||
func (c *Converter) FollowingToCSV(
|
||||
ctx context.Context,
|
||||
following []*gtsmodel.Follow,
|
||||
) ([][]string, error) {
|
||||
// Records should be length of
|
||||
// input + 1 so we can add headers.
|
||||
records := make([][]string, 1, len(following)+1)
|
||||
|
||||
// Add headers at the
|
||||
// top of records.
|
||||
records[0] = []string{
|
||||
"Account address",
|
||||
"Show boosts",
|
||||
}
|
||||
|
||||
// We need to know our own domain for this.
|
||||
// Try account domain, fall back to host.
|
||||
thisDomain := config.GetAccountDomain()
|
||||
if thisDomain == "" {
|
||||
thisDomain = config.GetHost()
|
||||
}
|
||||
|
||||
// For each item, add a record.
|
||||
for _, follow := range following {
|
||||
if follow.TargetAccount == nil {
|
||||
// Retrieve target account.
|
||||
var err error
|
||||
follow.TargetAccount, err = c.state.DB.GetAccountByID(
|
||||
// Barebones is fine here.
|
||||
gtscontext.SetBarebones(ctx),
|
||||
follow.TargetAccountID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"db error getting target account for follow %s: %w",
|
||||
follow.ID, err,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
domain := follow.TargetAccount.Domain
|
||||
if domain == "" {
|
||||
// Local account,
|
||||
// use our domain.
|
||||
domain = thisDomain
|
||||
}
|
||||
|
||||
records = append(records, []string{
|
||||
// Account address: eg., someone@example.org
|
||||
// -- NOTE: without the leading '@'!
|
||||
follow.TargetAccount.Username + "@" + domain,
|
||||
// Show boosts: eg., true
|
||||
strconv.FormatBool(*follow.ShowReblogs),
|
||||
})
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// FollowersToCSV converts a slice of follows into
|
||||
// a slice of CSV-compatible Followers records.
|
||||
func (c *Converter) FollowersToCSV(
|
||||
ctx context.Context,
|
||||
followers []*gtsmodel.Follow,
|
||||
) ([][]string, error) {
|
||||
// Records should be length of
|
||||
// input + 1 so we can add headers.
|
||||
records := make([][]string, 1, len(followers)+1)
|
||||
|
||||
// Add header at the
|
||||
// top of records.
|
||||
records[0] = []string{
|
||||
"Account address",
|
||||
}
|
||||
|
||||
// We need to know our own domain for this.
|
||||
// Try account domain, fall back to host.
|
||||
thisDomain := config.GetAccountDomain()
|
||||
if thisDomain == "" {
|
||||
thisDomain = config.GetHost()
|
||||
}
|
||||
|
||||
// For each item, add a record.
|
||||
for _, follow := range followers {
|
||||
if follow.Account == nil {
|
||||
// Retrieve account.
|
||||
var err error
|
||||
follow.Account, err = c.state.DB.GetAccountByID(
|
||||
// Barebones is fine here.
|
||||
gtscontext.SetBarebones(ctx),
|
||||
follow.AccountID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"db error getting account for follow %s: %w",
|
||||
follow.ID, err,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
domain := follow.Account.Domain
|
||||
if domain == "" {
|
||||
// Local account,
|
||||
// use our domain.
|
||||
domain = thisDomain
|
||||
}
|
||||
|
||||
records = append(records, []string{
|
||||
// Account address: eg., someone@example.org
|
||||
// -- NOTE: without the leading '@'!
|
||||
follow.Account.Username + "@" + domain,
|
||||
})
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// FollowersToCSV converts a slice of follows into
|
||||
// a slice of CSV-compatible Followers records.
|
||||
func (c *Converter) ListsToCSV(
|
||||
ctx context.Context,
|
||||
lists []*gtsmodel.List,
|
||||
) ([][]string, error) {
|
||||
// We need to know our own domain for this.
|
||||
// Try account domain, fall back to host.
|
||||
thisDomain := config.GetAccountDomain()
|
||||
if thisDomain == "" {
|
||||
thisDomain = config.GetHost()
|
||||
}
|
||||
|
||||
// NOTE: Mastodon-compatible lists
|
||||
// CSV doesn't use column headers.
|
||||
records := make([][]string, 0)
|
||||
|
||||
// For each item, add a record.
|
||||
for _, list := range lists {
|
||||
for _, entry := range list.ListEntries {
|
||||
if entry.Follow == nil {
|
||||
// Retrieve follow.
|
||||
var err error
|
||||
entry.Follow, err = c.state.DB.GetFollowByID(
|
||||
ctx,
|
||||
entry.FollowID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"db error getting follow for list entry %s: %w",
|
||||
entry.ID, err,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
if entry.Follow.TargetAccount == nil {
|
||||
// Retrieve account.
|
||||
var err error
|
||||
entry.Follow.TargetAccount, err = c.state.DB.GetAccountByID(
|
||||
// Barebones is fine here.
|
||||
gtscontext.SetBarebones(ctx),
|
||||
entry.Follow.TargetAccountID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"db error getting target account for list entry %s: %w",
|
||||
entry.ID, err,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
var (
|
||||
username = entry.Follow.TargetAccount.Username
|
||||
domain = entry.Follow.TargetAccount.Domain
|
||||
)
|
||||
|
||||
if domain == "" {
|
||||
// Local account,
|
||||
// use our domain.
|
||||
domain = thisDomain
|
||||
}
|
||||
|
||||
records = append(records, []string{
|
||||
// List title: eg., Very cool list
|
||||
list.Title,
|
||||
// Account address: eg., someone@example.org
|
||||
// -- NOTE: without the leading '@'!
|
||||
username + "@" + domain,
|
||||
})
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// BlocksToCSV converts a slice of blocks into
|
||||
// a slice of CSV-compatible blocks records.
|
||||
func (c *Converter) BlocksToCSV(
|
||||
ctx context.Context,
|
||||
blocks []*gtsmodel.Block,
|
||||
) ([][]string, error) {
|
||||
// We need to know our own domain for this.
|
||||
// Try account domain, fall back to host.
|
||||
thisDomain := config.GetAccountDomain()
|
||||
if thisDomain == "" {
|
||||
thisDomain = config.GetHost()
|
||||
}
|
||||
|
||||
// NOTE: Mastodon-compatible blocks
|
||||
// CSV doesn't use column headers.
|
||||
records := make([][]string, 0, len(blocks))
|
||||
|
||||
// For each item, add a record.
|
||||
for _, block := range blocks {
|
||||
if block.TargetAccount == nil {
|
||||
// Retrieve target account.
|
||||
var err error
|
||||
block.TargetAccount, err = c.state.DB.GetAccountByID(
|
||||
// Barebones is fine here.
|
||||
gtscontext.SetBarebones(ctx),
|
||||
block.TargetAccountID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"db error getting target account for block %s: %w",
|
||||
block.ID, err,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
domain := block.TargetAccount.Domain
|
||||
if domain == "" {
|
||||
// Local account,
|
||||
// use our domain.
|
||||
domain = thisDomain
|
||||
}
|
||||
|
||||
records = append(records, []string{
|
||||
// Account address: eg., someone@example.org
|
||||
// -- NOTE: without the leading '@'!
|
||||
block.TargetAccount.Username + "@" + domain,
|
||||
})
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
||||
|
||||
// MutesToCSV converts a slice of mutes into
|
||||
// a slice of CSV-compatible mute records.
|
||||
func (c *Converter) MutesToCSV(
|
||||
ctx context.Context,
|
||||
mutes []*gtsmodel.UserMute,
|
||||
) ([][]string, error) {
|
||||
// Records should be length of
|
||||
// input + 1 so we can add headers.
|
||||
records := make([][]string, 1, len(mutes)+1)
|
||||
|
||||
// Add headers at the
|
||||
// top of records.
|
||||
records[0] = []string{
|
||||
"Account address",
|
||||
"Hide notifications",
|
||||
}
|
||||
|
||||
// We need to know our own domain for this.
|
||||
// Try account domain, fall back to host.
|
||||
thisDomain := config.GetAccountDomain()
|
||||
if thisDomain == "" {
|
||||
thisDomain = config.GetHost()
|
||||
}
|
||||
|
||||
// For each item, add a record.
|
||||
for _, mute := range mutes {
|
||||
if mute.TargetAccount == nil {
|
||||
// Retrieve target account.
|
||||
var err error
|
||||
mute.TargetAccount, err = c.state.DB.GetAccountByID(
|
||||
// Barebones is fine here.
|
||||
gtscontext.SetBarebones(ctx),
|
||||
mute.TargetAccountID,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, gtserror.Newf(
|
||||
"db error getting target account for mute %s: %w",
|
||||
mute.ID, err,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
domain := mute.TargetAccount.Domain
|
||||
if domain == "" {
|
||||
// Local account,
|
||||
// use our domain.
|
||||
domain = thisDomain
|
||||
}
|
||||
|
||||
records = append(records, []string{
|
||||
// Account address: eg., someone@example.org
|
||||
// -- NOTE: without the leading '@'!
|
||||
mute.TargetAccount.Username + "@" + domain,
|
||||
// Hide notifications: eg., true
|
||||
strconv.FormatBool(*mute.Notifications),
|
||||
})
|
||||
}
|
||||
|
||||
return records, nil
|
||||
}
|
|
@ -48,6 +48,11 @@ export interface GTSFetchArgs extends FetchArgs {
|
|||
* as FormData before submission.
|
||||
*/
|
||||
asForm?: boolean;
|
||||
/**
|
||||
* If set, then Accept header will
|
||||
* be set to the provided contentType.
|
||||
*/
|
||||
acceptContentType?: string;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -77,6 +82,10 @@ const gtsBaseQuery: BaseQueryFn<
|
|||
// Derive baseUrl dynamically.
|
||||
let baseUrl: string | undefined;
|
||||
|
||||
// Assume Accept value of
|
||||
// "application/json" by default.
|
||||
let accept = "application/json";
|
||||
|
||||
// Check if simple string baseUrl provided
|
||||
// as args, or if more complex args provided.
|
||||
if (typeof args === "string") {
|
||||
|
@ -101,11 +110,16 @@ const gtsBaseQuery: BaseQueryFn<
|
|||
});
|
||||
}
|
||||
|
||||
if (args.acceptContentType !== undefined) {
|
||||
accept = args.acceptContentType;
|
||||
}
|
||||
|
||||
// Delete any of our extended arguments
|
||||
// to avoid confusing fetchBaseQuery.
|
||||
delete args.baseUrl;
|
||||
delete args.discardEmpty;
|
||||
delete args.asForm;
|
||||
delete args.acceptContentType;
|
||||
}
|
||||
|
||||
if (!baseUrl) {
|
||||
|
@ -124,9 +138,21 @@ const gtsBaseQuery: BaseQueryFn<
|
|||
if (token != undefined) {
|
||||
headers.set('Authorization', token);
|
||||
}
|
||||
headers.set("Accept", "application/json");
|
||||
|
||||
headers.set("Accept", accept);
|
||||
return headers;
|
||||
},
|
||||
responseHandler: (response) => {
|
||||
// Return just text if caller has
|
||||
// set a custom accept content-type.
|
||||
if (accept !== "application/json") {
|
||||
return response.text();
|
||||
}
|
||||
|
||||
// Else return good old
|
||||
// fashioned JSON baby!
|
||||
return response.json();
|
||||
},
|
||||
})(args, api, extraOptions);
|
||||
};
|
||||
|
||||
|
|
138
web/source/settings/lib/query/user/export-import.ts
Normal file
138
web/source/settings/lib/query/user/export-import.ts
Normal file
|
@ -0,0 +1,138 @@
|
|||
/*
|
||||
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/>.
|
||||
*/
|
||||
|
||||
import fileDownload from "js-file-download";
|
||||
|
||||
import { gtsApi } from "../gts-api";
|
||||
import { FetchBaseQueryError } from "@reduxjs/toolkit/query";
|
||||
import { AccountExportStats } from "../../types/account";
|
||||
|
||||
const extended = gtsApi.injectEndpoints({
|
||||
endpoints: (build) => ({
|
||||
exportStats: build.query<AccountExportStats, void>({
|
||||
query: () => ({
|
||||
url: `/api/v1/exports/stats`
|
||||
})
|
||||
}),
|
||||
|
||||
exportFollowing: build.mutation<string | null, void>({
|
||||
async queryFn(_arg, _api, _extraOpts, fetchWithBQ) {
|
||||
const csvRes = await fetchWithBQ({
|
||||
url: `/api/v1/exports/following.csv`,
|
||||
acceptContentType: "text/csv",
|
||||
});
|
||||
if (csvRes.error) {
|
||||
return { error: csvRes.error as FetchBaseQueryError };
|
||||
}
|
||||
|
||||
if (csvRes.meta?.response?.status !== 200) {
|
||||
return { error: csvRes.data };
|
||||
}
|
||||
|
||||
fileDownload(csvRes.data, "following.csv", "text/csv");
|
||||
return { data: null };
|
||||
}
|
||||
}),
|
||||
|
||||
exportFollowers: build.mutation<string | null, void>({
|
||||
async queryFn(_arg, _api, _extraOpts, fetchWithBQ) {
|
||||
const csvRes = await fetchWithBQ({
|
||||
url: `/api/v1/exports/followers.csv`,
|
||||
acceptContentType: "text/csv",
|
||||
});
|
||||
if (csvRes.error) {
|
||||
return { error: csvRes.error as FetchBaseQueryError };
|
||||
}
|
||||
|
||||
if (csvRes.meta?.response?.status !== 200) {
|
||||
return { error: csvRes.data };
|
||||
}
|
||||
|
||||
fileDownload(csvRes.data, "followers.csv", "text/csv");
|
||||
return { data: null };
|
||||
}
|
||||
}),
|
||||
|
||||
exportLists: build.mutation<string | null, void>({
|
||||
async queryFn(_arg, _api, _extraOpts, fetchWithBQ) {
|
||||
const csvRes = await fetchWithBQ({
|
||||
url: `/api/v1/exports/lists.csv`,
|
||||
acceptContentType: "text/csv",
|
||||
});
|
||||
if (csvRes.error) {
|
||||
return { error: csvRes.error as FetchBaseQueryError };
|
||||
}
|
||||
|
||||
if (csvRes.meta?.response?.status !== 200) {
|
||||
return { error: csvRes.data };
|
||||
}
|
||||
|
||||
fileDownload(csvRes.data, "lists.csv", "text/csv");
|
||||
return { data: null };
|
||||
}
|
||||
}),
|
||||
|
||||
exportBlocks: build.mutation<string | null, void>({
|
||||
async queryFn(_arg, _api, _extraOpts, fetchWithBQ) {
|
||||
const csvRes = await fetchWithBQ({
|
||||
url: `/api/v1/exports/blocks.csv`,
|
||||
acceptContentType: "text/csv",
|
||||
});
|
||||
if (csvRes.error) {
|
||||
return { error: csvRes.error as FetchBaseQueryError };
|
||||
}
|
||||
|
||||
if (csvRes.meta?.response?.status !== 200) {
|
||||
return { error: csvRes.data };
|
||||
}
|
||||
|
||||
fileDownload(csvRes.data, "blocks.csv", "text/csv");
|
||||
return { data: null };
|
||||
}
|
||||
}),
|
||||
|
||||
exportMutes: build.mutation<string | null, void>({
|
||||
async queryFn(_arg, _api, _extraOpts, fetchWithBQ) {
|
||||
const csvRes = await fetchWithBQ({
|
||||
url: `/api/v1/exports/mutes.csv`,
|
||||
acceptContentType: "text/csv",
|
||||
});
|
||||
if (csvRes.error) {
|
||||
return { error: csvRes.error as FetchBaseQueryError };
|
||||
}
|
||||
|
||||
if (csvRes.meta?.response?.status !== 200) {
|
||||
return { error: csvRes.data };
|
||||
}
|
||||
|
||||
fileDownload(csvRes.data, "mutes.csv", "text/csv");
|
||||
return { data: null };
|
||||
}
|
||||
}),
|
||||
})
|
||||
});
|
||||
|
||||
export const {
|
||||
useExportStatsQuery,
|
||||
useExportFollowingMutation,
|
||||
useExportFollowersMutation,
|
||||
useExportListsMutation,
|
||||
useExportBlocksMutation,
|
||||
useExportMutesMutation,
|
||||
} = extended;
|
|
@ -110,3 +110,13 @@ export interface ActionAccountParams {
|
|||
action: "suspend";
|
||||
reason: string;
|
||||
}
|
||||
|
||||
export interface AccountExportStats {
|
||||
media_storage: string;
|
||||
followers_count: number;
|
||||
following_count: number;
|
||||
statuses_count: number;
|
||||
lists_count: number;
|
||||
blocks_count: number;
|
||||
mutes_count: number;
|
||||
}
|
||||
|
|
|
@ -1464,6 +1464,39 @@ button.tab-button {
|
|||
}
|
||||
}
|
||||
|
||||
.export-data {
|
||||
.export-buttons-wrapper {
|
||||
display: grid;
|
||||
max-width: fit-content;
|
||||
gap: 0.5rem;
|
||||
|
||||
.stats-and-button {
|
||||
display: grid;
|
||||
grid-template-columns: 13rem 1fr;
|
||||
align-items: center;
|
||||
gap: 0.25rem;
|
||||
|
||||
.mutation-button {
|
||||
width: 100%;
|
||||
overflow-x: hidden;
|
||||
|
||||
button {
|
||||
font-size: 1rem;
|
||||
width: 100%;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@media screen and (max-width: 35rem) {
|
||||
gap: 1rem;
|
||||
|
||||
.stats-and-button {
|
||||
grid-template-columns: auto;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@media screen and (orientation: portrait) {
|
||||
.reports .report .byline {
|
||||
grid-template-columns: 1fr;
|
||||
|
|
173
web/source/settings/views/user/export-import/export.tsx
Normal file
173
web/source/settings/views/user/export-import/export.tsx
Normal file
|
@ -0,0 +1,173 @@
|
|||
/*
|
||||
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/>.
|
||||
*/
|
||||
|
||||
import React from "react";
|
||||
import {
|
||||
useExportFollowingMutation,
|
||||
useExportFollowersMutation,
|
||||
useExportListsMutation,
|
||||
useExportBlocksMutation,
|
||||
useExportMutesMutation,
|
||||
} from "../../../lib/query/user/export-import";
|
||||
import MutationButton from "../../../components/form/mutation-button";
|
||||
import useFormSubmit from "../../../lib/form/submit";
|
||||
import { useValue } from "../../../lib/form";
|
||||
import { AccountExportStats } from "../../../lib/types/account";
|
||||
|
||||
export default function Export({ exportStats }: { exportStats: AccountExportStats }) {
|
||||
const [exportFollowing, exportFollowingResult] = useFormSubmit(
|
||||
// Use a dummy value.
|
||||
{ type: useValue("exportFollowing", "exportFollowing") },
|
||||
// Mutation we're wrapping.
|
||||
useExportFollowingMutation(),
|
||||
// Form never changes but
|
||||
// we want to always trigger.
|
||||
{ changedOnly: false },
|
||||
);
|
||||
|
||||
const [exportFollowers, exportFollowersResult] = useFormSubmit(
|
||||
// Use a dummy value.
|
||||
{ type: useValue("exportFollowers", "exportFollowers") },
|
||||
// Mutation we're wrapping.
|
||||
useExportFollowersMutation(),
|
||||
// Form never changes but
|
||||
// we want to always trigger.
|
||||
{ changedOnly: false },
|
||||
);
|
||||
|
||||
const [exportLists, exportListsResult] = useFormSubmit(
|
||||
// Use a dummy value.
|
||||
{ type: useValue("exportLists", "exportLists") },
|
||||
// Mutation we're wrapping.
|
||||
useExportListsMutation(),
|
||||
// Form never changes but
|
||||
// we want to always trigger.
|
||||
{ changedOnly: false },
|
||||
);
|
||||
|
||||
|
||||
const [exportBlocks, exportBlocksResult] = useFormSubmit(
|
||||
// Use a dummy value.
|
||||
{ type: useValue("exportBlocks", "exportBlocks") },
|
||||
// Mutation we're wrapping.
|
||||
useExportBlocksMutation(),
|
||||
// Form never changes but
|
||||
// we want to always trigger.
|
||||
{ changedOnly: false },
|
||||
);
|
||||
|
||||
const [exportMutes, exportMutesResult] = useFormSubmit(
|
||||
// Use a dummy value.
|
||||
{ type: useValue("exportMutes", "exportMutes") },
|
||||
// Mutation we're wrapping.
|
||||
useExportMutesMutation(),
|
||||
// Form never changes but
|
||||
// we want to always trigger.
|
||||
{ changedOnly: false },
|
||||
);
|
||||
|
||||
return (
|
||||
<form className="export-data">
|
||||
<div className="form-section-docs">
|
||||
<h3>Export Data</h3>
|
||||
<a
|
||||
href="https://docs.gotosocial.org/en/latest/user_guide/export-import#export"
|
||||
target="_blank"
|
||||
className="docslink"
|
||||
rel="noreferrer"
|
||||
>
|
||||
Learn more about this section (opens in a new tab)
|
||||
</a>
|
||||
</div>
|
||||
|
||||
<div className="export-buttons-wrapper">
|
||||
<div className="stats-and-button">
|
||||
<span className="text-cutoff">
|
||||
Following {exportStats.following_count} account{ exportStats.following_count !== 1 && "s" }
|
||||
</span>
|
||||
<MutationButton
|
||||
className="text-cutoff"
|
||||
label="Download following.csv"
|
||||
type="button"
|
||||
onClick={() => exportFollowing()}
|
||||
result={exportFollowingResult}
|
||||
showError={true}
|
||||
disabled={exportStats.following_count === 0}
|
||||
/>
|
||||
</div>
|
||||
<div className="stats-and-button">
|
||||
<span className="text-cutoff">
|
||||
Followed by {exportStats.followers_count} account{ exportStats.followers_count !== 1 && "s" }
|
||||
</span>
|
||||
<MutationButton
|
||||
className="text-cutoff"
|
||||
label="Download followers.csv"
|
||||
type="button"
|
||||
onClick={() => exportFollowers()}
|
||||
result={exportFollowersResult}
|
||||
showError={true}
|
||||
disabled={exportStats.followers_count === 0}
|
||||
/>
|
||||
</div>
|
||||
<div className="stats-and-button">
|
||||
<span className="text-cutoff">
|
||||
Created {exportStats.lists_count} list{ exportStats.lists_count !== 1 && "s" }
|
||||
</span>
|
||||
<MutationButton
|
||||
className="text-cutoff"
|
||||
label="Download lists.csv"
|
||||
type="button"
|
||||
onClick={() => exportLists()}
|
||||
result={exportListsResult}
|
||||
showError={true}
|
||||
disabled={exportStats.lists_count === 0}
|
||||
/>
|
||||
</div>
|
||||
<div className="stats-and-button">
|
||||
<span className="text-cutoff">
|
||||
Blocking {exportStats.blocks_count} account{ exportStats.blocks_count !== 1 && "s" }
|
||||
</span>
|
||||
<MutationButton
|
||||
className="text-cutoff"
|
||||
label="Download blocks.csv"
|
||||
type="button"
|
||||
onClick={() => exportBlocks()}
|
||||
result={exportBlocksResult}
|
||||
showError={true}
|
||||
disabled={exportStats.blocks_count === 0}
|
||||
/>
|
||||
</div>
|
||||
<div className="stats-and-button">
|
||||
<span className="text-cutoff">
|
||||
Muting {exportStats.mutes_count} account{ exportStats.mutes_count !== 1 && "s" }
|
||||
</span>
|
||||
<MutationButton
|
||||
className="text-cutoff"
|
||||
label="Download mutes.csv"
|
||||
type="button"
|
||||
onClick={() => exportMutes()}
|
||||
result={exportMutesResult}
|
||||
showError={true}
|
||||
disabled={exportStats.mutes_count === 0}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
</form>
|
||||
);
|
||||
}
|
57
web/source/settings/views/user/export-import/index.tsx
Normal file
57
web/source/settings/views/user/export-import/index.tsx
Normal file
|
@ -0,0 +1,57 @@
|
|||
/*
|
||||
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/>.
|
||||
*/
|
||||
|
||||
import React from "react";
|
||||
import Export from "./export";
|
||||
import Loading from "../../../components/loading";
|
||||
import { Error } from "../../../components/error";
|
||||
import { useExportStatsQuery } from "../../../lib/query/user/export-import";
|
||||
|
||||
export default function ExportImport() {
|
||||
const {
|
||||
data: exportStats,
|
||||
isLoading,
|
||||
isFetching,
|
||||
isError,
|
||||
error,
|
||||
} = useExportStatsQuery();
|
||||
|
||||
if (isLoading || isFetching) {
|
||||
return <Loading />;
|
||||
}
|
||||
|
||||
if (isError) {
|
||||
return <Error error={error} />;
|
||||
}
|
||||
|
||||
if (exportStats === undefined) {
|
||||
throw "undefined account export stats";
|
||||
}
|
||||
|
||||
return (
|
||||
<>
|
||||
<h1>Export & Import</h1>
|
||||
<p>
|
||||
On this page you can export data from your GoToSocial account, or import data into
|
||||
your GoToSocial account. All exports and imports use Mastodon-compatible CSV files.
|
||||
</p>
|
||||
<Export exportStats={exportStats} />
|
||||
</>
|
||||
);
|
||||
}
|
|
@ -53,6 +53,11 @@ export default function UserMenu() {
|
|||
itemUrl="migration"
|
||||
icon="fa-exchange"
|
||||
/>
|
||||
<MenuItem
|
||||
name="Export & Import"
|
||||
itemUrl="export-import"
|
||||
icon="fa-floppy-o"
|
||||
/>
|
||||
</MenuItem>
|
||||
);
|
||||
}
|
||||
|
|
|
@ -25,12 +25,14 @@ import UserProfile from "./profile";
|
|||
import UserMigration from "./migration";
|
||||
import PostSettings from "./posts";
|
||||
import EmailPassword from "./emailpassword";
|
||||
import ExportImport from "./export-import";
|
||||
|
||||
/**
|
||||
* - /settings/user/profile
|
||||
* - /settings/user/posts
|
||||
* - /settings/user/emailpassword
|
||||
* - /settings/user/migration
|
||||
* - /settings/user/export-import
|
||||
*/
|
||||
export default function UserRouter() {
|
||||
const baseUrl = useBaseUrl();
|
||||
|
@ -46,6 +48,7 @@ export default function UserRouter() {
|
|||
<Route path="/posts" component={PostSettings} />
|
||||
<Route path="/emailpassword" component={EmailPassword} />
|
||||
<Route path="/migration" component={UserMigration} />
|
||||
<Route path="/export-import" component={ExportImport} />
|
||||
<Route><Redirect to="/profile" /></Route>
|
||||
</Switch>
|
||||
</ErrorBoundary>
|
||||
|
|
Loading…
Reference in a new issue