mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-31 05:51:59 +08:00
107 lines
3.0 KiB
Go
107 lines
3.0 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"go.signoz.io/signoz/ee/query-service/model"
|
|
basemodel "go.signoz.io/signoz/pkg/query-service/model"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func (m *modelDao) CreatePAT(ctx context.Context, p *model.PAT) basemodel.BaseApiError {
|
|
_, err := m.DB().ExecContext(ctx,
|
|
"INSERT INTO personal_access_tokens (user_id, token, name, created_at, expires_at) VALUES ($1, $2, $3, $4, $5)",
|
|
p.UserID,
|
|
p.Token,
|
|
p.Name,
|
|
p.CreatedAt,
|
|
p.ExpiresAt)
|
|
if err != nil {
|
|
zap.S().Errorf("Failed to insert PAT in db, err: %v", zap.Error(err))
|
|
return model.InternalError(fmt.Errorf("PAT insertion failed"))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *modelDao) ListPATs(ctx context.Context, userID string) ([]model.PAT, basemodel.BaseApiError) {
|
|
pats := []model.PAT{}
|
|
|
|
if err := m.DB().Select(&pats, `SELECT * FROM personal_access_tokens WHERE user_id=?;`, userID); err != nil {
|
|
zap.S().Errorf("Failed to fetch PATs for user: %s, err: %v", userID, zap.Error(err))
|
|
return nil, model.InternalError(fmt.Errorf("failed to fetch PATs"))
|
|
}
|
|
return pats, nil
|
|
}
|
|
|
|
func (m *modelDao) DeletePAT(ctx context.Context, id string) basemodel.BaseApiError {
|
|
_, err := m.DB().ExecContext(ctx, `DELETE from personal_access_tokens where id=?;`, id)
|
|
if err != nil {
|
|
zap.S().Errorf("Failed to delete PAT, err: %v", zap.Error(err))
|
|
return model.InternalError(fmt.Errorf("failed to delete PAT"))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *modelDao) GetPAT(ctx context.Context, token string) (*model.PAT, basemodel.BaseApiError) {
|
|
pats := []model.PAT{}
|
|
|
|
if err := m.DB().Select(&pats, `SELECT * FROM personal_access_tokens WHERE token=?;`, token); err != nil {
|
|
return nil, model.InternalError(fmt.Errorf("failed to fetch PAT"))
|
|
}
|
|
|
|
if len(pats) != 1 {
|
|
return nil, &model.ApiError{
|
|
Typ: model.ErrorInternal,
|
|
Err: fmt.Errorf("found zero or multiple PATs with same token, %s", token),
|
|
}
|
|
}
|
|
|
|
return &pats[0], nil
|
|
}
|
|
|
|
func (m *modelDao) GetPATByID(ctx context.Context, id string) (*model.PAT, basemodel.BaseApiError) {
|
|
pats := []model.PAT{}
|
|
|
|
if err := m.DB().Select(&pats, `SELECT * FROM personal_access_tokens WHERE id=?;`, id); err != nil {
|
|
return nil, model.InternalError(fmt.Errorf("failed to fetch PAT"))
|
|
}
|
|
|
|
if len(pats) != 1 {
|
|
return nil, &model.ApiError{
|
|
Typ: model.ErrorInternal,
|
|
Err: fmt.Errorf("found zero or multiple PATs with same token"),
|
|
}
|
|
}
|
|
|
|
return &pats[0], nil
|
|
}
|
|
|
|
func (m *modelDao) GetUserByPAT(ctx context.Context, token string) (*basemodel.UserPayload, basemodel.BaseApiError) {
|
|
users := []basemodel.UserPayload{}
|
|
|
|
query := `SELECT
|
|
u.id,
|
|
u.name,
|
|
u.email,
|
|
u.password,
|
|
u.created_at,
|
|
u.profile_picture_url,
|
|
u.org_id,
|
|
u.group_id
|
|
FROM users u, personal_access_tokens p
|
|
WHERE u.id = p.user_id and p.token=?;`
|
|
|
|
if err := m.DB().Select(&users, query, token); err != nil {
|
|
return nil, model.InternalError(fmt.Errorf("failed to fetch user from PAT, err: %v", err))
|
|
}
|
|
|
|
if len(users) != 1 {
|
|
return nil, &model.ApiError{
|
|
Typ: model.ErrorInternal,
|
|
Err: fmt.Errorf("found zero or multiple users with same PAT token"),
|
|
}
|
|
}
|
|
return &users[0], nil
|
|
}
|