add chat function
#
This commit is contained in:
Bartal Læarsson
@@ -0,0 +1,208 @@
|
||||
use axum::{
|
||||
extract::{Path, State, Form},
|
||||
response::Html,
|
||||
};
|
||||
use chrono::Utc;
|
||||
use std::sync::Arc;
|
||||
use uuid::Uuid;
|
||||
|
||||
use crate::{
|
||||
models::{ChatSession, Message, MessageSender, SendMessageForm, NewSessionForm, SessionStatus},
|
||||
state::AppState,
|
||||
};
|
||||
|
||||
fn render_message(msg: &Message) -> String {
|
||||
let (bubble_class, label) = match msg.sender {
|
||||
MessageSender::Customer => ("msg-customer", "You"),
|
||||
MessageSender::Agent => ("msg-agent", "Support"),
|
||||
MessageSender::System => ("msg-system", ""),
|
||||
};
|
||||
|
||||
if msg.sender == MessageSender::System {
|
||||
format!(
|
||||
"<div class=\"msg-system-wrap\"><span class=\"msg-system-text\">{}</span></div>",
|
||||
html_escape(&msg.content)
|
||||
)
|
||||
} else {
|
||||
let time = msg.timestamp.format("%H:%M").to_string();
|
||||
format!(
|
||||
"<div class=\"msg-wrap {bclass}-wrap\">\
|
||||
<div class=\"{bclass}\">\
|
||||
<span class=\"msg-label\">{label}</span>\
|
||||
<p class=\"msg-text\">{content}</p>\
|
||||
<span class=\"msg-time\">{time}</span>\
|
||||
</div>\
|
||||
</div>",
|
||||
bclass = bubble_class,
|
||||
content = html_escape(&msg.content),
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
fn html_escape(s: &str) -> String {
|
||||
s.replace('&', "&")
|
||||
.replace('<', "<")
|
||||
.replace('>', ">")
|
||||
.replace('"', """)
|
||||
}
|
||||
|
||||
/// GET /messages/:session_id
|
||||
pub async fn get_messages(
|
||||
Path(session_id): Path<String>,
|
||||
State(state): State<Arc<AppState>>,
|
||||
) -> Html<String> {
|
||||
let Some(session) = state.sessions.get(&session_id) else {
|
||||
return Html("<p class='error'>Session not found.</p>".to_string());
|
||||
};
|
||||
let html: String = session.messages.iter().map(render_message).collect();
|
||||
Html(html)
|
||||
}
|
||||
|
||||
/// POST /send/:session_id — customer sends a message
|
||||
pub async fn send_message(
|
||||
Path(session_id): Path<String>,
|
||||
State(state): State<Arc<AppState>>,
|
||||
Form(form): Form<SendMessageForm>,
|
||||
) -> Html<String> {
|
||||
let content = form.content.trim().to_string();
|
||||
if content.is_empty() {
|
||||
return Html(String::new());
|
||||
}
|
||||
let msg = Message {
|
||||
id: Uuid::new_v4().to_string(),
|
||||
session_id: session_id.clone(),
|
||||
sender: MessageSender::Customer,
|
||||
content,
|
||||
timestamp: Utc::now(),
|
||||
};
|
||||
let rendered = render_message(&msg);
|
||||
if let Some(mut session) = state.sessions.get_mut(&session_id) {
|
||||
session.messages.push(msg);
|
||||
if let Some(tx) = state.notifiers.get(&session_id) {
|
||||
let _ = tx.send(session_id.clone());
|
||||
}
|
||||
let _ = state.agent_notifier.send(session_id.clone());
|
||||
}
|
||||
Html(rendered)
|
||||
}
|
||||
|
||||
/// POST /agent/send/:session_id — agent sends a message
|
||||
pub async fn agent_send_message(
|
||||
Path(session_id): Path<String>,
|
||||
State(state): State<Arc<AppState>>,
|
||||
Form(form): Form<SendMessageForm>,
|
||||
) -> Html<String> {
|
||||
let content = form.content.trim().to_string();
|
||||
if content.is_empty() {
|
||||
return Html(String::new());
|
||||
}
|
||||
let msg = Message {
|
||||
id: Uuid::new_v4().to_string(),
|
||||
session_id: session_id.clone(),
|
||||
sender: MessageSender::Agent,
|
||||
content,
|
||||
timestamp: Utc::now(),
|
||||
};
|
||||
let rendered = render_message(&msg);
|
||||
if let Some(mut session) = state.sessions.get_mut(&session_id) {
|
||||
session.messages.push(msg);
|
||||
session.status = SessionStatus::Active;
|
||||
if let Some(tx) = state.notifiers.get(&session_id) {
|
||||
let _ = tx.send(session_id.clone());
|
||||
}
|
||||
}
|
||||
Html(rendered)
|
||||
}
|
||||
|
||||
/// GET /sessions — session list for agent dashboard
|
||||
pub async fn get_sessions(
|
||||
State(state): State<Arc<AppState>>,
|
||||
) -> Html<String> {
|
||||
let mut sessions: Vec<_> = state.sessions
|
||||
.iter()
|
||||
.filter(|s| s.status != SessionStatus::Closed)
|
||||
.map(|s| (s.id.clone(), s.customer_name.clone(), s.status.clone(), s.messages.len()))
|
||||
.collect();
|
||||
|
||||
sessions.sort_by(|a, b| a.0.cmp(&b.0));
|
||||
|
||||
if sessions.is_empty() {
|
||||
return Html("<p class=\"no-sessions\">No active conversations yet.</p>".to_string());
|
||||
}
|
||||
|
||||
let mut html = String::new();
|
||||
for (id, name, status, count) in &sessions {
|
||||
let status_class = match status {
|
||||
SessionStatus::Waiting => "status-waiting",
|
||||
SessionStatus::Active => "status-active",
|
||||
SessionStatus::Closed => "status-closed",
|
||||
};
|
||||
let status_label = match status {
|
||||
SessionStatus::Waiting => "Waiting",
|
||||
SessionStatus::Active => "Active",
|
||||
SessionStatus::Closed => "Closed",
|
||||
};
|
||||
let short_id = &id[..8];
|
||||
let escaped_name = html_escape(name);
|
||||
html.push_str(&format!(
|
||||
"<button class=\"session-item\" \
|
||||
hx-get=\"/messages/{id}\" \
|
||||
hx-target=\"#agent-messages\" \
|
||||
hx-swap=\"innerHTML\" \
|
||||
onclick=\"document.querySelectorAll('.session-item').forEach(function(b){{b.classList.remove('selected')}}); this.classList.add('selected'); setActiveSession('{id}', '{escaped_name}')\">\
|
||||
<span class=\"session-name\">{escaped_name}</span>\
|
||||
<span class=\"session-meta\">\
|
||||
<span class=\"session-badge {status_class}\">{status_label}</span>\
|
||||
<span class=\"session-count\">{count} msgs</span>\
|
||||
</span>\
|
||||
<span class=\"session-id\">#{short_id}</span>\
|
||||
</button>"
|
||||
));
|
||||
}
|
||||
Html(html)
|
||||
}
|
||||
|
||||
/// POST /sessions/new — start a new customer session
|
||||
pub async fn new_session(
|
||||
State(state): State<Arc<AppState>>,
|
||||
Form(form): Form<NewSessionForm>,
|
||||
) -> Html<String> {
|
||||
let name = form.name.trim().to_string();
|
||||
let name = if name.is_empty() { "Anonymous".to_string() } else { name };
|
||||
let session_id = Uuid::new_v4().to_string();
|
||||
let short = session_id[..8].to_string();
|
||||
|
||||
let session = ChatSession::new(session_id.clone(), name.clone());
|
||||
state.sessions.insert(session_id.clone(), session);
|
||||
state.get_or_create_notifier(&session_id);
|
||||
let _ = state.agent_notifier.send(session_id.clone());
|
||||
|
||||
// The response swaps out #start-form and fills #chat-container via OOB
|
||||
let html = format!(
|
||||
"<div id=\"chat-container\" style=\"display:flex;flex-direction:column\" hx-swap-oob=\"true\">\
|
||||
<div id=\"chat-header\">\
|
||||
<span class=\"chat-status-dot active\"></span>\
|
||||
<span>Support Chat</span>\
|
||||
<span class=\"chat-session-id\">Session #{short}</span>\
|
||||
</div>\
|
||||
<div id=\"messages\" \
|
||||
hx-get=\"/messages/{session_id}\" \
|
||||
hx-trigger=\"load, every 2s\" \
|
||||
hx-swap=\"innerHTML\">\
|
||||
</div>\
|
||||
<div id=\"typing-indicator\" class=\"htmx-indicator\">\
|
||||
<span class=\"typing-dot\"></span><span class=\"typing-dot\"></span><span class=\"typing-dot\"></span>\
|
||||
</div>\
|
||||
<form id=\"chat-form\" \
|
||||
hx-post=\"/send/{session_id}\" \
|
||||
hx-target=\"#messages\" \
|
||||
hx-swap=\"beforeend\" \
|
||||
hx-on=\"htmx:afterRequest: this.reset(); scrollMessages()\">\
|
||||
<input type=\"text\" name=\"content\" placeholder=\"Type your message...\" autocomplete=\"off\" required />\
|
||||
<button type=\"submit\">Send</button>\
|
||||
</form>\
|
||||
</div>\
|
||||
<script>document.getElementById('start-form').style.display='none'; window.__sessionId='{session_id}';</script>"
|
||||
);
|
||||
Html(html)
|
||||
}
|
||||
@@ -0,0 +1,3 @@
|
||||
pub mod pages;
|
||||
pub mod chat;
|
||||
pub mod ws;
|
||||
@@ -0,0 +1,9 @@
|
||||
use axum::response::Html;
|
||||
|
||||
pub async fn customer_page() -> Html<&'static str> {
|
||||
Html(include_str!("../../templates/customer.html"))
|
||||
}
|
||||
|
||||
pub async fn agent_page() -> Html<&'static str> {
|
||||
Html(include_str!("../../templates/agent.html"))
|
||||
}
|
||||
@@ -0,0 +1,74 @@
|
||||
use axum::{
|
||||
extract::{
|
||||
ws::{Message, WebSocket, WebSocketUpgrade},
|
||||
Path, State,
|
||||
},
|
||||
response::IntoResponse,
|
||||
};
|
||||
use std::sync::Arc;
|
||||
|
||||
use crate::state::AppState;
|
||||
|
||||
/// WebSocket for customer — pushes notifications when agent replies
|
||||
pub async fn customer_ws(
|
||||
ws: WebSocketUpgrade,
|
||||
Path(session_id): Path<String>,
|
||||
State(state): State<Arc<AppState>>,
|
||||
) -> impl IntoResponse {
|
||||
ws.on_upgrade(move |socket| handle_customer_socket(socket, session_id, state))
|
||||
}
|
||||
|
||||
async fn handle_customer_socket(mut socket: WebSocket, session_id: String, state: Arc<AppState>) {
|
||||
let tx = state.get_or_create_notifier(&session_id);
|
||||
let mut rx = tx.subscribe();
|
||||
|
||||
loop {
|
||||
tokio::select! {
|
||||
// Forward broadcast events to the customer browser
|
||||
Ok(sid) = rx.recv() => {
|
||||
if sid == session_id {
|
||||
// Signal HTMX to re-poll messages
|
||||
if socket.send(Message::Text("refresh".into())).await.is_err() {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
// Handle ping/close from client
|
||||
msg = socket.recv() => {
|
||||
match msg {
|
||||
Some(Ok(Message::Close(_))) | None => break,
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// WebSocket for agent dashboard — pushes notifications when any session updates
|
||||
pub async fn agent_ws(
|
||||
ws: WebSocketUpgrade,
|
||||
Path(_agent_id): Path<String>,
|
||||
State(state): State<Arc<AppState>>,
|
||||
) -> impl IntoResponse {
|
||||
ws.on_upgrade(move |socket| handle_agent_socket(socket, state))
|
||||
}
|
||||
|
||||
async fn handle_agent_socket(mut socket: WebSocket, state: Arc<AppState>) {
|
||||
let mut rx = state.agent_notifier.subscribe();
|
||||
|
||||
loop {
|
||||
tokio::select! {
|
||||
Ok(session_id) = rx.recv() => {
|
||||
if socket.send(Message::Text(session_id.into())).await.is_err() {
|
||||
break;
|
||||
}
|
||||
}
|
||||
msg = socket.recv() => {
|
||||
match msg {
|
||||
Some(Ok(Message::Close(_))) | None => break,
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
+46
@@ -0,0 +1,46 @@
|
||||
mod state;
|
||||
mod handlers;
|
||||
mod models;
|
||||
|
||||
use axum::{
|
||||
routing::{get, post},
|
||||
Router,
|
||||
};
|
||||
use std::sync::Arc;
|
||||
use tower_http::services::ServeDir;
|
||||
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
||||
|
||||
use state::AppState;
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
tracing_subscriber::registry()
|
||||
.with(tracing_subscriber::EnvFilter::new(
|
||||
std::env::var("RUST_LOG").unwrap_or_else(|_| "chat_service=debug,info".into()),
|
||||
))
|
||||
.with(tracing_subscriber::fmt::layer())
|
||||
.init();
|
||||
|
||||
let state = Arc::new(AppState::new());
|
||||
|
||||
let app = Router::new()
|
||||
.route("/", get(handlers::pages::customer_page))
|
||||
.route("/agent", get(handlers::pages::agent_page))
|
||||
.route("/ws/customer/:session_id", get(handlers::ws::customer_ws))
|
||||
.route("/ws/agent/:agent_id", get(handlers::ws::agent_ws))
|
||||
.route("/messages/:session_id", get(handlers::chat::get_messages))
|
||||
.route("/send/:session_id", post(handlers::chat::send_message))
|
||||
.route("/agent/send/:session_id", post(handlers::chat::agent_send_message))
|
||||
.route("/sessions", get(handlers::chat::get_sessions))
|
||||
.route("/sessions/new", post(handlers::chat::new_session))
|
||||
.nest_service("/static", ServeDir::new("static"))
|
||||
.with_state(state);
|
||||
|
||||
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
|
||||
tracing::info!("Chat service running — http://0.0.0.0:3000");
|
||||
axum::Server::from_tcp(listener.into_std().unwrap())
|
||||
.unwrap()
|
||||
.serve(app.into_make_service())
|
||||
.await
|
||||
.unwrap();
|
||||
}
|
||||
@@ -0,0 +1,66 @@
|
||||
use chrono::{DateTime, Utc};
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
|
||||
#[serde(rename_all = "lowercase")]
|
||||
pub enum MessageSender {
|
||||
Customer,
|
||||
Agent,
|
||||
System,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Message {
|
||||
pub id: String,
|
||||
pub session_id: String,
|
||||
pub sender: MessageSender,
|
||||
pub content: String,
|
||||
pub timestamp: DateTime<Utc>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
|
||||
#[serde(rename_all = "lowercase")]
|
||||
pub enum SessionStatus {
|
||||
Waiting,
|
||||
Active,
|
||||
Closed,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct ChatSession {
|
||||
pub id: String,
|
||||
pub customer_name: String,
|
||||
pub status: SessionStatus,
|
||||
pub created_at: DateTime<Utc>,
|
||||
pub messages: Vec<Message>,
|
||||
}
|
||||
|
||||
impl ChatSession {
|
||||
pub fn new(id: String, customer_name: String) -> Self {
|
||||
let now = Utc::now();
|
||||
let welcome = Message {
|
||||
id: uuid::Uuid::new_v4().to_string(),
|
||||
session_id: id.clone(),
|
||||
sender: MessageSender::System,
|
||||
content: "Welcome! A support agent will be with you shortly.".to_string(),
|
||||
timestamp: now,
|
||||
};
|
||||
Self {
|
||||
id,
|
||||
customer_name,
|
||||
status: SessionStatus::Waiting,
|
||||
created_at: now,
|
||||
messages: vec![welcome],
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct SendMessageForm {
|
||||
pub content: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct NewSessionForm {
|
||||
pub name: String,
|
||||
}
|
||||
@@ -0,0 +1,32 @@
|
||||
use dashmap::DashMap;
|
||||
use tokio::sync::broadcast;
|
||||
use crate::models::ChatSession;
|
||||
|
||||
pub struct AppState {
|
||||
pub sessions: DashMap<String, ChatSession>,
|
||||
// broadcast channel: session_id -> sender
|
||||
pub notifiers: DashMap<String, broadcast::Sender<String>>,
|
||||
// agent broadcast for new sessions
|
||||
pub agent_notifier: broadcast::Sender<String>,
|
||||
}
|
||||
|
||||
impl AppState {
|
||||
pub fn new() -> Self {
|
||||
let (agent_tx, _) = broadcast::channel(64);
|
||||
Self {
|
||||
sessions: DashMap::new(),
|
||||
notifiers: DashMap::new(),
|
||||
agent_notifier: agent_tx,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn get_or_create_notifier(&self, session_id: &str) -> broadcast::Sender<String> {
|
||||
if let Some(tx) = self.notifiers.get(session_id) {
|
||||
tx.clone()
|
||||
} else {
|
||||
let (tx, _) = broadcast::channel(64);
|
||||
self.notifiers.insert(session_id.to_string(), tx.clone());
|
||||
tx
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user