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/app"
12 "bnema.dev/ermite/internal/domain"
13)
14
15// tokenService implements ermiteconnect.TokenServiceHandler.
16type tokenService struct{ tokens *app.Tokens }
17
18func (s *tokenService) CreateToken(ctx context.Context, req *connect.Request[pb.CreateTokenRequest]) (*connect.Response[pb.CreateTokenResponse], error) {
19 params := app.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 created, err := s.tokens.Create(ctx, caller(ctx), params)
38 if err != nil {
39 return nil, connectError(ctx, err)
40 }
41 return connect.NewResponse(&pb.CreateTokenResponse{Token: tokenToPB(created.Token), Secret: created.Secret}), nil
42}
43
44func (s *tokenService) ListTokens(ctx context.Context, req *connect.Request[pb.ListTokensRequest]) (*connect.Response[pb.ListTokensResponse], error) {
45 page, err := s.tokens.ListPage(ctx, caller(ctx), int(req.Msg.GetPageSize()), req.Msg.GetPageToken())
46 if err != nil {
47 return nil, connectError(ctx, err)
48 }
49 out := &pb.ListTokensResponse{NextPageToken: page.NextPageToken, Tokens: make([]*pb.Token, 0, len(page.Tokens))}
50 for _, t := range page.Tokens {
51 out.Tokens = append(out.Tokens, tokenToPB(t))
52 }
53 return connect.NewResponse(out), nil
54}
55
56func (s *tokenService) RevokeToken(ctx context.Context, req *connect.Request[pb.RevokeTokenRequest]) (*connect.Response[pb.RevokeTokenResponse], error) {
57 if err := s.tokens.Revoke(ctx, caller(ctx), req.Msg.GetId()); err != nil {
58 return nil, connectError(ctx, err)
59 }
60 return connect.NewResponse(&pb.RevokeTokenResponse{}), nil
61}