commit d42f70cf91464c75f9fbcd4501e27ddf7b02ac5d · nekomimi.pet/microcosm-rs

+32 -3

reflector/src/main.rs

···

       1
        
       use clap::Parser;

     

       2
        
       use poem::{

     

       3
       -
           EndpointExt, Route, Server, get, handler,

     

       0
        
       
     

       4
        
           listener::TcpListener,

     

       5
        
           middleware::{AddData, Tracing},

     

       6
       -
           web::{Data, Json, TypedHeader, headers::Host},

     

       7
        
       };

     

       8
       -
       use serde::Serialize;

     

       9
        
       

     

       10
        
       #[handler]

     

       11
        
       fn hello() -> String {

     
···

       33
        
           })

     

       34
        
       }

     

       35
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       36
        
       /// Slingshot record edge cache

     

       37
        
       #[derive(Parser, Debug, Clone)]

     

       38
        
       #[command(version, about, long_about = None)]

     
···

       50
        
           /// The HTTPS endpoint for the service

     

       51
        
           #[arg(long)]

     

       52
        
           service_endpoint: String,

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       53
        
       }

     

       54
        
       

     

       55
        
       impl From<Args> for DidService {

     
···

       68
        
           log::info!("ɹoʇɔǝʅⅎǝɹ");

     

       69
        
       

     

       70
        
           let args = Args::parse();

     

       0
        
       
     

       71
        
           let service: DidService = args.into();

     

       72
        
       

     

       73
        
           Server::new(TcpListener::bind("0.0.0.0:3001"))

     
···

       75
        
                   Route::new()

     

       76
        
                       .at("/", get(hello))

     

       77
        
                       .at("/.well-known/did.json", get(did_doc))

     

       0
        
       
     

       78
        
                       .with(AddData::new(service))

     

       0
        
       
     

       79
        
                       .with(Tracing),

     

       80
        
               )

     

       81
        
               .await

···

       1
        
       use clap::Parser;

     

       2
        
       use poem::{

     

       3
       +
           EndpointExt, Response, Route, Server, get, handler,

     

       4
       +
           http::StatusCode,

     

       5
        
           listener::TcpListener,

     

       6
        
           middleware::{AddData, Tracing},

     

       7
       +
           web::{Data, Json, Query, TypedHeader, headers::Host},

     

       8
        
       };

     

       9
       +
       use serde::{Deserialize, Serialize};

     

       10
        
       

     

       11
        
       #[handler]

     

       12
        
       fn hello() -> String {

     
···

       34
        
           })

     

       35
        
       }

     

       36
        
       

     

       37
       +
       #[derive(Deserialize)]

     

       38
       +
       struct AskQuery {

     

       39
       +
           domain: String,

     

       40
       +
       }

     

       41
       +
       #[handler]

     

       42
       +
       fn ask_caddy(

     

       43
       +
           Data(parent): Data<&Option<String>>,

     

       44
       +
           Query(AskQuery { domain }): Query<AskQuery>,

     

       45
       +
       ) -> Response {

     

       46
       +
           if let Some(parent) = parent {

     

       47
       +
               if let Some(prefix) = domain.strip_suffix(&format!(".{parent}")) {

     

       48
       +
                   if !prefix.contains('.') {

     

       49
       +
                       // no sub-sub-domains allowed

     

       50
       +
                       return Response::builder().body("ok");

     

       51
       +
                   }

     

       52
       +
               }

     

       53
       +
           };

     

       54
       +
           Response::builder()

     

       55
       +
               .status(StatusCode::FORBIDDEN)

     

       56
       +
               .body("nope")

     

       57
       +
       }

     

       58
       +
       

     

       59
        
       /// Slingshot record edge cache

     

       60
        
       #[derive(Parser, Debug, Clone)]

     

       61
        
       #[command(version, about, long_about = None)]

     
···

       73
        
           /// The HTTPS endpoint for the service

     

       74
        
           #[arg(long)]

     

       75
        
           service_endpoint: String,

     

       76
       +
           /// The parent domain; requests should come from subdomains of this

     

       77
       +
           #[arg(long)]

     

       78
       +
           domain: Option<String>,

     

       79
        
       }

     

       80
        
       

     

       81
        
       impl From<Args> for DidService {

     
···

       94
        
           log::info!("ɹoʇɔǝʅⅎǝɹ");

     

       95
        
       

     

       96
        
           let args = Args::parse();

     

       97
       +
           let domain = args.domain.clone();

     

       98
        
           let service: DidService = args.into();

     

       99
        
       

     

       100
        
           Server::new(TcpListener::bind("0.0.0.0:3001"))

     
···

       102
        
                   Route::new()

     

       103
        
                       .at("/", get(hello))

     

       104
        
                       .at("/.well-known/did.json", get(did_doc))

     

       105
       +
                       .at("/ask", get(ask_caddy))

     

       106
        
                       .with(AddData::new(service))

     

       107
       +
                       .with(AddData::new(domain))

     

       108
        
                       .with(Tracing),

     

       109
        
               )

     

       110
        
               .await