commit d42f70cf91464c75f9fbcd4501e27ddf7b02ac5d · nekomimi.pet/microcosm-rs

+32 -3
reflector/src/main.rs
···

       1
       1
        
       use clap::Parser;

     

       2
       2
        
       use poem::{

     

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

     

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

     

       4
       4
       +
           http::StatusCode,

     

       4
       5
        
           listener::TcpListener,

     

       5
       6
        
           middleware::{AddData, Tracing},

     

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

     

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

     

       7
       8
        
       };

     

       8
       8
       -
       use serde::Serialize;

     

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

     

       9
       10
        
       

     

       10
       11
        
       #[handler]

     

       11
       12
        
       fn hello() -> String {

     
···

       33
       34
        
           })

     

       34
       35
        
       }

     

       35
       36
        
       

     

       37
       37
       +
       #[derive(Deserialize)]

     

       38
       38
       +
       struct AskQuery {

     

       39
       39
       +
           domain: String,

     

       40
       40
       +
       }

     

       41
       41
       +
       #[handler]

     

       42
       42
       +
       fn ask_caddy(

     

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

     

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

     

       45
       45
       +
       ) -> Response {

     

       46
       46
       +
           if let Some(parent) = parent {

     

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

     

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

     

       49
       49
       +
                       // no sub-sub-domains allowed

     

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

     

       51
       51
       +
                   }

     

       52
       52
       +
               }

     

       53
       53
       +
           };

     

       54
       54
       +
           Response::builder()

     

       55
       55
       +
               .status(StatusCode::FORBIDDEN)

     

       56
       56
       +
               .body("nope")

     

       57
       57
       +
       }

     

       58
       58
       +
       

     

       36
       59
        
       /// Slingshot record edge cache

     

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

     

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

     
···

       50
       73
        
           /// The HTTPS endpoint for the service

     

       51
       74
        
           #[arg(long)]

     

       52
       75
        
           service_endpoint: String,

     

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

     

       77
       77
       +
           #[arg(long)]

     

       78
       78
       +
           domain: Option<String>,

     

       53
       79
        
       }

     

       54
       80
        
       

     

       55
       81
        
       impl From<Args> for DidService {

     
···

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

     

       69
       95
        
       

     

       70
       96
        
           let args = Args::parse();

     

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

     

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

     

       72
       99
        
       

     

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

     
···

       75
       102
        
                   Route::new()

     

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

     

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

     

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

     

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

     

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

     

       79
       108
        
                       .with(Tracing),

     

       80
       109
        
               )

     

       81
       110
        
               .await