AP-5/src/lib.rs

96 lines
4.1 KiB
Rust

use serde;
use worker::*;
#[event(fetch)]
pub async fn main(req: Request, env: Env, _ctx: worker::Context) -> Result<Response> {
// Create an instance of the Router, which can use parameters (/user/:name) or wildcard values
// (/file/*pathname). Alternatively, use `Router::with_data(D)` and pass in arbitrary data for
// routes to access and share using the `ctx.data()` method.
let router = Router::new();
#[derive(serde::Serialize, serde::Deserialize)]
struct UserInfo {
ext: String,
passwd: String,
comment: String,
created_at: String,
}
router
.get_async("/asterisk/:hostname", |_req, ctx| async move {
if let Some(hostname) = ctx.param("hostname") {
let kv = ctx.kv("GONK_NODES")?;
return match kv.get("base_asterisk").cache_ttl(300).text().await? {
Some(a) => Response::ok(&a),
None => Response::error(
format!("Could not find asterisk.conf for {}", hostname),
404,
),
};
}
Response::error("Bad Request", 400)
})
.get_async("/pjsip/:hostname", |_req, ctx| async move {
if let Some(hostname) = ctx.param("hostname") {
let kv = ctx.kv("GONK_NODES")?;
return match kv.get("base_pjsip").cache_ttl(300).text().await? {
Some(base) => {
let mut conf = base + "";
let remote_ip =
match _req.headers().get("CF-Connecting-IP").unwrap_or_default() {
Some(ip) => ip,
None => return Response::error("Could not find remote IP", 500),
};
conf = conf.replace("##EXTERNAL_IP##", &remote_ip) + "\nxx";
match kv.get("users").cache_ttl(300).text().await? {
Some(users) => {
let users = users.split(",");
let base_user =
match kv.get("base_user").cache_ttl(300).text().await? {
Some(base) => base,
None => {
return Response::error("Could not find user_base", 500)
}
};
for u in users {
let mut user = base_user.clone();
let info: UserInfo =
match kv.get(u).cache_ttl(300).json().await? {
Some(info) => info,
None => {
return Response::error(
format!("Could not find user {}", u),
500,
)
}
};
user = user.replace("##CALL##", u);
user = user.replace("##EXT##", &info.ext);
user = user.replace("##PASSWD##", &info.passwd);
conf = conf + &user;
}
}
None => {
return Response::error(format!("Could not find gonk users"), 404);
}
}
Response::ok(&conf)
}
None => {
Response::error(format!("Could not find pjsip.conf for {}", hostname), 404)
}
};
}
Response::error("Bad Request", 400)
})
.run(req, env)
.await
}