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

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

#[tracing::instrument(skip(context))]
pub async fn get_captcha(
  context: Data<LemmyContext>,
) -> Result<Json<GetCaptchaResponse>, LemmyError> {
  let local_site = LocalSite::read(&mut context.pool()).await?;

  if !local_site.captcha_enabled {
    return Ok(Json(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(&mut context.pool(), &captcha_form).await?;

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