Merge pull request #231 from H1rono/handler

Handler

Author: H1rono

Cargo.lock

@@ -56,6 +56,27 @@ version = "0.1"
 features = []
 optional = true
 
+[dependencies.tower]
+version = "0.4"
+default-features = false
+features = ["util"]
+optional = true
+
+[dependencies.tower-http]
+version = "0.6"
+default-features = false
+optional = true
+
+[dependencies.futures]
+version = "0.3"
+features = []
+optional = true
+
+[dependencies.pin-project-lite]
+version = "0.2"
+features = []
+optional = true
+
 [dev-dependencies]
 http = "1"
 futures = { version = "0.3", features = ["executor"] }
@@ -65,3 +86,4 @@ uuid = ["dep:uuid"]
 time = ["dep:time"]
 chrono = ["dep:chrono"]
 http = ["dep:http", "dep:http-body", "dep:http-body-util"]
+tower = ["http", "dep:tower", "dep:tower-http", "dep:futures", "dep:pin-project-lite"]

Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "handler-with-axum"
+version = "0.1.0"
+edition = "2021"
+publish = false
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+http = "1"
+tower = "0.4"
+axum = "0.7"
+tokio = { version = "1", features = ["full"] }
+traq-bot-http.path = "../.."
+traq-bot-http.features = ["tower"]

examples/handler-with-axum/Cargo.toml

@@ -0,0 +1,34 @@
+use std::{env, net::SocketAddr};
+
+use axum::{routing::post_service, Router};
+use http::StatusCode;
+use tokio::net::TcpListener;
+use tower::service_fn;
+
+use traq_bot_http::{payloads, RequestParser};
+
+#[tokio::main]
+async fn main() {
+    let verification_token = env::var("VERIFICATION_TOKEN").unwrap();
+    let parser = RequestParser::new(&verification_token);
+    let handler = parser
+        .into_handler()
+        .on_message_created(service_fn(on_message_created));
+    let app = Router::new().route(
+        "/",
+        post_service(handler).handle_error(|_| async { StatusCode::INTERNAL_SERVER_ERROR }),
+    );
+    let addr = SocketAddr::from(([127, 0, 0, 1], 8080));
+    let server = TcpListener::bind(addr).await.unwrap();
+    axum::serve(server, app).await.unwrap();
+}
+
+async fn on_message_created(
+    payload: payloads::MessageCreatedPayload,
+) -> Result<(), std::convert::Infallible> {
+    print!(
+        "{}さんがメッセージを投稿しました。\n内容: {}\n",
+        payload.message.user.display_name, payload.message.text
+    );
+    Ok(())
+}

examples/handler-with-axum/src/main.rs

@@ -57,6 +57,8 @@ pub enum ErrorKind {
     ReadBodyFailed,
     /// リクエストボディの値をパースできなかった
     ParseBodyFailed,
+    /// イベントハンドラ実行中のエラー
+    Handler,
 }
 
 /// type alias
@@ -100,6 +102,8 @@ impl Error {
     error_with_source! {pub(crate) BotEventMismatch}
     error_with_source! {pub(crate) ReadBodyFailed}
     error_with_source! {pub(crate) ParseBodyFailed}
+    // cfg(not(feature = "tower")) でdead_codeになる
+    error_with_source! {#[allow(dead_code)] pub(crate) Handler}
 }
 
 impl From<ErrorKind> for Error {
@@ -135,6 +139,7 @@ impl ErrorKind {
             Self::BotEventMismatch => "X-TRAQ-BOT-EVENT value is wrong",
             Self::ReadBodyFailed => "Failed to read request body",
             Self::ParseBodyFailed => "Failed to parse request body",
+            Self::Handler => "Event handler raised an error",
         }
     }
 }