commit 998eafedae0ee8ad443209874a63d1a74ccd1353 · nekomimi.pet/microcosm-rs

+52 -69

who-am-i/src/server.rs

···

       247
        
           use atrium_identity::Error as IdError;

     

       248
        
           use atrium_oauth::Error as OAuthError;

     

       249
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       250
        
           match oauth.begin(&params.handle).await {

     

       251
        
               Ok(auth_url) => (jar, Redirect::to(&auth_url)).into_response(),

     

       252
       -
               Err(OAuthError::Identity(IdError::NotFound)) => {

     

       253
       -
                   let info = json!({ "reason": "handle not found" });

     

       254
       -
                   (StatusCode::NOT_FOUND, RenderHtml("auth-fail", engine, info)).into_response()

     

       255
       -
               }

     

       256
       -
               Err(OAuthError::Identity(IdError::AtIdentifier(r))) => {

     

       257
       -
                   let info = json!({ "reason": r });

     

       258
       -
                   (StatusCode::NOT_FOUND, RenderHtml("auth-fail", engine, info)).into_response()

     

       259
       -
               }

     

       260
       -
               Err(OAuthError::Identity(IdError::HttpStatus(StatusCode::NOT_FOUND))) => {

     

       261
       -
                   let info = json!({ "reason": "handle not found" });

     

       262
       -
                   (StatusCode::NOT_FOUND, RenderHtml("auth-fail", engine, info)).into_response()

     

       263
       -
               }

     

       264
        
               Err(e) => {

     

       265
        
                   eprintln!("begin auth failed: {e:?}");

     

       266
       -
                   let info = json!({ "reason": "unknown" });

     

       267
       -
                   (

     

       268
       -
                       StatusCode::INTERNAL_SERVER_ERROR,

     

       269
       -
                       RenderHtml("auth-fail", engine, info),

     

       270
       -
                   )

     

       271
       -
                       .into_response()

     

       272
        
               }

     

       273
        
           }

     

       274
        
       }

     

       275
        
       

     

       276
       -
       impl OAuthCompleteError {

     

       277
       -
           fn to_error_response(&self, engine: AppEngine) -> Response {

     

       278
       -
               let (level, desc) = match self {

     

       279
       -
                   OAuthCompleteError::Denied { description, .. } => {

     

       280
       -
                       ("warn", format!("asdf: {description:?}"))

     

       281
       -
                   }

     

       282
       -
                   OAuthCompleteError::Failed { .. } => (

     

       283
       -
                       "error",

     

       284
       -
                       "Something went wrong while requesting permission, sorry!".to_string(),

     

       285
       -
                   ),

     

       286
       -
                   OAuthCompleteError::CallbackFailed(_) => (

     

       287
       -
                       "error",

     

       288
       -
                       "Something went wrong after permission was granted, sorry!".to_string(),

     

       289
       -
                   ),

     

       290
       -
                   OAuthCompleteError::NoDid => (

     

       291
       -
                       "error",

     

       292
       -
                       "Something went wrong when trying to confirm your identity, sorry!".to_string(),

     

       293
       -
                   ),

     

       294
       -
               };

     

       295
       -
               (

     

       296
       -
                   if level == "warn" {

     

       297
       -
                       StatusCode::FORBIDDEN

     

       298
       -
                   } else {

     

       299
       -
                       StatusCode::INTERNAL_SERVER_ERROR

     

       300
       -
                   },

     

       301
       -
                   RenderHtml(

     

       302
       -
                       "auth-fail",

     

       303
       -
                       engine,

     

       304
       -
                       json!({

     

       305
       -
                           "reason": desc,

     

       306
       -
                       }),

     

       307
       -
                   ),

     

       308
       -
               )

     

       309
       -
                   .into_response()

     

       310
       -
           }

     

       311
       -
       }

     

       312
       -
       

     

       313
        
       async fn complete_oauth(

     

       314
        
           State(AppState {

     

       315
        
               engine,

     
···

       320
        
           }): State<AppState>,

     

       321
        
           Query(params): Query<OAuthCallbackParams>,

     

       322
        
           jar: SignedCookieJar,

     

       323
       -
       ) -> Result<(SignedCookieJar, impl IntoResponse), Response> {

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       324
        
           let did = match oauth.complete(params).await {

     

       325
        
               Ok(did) => did,

     

       326
       -
               Err(e) => return Err(e.to_error_response(engine)),

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       327
        
           };

     

       328
        
       

     

       329
        
           let cookie = Cookie::build((DID_COOKIE_KEY, did.to_string()))

     
···

       342
        
               },

     

       343
        
               shutdown.child_token(),

     

       344
        
           );

     

       345
       -
       

     

       346
       -
           Ok((

     

       347
       -
               jar,

     

       348
       -
               RenderHtml(

     

       349
       -
                   "authorized",

     

       350
       -
                   engine,

     

       351
       -
                   json!({

     

       352
       -
                       "did": did,

     

       353
       -
                       "fetch_key": fetch_key,

     

       354
       -
                   }),

     

       355
       -
               ),

     

       356
       -
           ))

     

       357
        
       }

