From e02b9ef3ee4394db1a387b146aa5faeea95efa3a Mon Sep 17 00:00:00 2001 From: Xe Iaso Date: Mon, 12 Feb 2024 04:32:43 -0800 Subject: cmd/patreon-saasproxy: use twirp/protobuf instead of yolo json Signed-off-by: Xe Iaso --- cmd/patreon-saasproxy/main.go | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'cmd') diff --git a/cmd/patreon-saasproxy/main.go b/cmd/patreon-saasproxy/main.go index cfe9717..b44a697 100644 --- a/cmd/patreon-saasproxy/main.go +++ b/cmd/patreon-saasproxy/main.go @@ -1,6 +1,7 @@ package main import ( + "context" "encoding/base64" "encoding/json" "expvar" @@ -14,6 +15,8 @@ import ( "github.com/facebookgo/flagenv" _ "github.com/joho/godotenv/autoload" "golang.org/x/oauth2" + "google.golang.org/protobuf/types/known/emptypb" + "google.golang.org/protobuf/types/known/timestamppb" "gopkg.in/mxpv/patreon-go.v1" "tailscale.com/client/tailscale" "tailscale.com/hostinfo" @@ -21,6 +24,7 @@ import ( "tailscale.com/tsnet" "tailscale.com/tsweb" "xeiaso.net/v4/internal" + "xeiaso.net/v4/internal/adminpb" ) var ( @@ -110,6 +114,9 @@ func main() { log.Fatal(err) } + ph := adminpb.NewPatreonServer(s) + http.Handle(adminpb.PatreonPathPrefix, ph) + slog.Info("listening over tailscale", "hostname", *tailscaleHostname) log.Fatal(http.Serve(ln, nil)) @@ -120,6 +127,21 @@ type Server struct { cts oauth2.TokenSource } +func (s *Server) GetToken(ctx context.Context, _ *emptypb.Empty) (*adminpb.PatreonToken, error) { + token, err := s.cts.Token() + if err != nil { + slog.Error("token fetch failed", "err", err) + return nil, err + } + + return &adminpb.PatreonToken{ + AccessToken: token.AccessToken, + TokenType: token.TokenType, + RefreshToken: token.RefreshToken, + Expiry: timestamppb.New(token.Expiry), + }, nil +} + func (s *Server) GiveToken(w http.ResponseWriter, r *http.Request) { whois, err := s.lc.WhoIs(r.Context(), r.RemoteAddr) if err != nil { -- cgit v1.2.3