lemmy/crates/api/src/local_user/get_captcha.rs

use crate::{captcha_as_wav_base64, Perform};
use actix_web::web::Data;
use captcha::{gen, Difficulty};
use lemmy_api_common::{
  context::LemmyContext,
  person::{CaptchaResponse, GetCaptcha, GetCaptchaResponse},
};
use lemmy_db_schema::source::{
  captcha_answer::{CaptchaAnswer, CaptchaAnswerForm},
  local_site::LocalSite,
};
use lemmy_utils::error::LemmyError;

#[async_trait::async_trait(?Send)]
impl Perform for GetCaptcha {
  type Response = GetCaptchaResponse;

  #[tracing::instrument(skip(context))]
  async fn perform(&self, context: &Data<LemmyContext>) -> Result<Self::Response, LemmyError> {
    let mut conn = context.conn().await?;

    let local_site = LocalSite::read(&mut conn).await?;

    if !local_site.captcha_enabled {
      return Ok(GetCaptchaResponse { ok: None });
    }

    let captcha = gen(match local_site.captcha_difficulty.as_str() {
      "easy" => Difficulty::Easy,
      "hard" => Difficulty::Hard,
      _ => Difficulty::Medium,
    });

    let answer = captcha.chars_as_string();

    let png = captcha.as_base64().expect("failed to generate captcha");

    let wav = captcha_as_wav_base64(&captcha);

    let captcha_form: CaptchaAnswerForm = CaptchaAnswerForm { answer };
    // Stores the captcha item in the db
    let captcha = CaptchaAnswer::insert(context.pool(), &captcha_form).await?;

    Ok(GetCaptchaResponse {
      ok: Some(CaptchaResponse {
        png,
        wav,
        uuid: captcha.uuid.to_string(),
      }),
    })
  }
}