focalboard/server/services/store/sqlstore/workspaces.go

148 lines
3.1 KiB
Go
Raw Normal View History

2021-01-14 01:56:01 +01:00
package sqlstore
import (
"encoding/json"
"time"
2021-01-26 23:13:46 +01:00
"github.com/mattermost/focalboard/server/model"
"github.com/mattermost/focalboard/server/utils"
2021-01-14 01:56:01 +01:00
"github.com/mattermost/mattermost-server/v6/shared/mlog"
2021-01-14 01:56:01 +01:00
sq "github.com/Masterminds/squirrel"
)
func (s *SQLStore) UpsertWorkspaceSignupToken(workspace model.Workspace) error {
now := time.Now().Unix()
query := s.getQueryBuilder().
2021-04-17 09:06:57 +02:00
Insert(s.tablePrefix+"workspaces").
2021-01-14 01:56:01 +01:00
Columns(
"id",
"signup_token",
"modified_by",
"update_at",
).
Values(
workspace.ID,
workspace.SignupToken,
workspace.ModifiedBy,
now,
)
if s.dbType == mysqlDBType {
query = query.Suffix("ON DUPLICATE KEY UPDATE signup_token = ?, modified_by = ?, update_at = ?",
workspace.SignupToken, workspace.ModifiedBy, now)
} else {
query = query.Suffix(
`ON CONFLICT (id)
DO UPDATE SET signup_token = EXCLUDED.signup_token, modified_by = EXCLUDED.modified_by, update_at = EXCLUDED.update_at`,
)
}
2021-01-14 01:56:01 +01:00
_, err := query.Exec()
return err
}
func (s *SQLStore) UpsertWorkspaceSettings(workspace model.Workspace) error {
now := time.Now().Unix()
signupToken := utils.CreateGUID()
2021-01-14 01:56:01 +01:00
settingsJSON, err := json.Marshal(workspace.Settings)
if err != nil {
return err
}
query := s.getQueryBuilder().
2021-04-17 09:06:57 +02:00
Insert(s.tablePrefix+"workspaces").
2021-01-14 01:56:01 +01:00
Columns(
"id",
"signup_token",
2021-01-14 01:56:01 +01:00
"settings",
"modified_by",
"update_at",
).
Values(
workspace.ID,
signupToken,
2021-01-14 01:56:01 +01:00
settingsJSON,
workspace.ModifiedBy,
now,
)
if s.dbType == mysqlDBType {
query = query.Suffix("ON DUPLICATE KEY UPDATE settings = ?, modified_by = ?, update_at = ?", settingsJSON, workspace.ModifiedBy, now)
} else {
query = query.Suffix(
`ON CONFLICT (id)
DO UPDATE SET settings = EXCLUDED.settings, modified_by = EXCLUDED.modified_by, update_at = EXCLUDED.update_at`,
)
}
2021-01-14 01:56:01 +01:00
_, err = query.Exec()
return err
}
func (s *SQLStore) GetWorkspace(id string) (*model.Workspace, error) {
2021-01-14 01:56:01 +01:00
var settingsJSON string
query := s.getQueryBuilder().
Select(
"id",
"signup_token",
"COALESCE(settings, '{}')",
2021-01-14 01:56:01 +01:00
"modified_by",
"update_at",
).
2021-04-17 09:06:57 +02:00
From(s.tablePrefix + "workspaces").
Where(sq.Eq{"id": id})
2021-01-14 01:56:01 +01:00
row := query.QueryRow()
workspace := model.Workspace{}
err := row.Scan(
&workspace.ID,
&workspace.SignupToken,
&settingsJSON,
&workspace.ModifiedBy,
&workspace.UpdateAt,
)
if err != nil {
return nil, err
}
err = json.Unmarshal([]byte(settingsJSON), &workspace.Settings)
if err != nil {
s.logger.Error(`ERROR GetWorkspace settings json.Unmarshal`, mlog.Err(err))
2021-01-14 01:56:01 +01:00
return nil, err
}
return &workspace, nil
}
func (s *SQLStore) HasWorkspaceAccess(userID string, workspaceID string) (bool, error) {
return true, nil
}
func (s *SQLStore) GetWorkspaceCount() (int64, error) {
query := s.getQueryBuilder().
Select(
"COUNT(*) AS count",
).
From(s.tablePrefix + "workspaces")
rows, err := query.Query()
if err != nil {
s.logger.Error("ERROR GetWorkspaceCount", mlog.Err(err))
return 0, err
}
defer s.CloseRows(rows)
var count int64
rows.Next()
err = rows.Scan(&count)
if err != nil {
s.logger.Error("Failed to fetch workspace count", mlog.Err(err))
return 0, err
}
return count, nil
}