commit b30dd42c5306ef755e29e4a10e9f882d8a1ac71e · hailey.at/atproto-oauth-golang

.gitignore

+1 -1

cmd/web_server_demo/handle_auth.go

···

       101
       101
        
       	}

     

       102
       102
        
       

     

       103
       103
        
       	u, _ := url.Parse(meta.AuthorizationEndpoint)

     

       104
       104
       -
       	u.RawQuery = fmt.Sprintf("client_id=%s&request_uri=%s", url.QueryEscape(serverMetadataUrl), parResp.RequestUri)

     

       104
       104
       +
       	u.RawQuery = fmt.Sprintf("client_id=%s&request_uri=%s", url.QueryEscape(s.args.UrlRoot+serverMetadataPath), parResp.RequestUri)

     

       105
       105
        
       

     

       106
       106
        
       	sess, err := session.Get("session", e)

     

       107
       107
        
       	if err != nil {

+1 -1

cmd/web_server_demo/handle_post.go

···

       35
       35
        
       		return err

     

       36
       36
        
       	}

     

       37
       37
        
       

     

       38
       38
       -
       	return e.File(getFilePath("make-post.html"))

     

       38
       38
       +
       	return e.File(s.getFilePath("make-post.html"))

     

       39
       39
        
       }

+70 -39

cmd/web_server_demo/main.go

