98 lines
1.9 KiB
Go
98 lines
1.9 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost-octo-tasks/server/model"
|
|
|
|
sq "github.com/Masterminds/squirrel"
|
|
)
|
|
|
|
func (s *SQLStore) UpsertWorkspaceSignupToken(workspace model.Workspace) error {
|
|
now := time.Now().Unix()
|
|
|
|
query := s.getQueryBuilder().
|
|
Insert("workspaces").
|
|
Columns(
|
|
"id",
|
|
"signup_token",
|
|
"modified_by",
|
|
"update_at",
|
|
).
|
|
Values(
|
|
workspace.ID,
|
|
workspace.SignupToken,
|
|
workspace.ModifiedBy,
|
|
now,
|
|
).
|
|
Suffix("ON CONFLICT (id) DO UPDATE SET signup_token = EXCLUDED.signup_token, modified_by = EXCLUDED.modified_by, update_at = EXCLUDED.update_at")
|
|
|
|
_, err := query.Exec()
|
|
return err
|
|
}
|
|
|
|
func (s *SQLStore) UpsertWorkspaceSettings(workspace model.Workspace) error {
|
|
now := time.Now().Unix()
|
|
|
|
settingsJSON, err := json.Marshal(workspace.Settings)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
query := s.getQueryBuilder().
|
|
Insert("workspaces").
|
|
Columns(
|
|
"id",
|
|
"settings",
|
|
"modified_by",
|
|
"update_at",
|
|
).
|
|
Values(
|
|
workspace.ID,
|
|
settingsJSON,
|
|
workspace.ModifiedBy,
|
|
now,
|
|
).
|
|
Suffix("ON CONFLICT (id) DO UPDATE SET settings = EXCLUDED.settings, modified_by = EXCLUDED.modified_by, update_at = EXCLUDED.update_at")
|
|
|
|
_, err = query.Exec()
|
|
return err
|
|
}
|
|
|
|
func (s *SQLStore) GetWorkspace(ID string) (*model.Workspace, error) {
|
|
var settingsJSON string
|
|
|
|
query := s.getQueryBuilder().
|
|
Select(
|
|
"id",
|
|
"signup_token",
|
|
"COALESCE(\"settings\", '{}')",
|
|
"modified_by",
|
|
"update_at",
|
|
).
|
|
From("workspaces").
|
|
Where(sq.Eq{"id": ID})
|
|
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 {
|
|
log.Printf(`ERROR GetWorkspace settings json.Unmarshal: %v`, err)
|
|
return nil, err
|
|
}
|
|
|
|
return &workspace, nil
|
|
}
|