mirror/mCaptcha
1
0
Fork 1
mirror of https://github.com/mCaptcha/mCaptcha.git synced 2026-02-13 19:15:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: get rid of direct DB init and use db_*

Browse Source
This commit is contained in:
realaravinth
2022-05-27 18:25:27 +05:30
parent cd72ae6ffe
commit d7fd23f565
25 changed files with 55 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/pages/panel/sitekey/edit.rs
View File

@@ -61,8 +61,8 @@ pub async fn advance(
let username = id.identity().unwrap();
let key = path.into_inner();
let config = data.dblib.get_captcha_config(&username, &key).await?;
let levels = data.dblib.get_captcha_levels(Some(&username), &key).await?;
let config = data.db.get_captcha_config(&username, &key).await?;
let levels = data.db.get_captcha_levels(Some(&username), &key).await?;
let body = AdvanceEditPage::new(config, levels, key)
.render_once()
@@ -103,9 +103,9 @@ pub async fn easy(
let username = id.identity().unwrap();
let key = path.into_inner();
match data.dblib.get_traffic_pattern(&username, &key).await {
match data.db.get_traffic_pattern(&username, &key).await {
Ok(c) => {
let config = data.dblib.get_captcha_config(&username, &key).await?;
let config = data.db.get_captcha_config(&username, &key).await?;
let pattern = TrafficPatternRequest {
peak_sustainable_traffic: c.peak_sustainable_traffic as u32,
avg_traffic: c.avg_traffic as u32,

2
src/pages/panel/sitekey/list.rs
View File

@@ -45,7 +45,7 @@ impl IndexPage {
)]
pub async fn list_sitekeys(data: AppData, id: Identity) -> PageResult<impl Responder> {
let username = id.identity().unwrap();
let res = data.dblib.get_all_user_captchas(&username).await?;
let res = data.db.get_all_user_captchas(&username).await?;
let body = IndexPage::new(res).render_once().unwrap();
Ok(HttpResponse::Ok()
.content_type("text/html; charset=utf-8")

4
src/pages/panel/sitekey/view.rs
View File

@@ -67,8 +67,8 @@ pub async fn view_sitekey(
) -> PageResult<impl Responder> {
let username = id.identity().unwrap();
let key = path.into_inner();
let config = data.dblib.get_captcha_config(&username, &key).await?;
let levels = data.dblib.get_captcha_levels(Some(&username), &key).await?;
let config = data.db.get_captcha_config(&username, &key).await?;
let levels = data.db.get_captcha_levels(Some(&username), &key).await?;
let stats = data.stats.fetch(&data, &username, &key).await?;
let body = IndexPage::new(stats, config, levels, key)