···

       247
        
           use atrium_identity::Error as IdError;

     

       248
        
           use atrium_oauth::Error as OAuthError;

     

       249
        
       

     

       250
       +
           let err = |code, reason| {

     

       251
       +
               let info = json!({

     

       252
       +
                   "result": "fail",

     

       253
       +
                   "reason": reason,

     

       254
       +
               });

     

       255
       +
               (code, RenderHtml("auth-fail", engine.clone(), info)).into_response()

     

       256
       +
           };

     

       257
       +
       

     

       258
        
           match oauth.begin(&params.handle).await {

     

       259
        
               Ok(auth_url) => (jar, Redirect::to(&auth_url)).into_response(),

     

       260
       +
               Err(OAuthError::Identity(

     

       261
       +
                   IdError::NotFound | IdError::HttpStatus(StatusCode::NOT_FOUND),

     

       262
       +
               )) => err(StatusCode::NOT_FOUND, "handle not found"),

     

       263
       +
               Err(OAuthError::Identity(IdError::AtIdentifier(r))) => err(StatusCode::BAD_REQUEST, &r),

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       264
        
               Err(e) => {

     

       265
        
                   eprintln!("begin auth failed: {e:?}");

     

       266
       +
                   err(StatusCode::INTERNAL_SERVER_ERROR, "unknown")

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       267
        
               }

     

       268
        
           }

     

       269
        
       }

     

       270
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       271
        
       async fn complete_oauth(

     

       272
        
           State(AppState {

     

       273
        
               engine,

     
···

       278
        
           }): State<AppState>,

     

       279
        
           Query(params): Query<OAuthCallbackParams>,

     

       280
        
           jar: SignedCookieJar,

     

       281
       +
       ) -> Response {

     

       282
       +
           let err = |code, result, reason| {

     

       283
       +
               let info = json!({

     

       284
       +
                   "result": result,

     

       285
       +
                   "reason": reason,

     

       286
       +
               });

     

       287
       +
               (code, RenderHtml("auth-fail", engine.clone(), info)).into_response()

     

       288
       +
           };

     

       289
       +
       

     

       290
        
           let did = match oauth.complete(params).await {

     

       291
        
               Ok(did) => did,

     

       292
       +
               Err(e) => {

     

       293
       +
                   return match e {

     

       294
       +
                       OAuthCompleteError::Denied { description, .. } => {

     

       295
       +
                           let desc = description.unwrap_or("permission to share was denied".to_string());

     

       296
       +
                           err(StatusCode::FORBIDDEN, "deny", desc.as_str())

     

       297
       +
                       }

     

       298
       +
                       OAuthCompleteError::Failed { .. } => {

     

       299
       +
                           eprintln!("auth completion failed: {e:?}");

     

       300
       +
                           err(

     

       301
       +
                               StatusCode::INTERNAL_SERVER_ERROR,

     

       302
       +
                               "fail",

     

       303
       +
                               "failed to complete",

     

       304
       +
                           )

     

       305
       +
                       }

     

       306
       +
                       OAuthCompleteError::CallbackFailed(e) => {

     

       307
       +
                           eprintln!("auth callback failed: {e:?}");

     

       308
       +
                           err(

     

       309
       +
                               StatusCode::INTERNAL_SERVER_ERROR,

     

       310
       +
                               "fail",

     

       311
       +
                               "failed to complete callback",

     

       312
       +
                           )

     

       313
       +
                       }

     

       314
       +
                       OAuthCompleteError::NoDid => err(StatusCode::BAD_REQUEST, "fail", "no DID found"),

     

       315
       +
                   };

     

       316
       +
               }

     

       317
        
           };

     

       318
        
       

     

       319
        
           let cookie = Cookie::build((DID_COOKIE_KEY, did.to_string()))

     
···

       332
        
               },

     

       333
        
               shutdown.child_token(),

     

       334
        
           );

     

       335
       +
           let info = json!({

     

       336
       +
               "did": did,

     

       337
       +
               "fetch_key": fetch_key,

     

       338
       +
           });

     

       339
       +
           (jar, RenderHtml("authorized", engine, info)).into_response()

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       340
        
       }

+1 -1

who-am-i/templates/auth-fail.hbs

···

       9
        
       

     

       10
        
       <script>

     

       11
        
       localStorage.setItem("who-am-i", JSON.stringify({

     

       12
       -
         result: "fail",

     

       13
        
         reason: {{{json reason}}},

     

       14
        
       }));

     

       15

···

       9
        
       

     

       10
        
       <script>

     

       11
        
       localStorage.setItem("who-am-i", JSON.stringify({

     

       12
       +
         result: {{{json result}}},

     

       13
        
         reason: {{{json reason}}},

     

       14
        
       }));

     

       15

+5 -1

who-am-i/templates/prompt.hbs

···

       87
        
           fail(`uh oh: ${parsed.reason}`);

     

       88
        
         }

     

       89
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       90
        
         infoEl.classList.add('hidden');

     

       91
        
       

     

       92
        
         const handle = await lookUp(parsed.fetch_key);

     
···

       111
        
       const shareAllow = handle => {

     

       112
        
         top.postMessage(

     

       113
        
           { action: "allow", handle },

     

       114
       -
           {{{json parent_host}}},

     

       115
        
         );

     

       116
        
       }

     

       117

···

       87
        
           fail(`uh oh: ${parsed.reason}`);

     

       88
        
         }

     

       89
        
       

     

       90
       +
         if (parsed.result === "deny") {

     

       91
       +
           fail(parsed.reason);

     

       92
       +
         }

     

       93
       +
       

     

       94
        
         infoEl.classList.add('hidden');

     

       95
        
       

     

       96
        
         const handle = await lookUp(parsed.fetch_key);

     
···

       115
        
       const shareAllow = handle => {

     

       116
        
         top.postMessage(

     

       117
        
           { action: "allow", handle },

     

       118
       +
           {{{json parent_origin}}},

     

       119
        
         );

     

       120
        
       }

     

       121