···

       22
       22
        
       )

     

       23
       23
        
       

     

       24
       24
        
       var (

     

       25
       25
       -
       	ctx               = context.Background()

     

       26
       26
       -
       	serverAddr        = os.Getenv("OAUTH_TEST_SERVER_ADDR")

     

       27
       27
       -
       	serverUrlRoot     = os.Getenv("OAUTH_TEST_SERVER_URL_ROOT")

     

       28
       28
       -
       	staticFilePath    = os.Getenv("OAUTH_TEST_SERVER_STATIC_PATH")

     

       29
       29
       -
       	sessionSecret     = os.Getenv("OAUTH_TEST_SESSION_SECRET")

     

       30
       30
       -
       	serverMetadataUrl = fmt.Sprintf("%s/oauth/client-metadata.json", serverUrlRoot)

     

       31
       31
       -
       	serverCallbackUrl = fmt.Sprintf("%s/callback", serverUrlRoot)

     

       32
       32
       -
       	pdsUrl            = os.Getenv("OAUTH_TEST_PDS_URL")

     

       33
       33
       -
       	scope             = "atproto transition:generic"

     

       25
       25
       +
       	ctx                = context.Background()

     

       26
       26
       +
       	serverMetadataPath = "/oauth/client-metadata.json"

     

       27
       27
       +
       	serverCallbackPath = "/callback"

     

       28
       28
       +
       	scope              = "atproto transition:generic"

     

       34
       29
        
       )

     

       35
       30
        
       

     

       36
       31
        
       func main() {

     

       37
       32
        
       	app := &cli.App{

     

       38
       33
        
       		Name:   "atproto-goauth-demo-webserver",

     

       39
       34
        
       		Action: run,

     

       40
       40
       -
       	}

     

       41
       41
       -
       

     

       42
       42
       -
       	if serverUrlRoot == "" {

     

       43
       43
       -
       		panic(fmt.Errorf("no server url root set in env file"))

     

       35
       35
       +
       		Flags: []cli.Flag{

     

       36
       36
       +
       			&cli.StringFlag{

     

       37
       37
       +
       				Name:    "addr",

     

       38
       38
       +
       				Value:   ":8080",

     

       39
       39
       +
       				EnvVars: []string{"OAUTH_TEST_SERVER_ADDR"},

     

       40
       40
       +
       			},

     

       41
       41
       +
       			&cli.StringFlag{

     

       42
       42
       +
       				Name:     "url-root",

     

       43
       43
       +
       				Required: true,

     

       44
       44
       +
       				EnvVars:  []string{"OAUTH_TEST_SERVER_URL_ROOT"},

     

       45
       45
       +
       			},

     

       46
       46
       +
       			&cli.StringFlag{

     

       47
       47
       +
       				Name:    "static-file-path",

     

       48
       48
       +
       				Value:   "./cmd/web_server_demo/html",

     

       49
       49
       +
       				EnvVars: []string{"OAUTH_TEST_SERVER_STATIC_PATH"},

     

       50
       50
       +
       			},

     

       51
       51
       +
       			&cli.StringFlag{

     

       52
       52
       +
       				Name:    "session-secret",

     

       53
       53
       +
       				Value:   "session-secret",

     

       54
       54
       +
       				EnvVars: []string{"OAUTH_TEST_SERVER_SESSION_SECRET"},

     

       55
       55
       +
       			},

     

       56
       56
       +
       		},

     

       44
       57
        
       	}

     

       45
       58
        
       

     

       46
       46
       -
       	app.RunAndExitOnError()

     

       59
       59
       +
       	app.Run(os.Args)

     

       47
       60
        
       }

     

       48
       61
        
       

     

       49
       62
        
       type TestServer struct {

     
···

       53
       66
        
       	oauthClient  *oauth.Client

     

       54
       67
        
       	xrpcCli      *oauth.XrpcClient

     

       55
       68
        
       	jwksResponse *oauth_helpers.JwksResponseObject

     

       69
       69
       +
       	args         ServerArgs

     

       56
       70
        
       }

     

       57
       71
        
       

     

       58
       72
        
       type TemplateRenderer struct {

     

       59
       73
        
       	templates *template.Template

     

       60
       74
        
       }

     

       61
       75
        
       

     

       62
       62
       -
       func (t *TemplateRenderer) Render(w io.Writer, name string, data interface{}, c echo.Context) error {

     

       63
       63
       -
       	if viewContext, isMap := data.(map[string]interface{}); isMap {

     

       76
       76
       +
       func (t *TemplateRenderer) Render(w io.Writer, name string, data any, c echo.Context) error {

     

       77
       77
       +
       	if viewContext, isMap := data.(map[string]any); isMap {

     

       64
       78
        
       		viewContext["reverse"] = c.Echo().Reverse

     

       65
       79
        
       	}

     

       66
       80
        
       

     
···

       68
       82
        
       }

     

       69
       83
        
       

     

       70
       84
        
       func run(cmd *cli.Context) error {

     

       71
       71
       -
       	s, err := NewServer()

     

       85
       85
       +
       	s, err := NewServer(ServerArgs{

     

       86
       86
       +
       		Addr:           cmd.String("addr"),

     

       87
       87
       +
       		UrlRoot:        cmd.String("url-root"),

     

       88
       88
       +
       		StaticFilePath: cmd.String("static-file-path"),

     

       89
       89
       +
       		SessionSecret:  cmd.String("session-secret"),

     

       90
       90
       +
       	})

     

       72
       91
        
       	if err != nil {

     

       73
       92
        
       		panic(err)

     

       74
       93
        
       	}

     
···

       78
       97
        
       	return nil

     

       79
       98
        
       }

     

       80
       99
        
       

     

       81
       81
       -
       func NewServer() (*TestServer, error) {

     

       100
       100
       +
       type ServerArgs struct {

     

       101
       101
       +
       	Addr           string

     

       102
       102
       +
       	UrlRoot        string

     

       103
       103
       +
       	StaticFilePath string

     

       104
       104
       +
       	SessionSecret  string

     

       105
       105
       +
       }

     

       106
       106
       +
       

     

       107
       107
       +
       func NewServer(args ServerArgs) (*TestServer, error) {

     

       82
       108
        
       	e := echo.New()

     

       83
       109
        
       

     

       84
       110
        
       	e.Use(slogecho.New(slog.Default()))

     

       85
       85
       -
       	e.Use(session.Middleware(sessions.NewCookieStore([]byte(sessionSecret))))

     

       86
       86
       -
       

     

       87
       87
       -
       	renderer := &TemplateRenderer{

     

       88
       88
       -
       		templates: template.Must(template.ParseGlob(getFilePath("*.html"))),

     

       89
       89
       -
       	}

     

       90
       90
       -
       	e.Renderer = renderer

     

       111
       111
       +
       	e.Use(session.Middleware(sessions.NewCookieStore([]byte(args.SessionSecret))))

     

       91
       112
        
       

     

       92
       113
        
       	fmt.Println("atproto goauth demo webserver")

     

       93
       114
        
       

     
···

       113
       134
        
       

     

       114
       135
        
       	c, err := oauth.NewClient(oauth.ClientArgs{

     

       115
       136
        
       		ClientJwk:   k,

     

       116
       116
       -
       		ClientId:    serverMetadataUrl,

     

       117
       117
       -
       		RedirectUri: serverCallbackUrl,

     

       137
       137
       +
       		ClientId:    args.UrlRoot + serverMetadataPath,

     

       138
       138
       +
       		RedirectUri: args.UrlRoot + serverCallbackPath,

     

       118
       139
        
       	})

     

       119
       140
        
       	if err != nil {

     

       120
       141
        
       		return nil, err

     

       121
       142
        
       	}

     

       122
       143
        
       

     

       123
       144
        
       	httpd := &http.Server{

     

       124
       124
       -
       		Addr:    serverAddr,

     

       145
       145
       +
       		Addr:    args.Addr,

     

       125
       146
        
       		Handler: e,

     

       126
       147
        
       	}

     

       127
       148
        
       

     
···

       140
       161
        
       		},

     

       141
       162
        
       	}

     

       142
       163
        
       

     

       143
       143
       -
       	return &TestServer{

     

       164
       164
       +
       	s := &TestServer{

     

       144
       165
        
       		httpd:        httpd,

     

       145
       166
        
       		e:            e,

     

       146
       167
        
       		db:           db,

     

       147
       168
        
       		oauthClient:  c,

     

       148
       169
        
       		xrpcCli:      xrpcCli,

     

       149
       170
        
       		jwksResponse: oauth_helpers.CreateJwksResponseObject(pubKey),

     

       150
       150
       -
       	}, nil

     

       171
       171
       +
       		args:         args,

     

       172
       172
       +
       	}

     

       173
       173
       +
       

     

       174
       174
       +
       	renderer := &TemplateRenderer{

     

       175
       175
       +
       		templates: template.Must(template.ParseGlob(s.getFilePath("*.html"))),

     

       176
       176
       +
       	}

     

       177
       177
       +
       	e.Renderer = renderer

     

       178
       178
       +
       

     

       179
       179
       +
       	return s, nil

     

       151
       180
        
       }

     

       152
       181
        
       

     

       153
       182
        
       func (s *TestServer) run() error {

     

       154
       183
        
       	s.e.GET("/", s.handleHome)

     

       155
       155
       -
       	s.e.File("/login", getFilePath("login.html"))

     

       184
       184
       +
       	s.e.File("/login", s.getFilePath("login.html"))

     

       156
       185
        
       	s.e.POST("/login", s.handleLoginSubmit)

     

       157
       186
        
       	s.e.GET("/logout", s.handleLogout)

     

       158
       187
        
       	s.e.GET("/profile", s.handleProfile)

     
···

       161
       190
        
       	s.e.GET("/oauth/client-metadata.json", s.handleClientMetadata)

     

       162
       191
        
       	s.e.GET("/oauth/jwks.json", s.handleJwks)

     

       163
       192
        
       

     

       193
       193
       +
       	slog.Default().Info("starting http server", "addr", s.args.Addr)

     

       194
       194
       +
       

     

       164
       195
        
       	if err := s.httpd.ListenAndServe(); err != nil {

     

       165
       196
        
       		return err

     

       166
       197
        
       	}

     
···

       181
       212
        
       

     

       182
       213
        
       func (s *TestServer) handleClientMetadata(e echo.Context) error {

     

       183
       214
        
       	metadata := map[string]any{

     

       184
       184
       -
       		"client_id":                       serverMetadataUrl,

     

       215
       215
       +
       		"client_id":                       s.args.UrlRoot + serverMetadataPath,

     

       185
       216
        
       		"client_name":                     "Atproto GoAuth Demo Webserver",

     

       186
       186
       -
       		"client_uri":                      serverUrlRoot,

     

       187
       187
       -
       		"logo_uri":                        fmt.Sprintf("%s/logo.png", serverUrlRoot),

     

       188
       188
       -
       		"tos_uri":                         fmt.Sprintf("%s/tos", serverUrlRoot),

     

       189
       189
       -
       		"policy_url":                      fmt.Sprintf("%s/policy", serverUrlRoot),

     

       190
       190
       -
       		"redirect_uris":                   []string{serverCallbackUrl},

     

       217
       217
       +
       		"client_uri":                      s.args.UrlRoot,

     

       218
       218
       +
       		"logo_uri":                        fmt.Sprintf("%s/logo.png", s.args.UrlRoot),

     

       219
       219
       +
       		"tos_uri":                         fmt.Sprintf("%s/tos", s.args.UrlRoot),

     

       220
       220
       +
       		"policy_url":                      fmt.Sprintf("%s/policy", s.args.UrlRoot),

     

       221
       221
       +
       		"redirect_uris":                   []string{s.args.UrlRoot + serverCallbackPath},

     

       191
       222
        
       		"grant_types":                     []string{"authorization_code", "refresh_token"},

     

       192
       223
        
       		"response_types":                  []string{"code"},

     

       193
       224
        
       		"application_type":                "web",

     

       194
       225
        
       		"dpop_bound_access_tokens":        true,

     

       195
       195
       -
       		"jwks_uri":                        fmt.Sprintf("%s/oauth/jwks.json", serverUrlRoot),

     

       226
       226
       +
       		"jwks_uri":                        fmt.Sprintf("%s/oauth/jwks.json", s.args.UrlRoot),

     

       196
       227
        
       		"scope":                           "atproto transition:generic",

     

       197
       228
        
       		"token_endpoint_auth_method":      "private_key_jwt",

     

       198
       229
        
       		"token_endpoint_auth_signing_alg": "ES256",

     
···

       205
       236
        
       	return e.JSON(200, s.jwksResponse)

     

       206
       237
        
       }

     

       207
       238
        
       

     

       208
       208
       -
       func getFilePath(file string) string {

     

       209
       209
       -
       	return fmt.Sprintf("%s/%s", staticFilePath, file)

     

       239
       239
       +
       func (s *TestServer) getFilePath(file string) string {

     

       240
       240
       +
       	return fmt.Sprintf("%s/%s", s.args.StaticFilePath, file)

     

       210
       241
        
       }