check oauth token existence

crates/service/storyteller_web/src/http_server/endpoints/twitch/oauth/check_oauth_status.rs

@@ -0,0 +1,98 @@
+use actix_http::{StatusCode, header};
+use actix_web::{HttpResponse, HttpRequest, web, ResponseError};
+use crate::server_state::ServerState;
+use database_queries::queries::twitch::twitch_oauth::insert::TwitchOauthTokenInsertBuilder;
+use http_server_common::request::get_request_ip::get_request_ip;
+use http_server_common::response::response_error_helpers::to_simple_json_error;
+use log::error;
+use log::info;
+use log::warn;
+use std::fmt;
+use std::sync::Arc;
+use twitch_common::oauth_token_builder::get_oauth_token_builder;
+use twitch_oauth2::tokens::BearerTokenType::UserToken;
+use twitch_oauth2::{CsrfToken, TwitchToken};
+use database_queries::tokens::Tokens;
+use database_queries::queries::twitch::twitch_oauth::find::{TwitchOauthTokenFinder, TwitchOauthTokenRecord};
+
+#[derive(Serialize)]
+pub struct CheckOauthResponse {
+  pub success: bool,
+
+  /// This is false if the user isn't logged in, doesn't have oauth, or oauth is invalid
+  pub oauth_token_found: bool,
+}
+
+#[derive(Debug)]
+pub enum CheckOauthStatusError {
+  ServerError,
+}
+
+impl ResponseError for CheckOauthStatusError {
+  fn status_code(&self) -> StatusCode {
+    match *self {
+      Self::ServerError => StatusCode::INTERNAL_SERVER_ERROR,
+    }
+  }
+
+  fn error_response(&self) -> HttpResponse {
+    let error_reason = match self {
+      Self::ServerError => "server error".to_string(),
+    };
+
+    to_simple_json_error(&error_reason, self.status_code())
+  }
+}
+
+// NB: Not using derive_more::Display since Clion doesn't understand it.
+impl fmt::Display for CheckOauthStatusError {
+  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+    write!(f, "{:?}", self)
+  }
+}
+
+pub async fn check_oauth_status_handler(
+  http_request: HttpRequest,
+  server_state: web::Data<Arc<ServerState>>
+)
+  -> Result<HttpResponse, CheckOauthStatusError>
+{
+  let maybe_user_session = server_state
+      .session_checker
+      .maybe_get_user_session(&http_request, &server_state.mysql_pool)
+      .await
+      .map_err(|e| {
+        warn!("Session checker error: {:?}", e);
+        CheckOauthStatusError::ServerError
+      })?;
+
+  let oauth_token_found = match maybe_user_session {
+    None => false, // Not logged in - not found
+    Some(session) => {
+      let finder = TwitchOauthTokenFinder::new()
+          .allow_expired_tokens(true)
+          .scope_user_token(Some(&session.user_token));
+
+      let record = finder.perform_query(&server_state.mysql_pool).await
+          .map_err(|e| {
+            warn!("lookup error: {:?}", e);
+            CheckOauthStatusError::ServerError
+          })?;
+
+      // Existence is sufficient.
+      record.is_some()
+    },
+  };
+
+  let response = CheckOauthResponse {
+    success: true,
+    oauth_token_found,
+  };
+
+  let body = serde_json::to_string(&response)
+      .map_err(|e| CheckOauthStatusError::ServerError)?;
+
+  return Ok(HttpResponse::Ok()
+      .content_type("application/json")
+      .body(body));
+}

crates/service/storyteller_web/src/http_server/endpoints/twitch/oauth/mod.rs

@@ -1,3 +1,4 @@
+pub mod check_oauth_status;
 pub mod oauth_begin_json;
 pub mod oauth_begin_redirect;
 pub mod oauth_end;

crates/service/storyteller_web/src/routes.rs

@@ -83,6 +83,7 @@ use crate::http_server::endpoints::w2l::get_w2l_template_use_count::get_w2l_temp
 use crate::http_server::endpoints::w2l::get_w2l_upload_template_job_status::get_w2l_upload_template_job_status_handler;
 use crate::http_server::endpoints::w2l::list_w2l_templates::list_w2l_templates_handler;
 use crate::http_server::endpoints::w2l::set_w2l_template_mod_approval::set_w2l_template_mod_approval_handler;
+use crate::http_server::endpoints::twitch::oauth::check_oauth_status::check_oauth_status_handler;
 
 pub fn add_routes<T, B> (app: App<T, B>) -> App<T, B>
   where
@@ -632,6 +633,12 @@ fn add_twitch_routes<T, B> (app: App<T, B>) -> App<T, B>
       .route(web::get().to(oauth_end_enroll_from_redirect))
       .route(web::head().to(|| HttpResponse::Ok()))
     )
+    .service(web::scope("/oauth")
+        .service(web::resource("/check")
+            .route(web::get().to(check_oauth_status_handler))
+            .route(web::head().to(|| HttpResponse::Ok()))
+        )
+    )
     .service(web::scope("/event_rule")
         .service(web::resource("/create")
             .route(web::post().to(create_twitch_event_rule_handler))