token.go RawHistoryFolder

1// SPDX-License-Identifier: AGPL-3.0-or-later
2
3package rpc
4
5import (
6	"context"
7
8	"connectrpc.com/connect"
9
10	pb "bnema.dev/ermite-api/gen/ermite"
11	"bnema.dev/ermite/internal/domain"
12	"bnema.dev/ermite/internal/ports"
13)
14
15// tokenService implements ermiteconnect.TokenServiceHandler.
16type tokenService struct{ tokens ports.TokenService }
17
18func (s *tokenService) CreateToken(ctx context.Context, req *connect.Request[pb.CreateTokenRequest]) (*connect.Response[pb.CreateTokenResponse], error) {
19	params := ports.CreateTokenParams{
20		Label:     req.Msg.GetLabel(),
21		ExpiresAt: goTime(req.Msg.GetExpiresAt()),
22	}
23	for _, sc := range req.Msg.GetScopes() {
24		scope, err := scopeFromPB(sc)
25		if err != nil {
26			return nil, connectError(ctx, err)
27		}
28		params.Scopes = append(params.Scopes, scope)
29	}
30	for _, r := range req.Msg.GetRepos() {
31		name, err := domain.ParseRepoName(r)
32		if err != nil {
33			return nil, connectError(ctx, err)
34		}
35		params.Repos = append(params.Repos, name)
36	}
37	params.BranchPrefixes = req.Msg.GetBranchPrefixes()
38	created, err := s.tokens.Create(ctx, caller(ctx), params)
39	if err != nil {
40		return nil, connectError(ctx, err)
41	}
42	return connect.NewResponse(&pb.CreateTokenResponse{Token: tokenToPB(created.Token), Secret: created.Secret}), nil
43}
44
45func (s *tokenService) ListTokens(ctx context.Context, req *connect.Request[pb.ListTokensRequest]) (*connect.Response[pb.ListTokensResponse], error) {
46	page, err := s.tokens.ListPage(ctx, caller(ctx), int(req.Msg.GetPageSize()), req.Msg.GetPageToken())
47	if err != nil {
48		return nil, connectError(ctx, err)
49	}
50	out := &pb.ListTokensResponse{NextPageToken: page.NextPageToken, Tokens: make([]*pb.Token, 0, len(page.Tokens))}
51	for _, t := range page.Tokens {
52		out.Tokens = append(out.Tokens, tokenToPB(t))
53	}
54	return connect.NewResponse(out), nil
55}
56
57func (s *tokenService) RevokeToken(ctx context.Context, req *connect.Request[pb.RevokeTokenRequest]) (*connect.Response[pb.RevokeTokenResponse], error) {
58	if err := s.tokens.Revoke(ctx, caller(ctx), req.Msg.GetId()); err != nil {
59		return nil, connectError(ctx, err)
60	}
61	return connect.NewResponse(&pb.RevokeTokenResponse{}), nil
62}