diff --git a/flake.nix b/flake.nix index 8e8ddf5..2d32066 100644 --- a/flake.nix +++ b/flake.nix @@ -8,7 +8,7 @@ outputs = { self, nixpkgs, rust-overlay, flake-utils, ... }: flake-utils.lib.eachDefaultSystem (system: let - tag = "v0.1.4_4-alpha"; + tag = "v0.1.5-alpha"; overlays = [ (import rust-overlay) ]; pkgs = import nixpkgs { inherit system overlays; diff --git a/src/htmx.rs b/src/htmx.rs index 98c5e4f..751f4e0 100644 --- a/src/htmx.rs +++ b/src/htmx.rs @@ -23,6 +23,7 @@ pub enum Tag { Dialog, Article, Header, + Footer, Div, @@ -51,6 +52,7 @@ pub enum Tag { Anchor, Button, + HeaderGroup, Header1, Header2, Header3, @@ -85,6 +87,7 @@ impl Tag { Self::Dialog => "dialog", Self::Article => "article", Self::Header => "header", + Self::Footer => "footer", Self::Div => "div", @@ -111,6 +114,7 @@ impl Tag { Self::Anchor => "a", Self::Button => "button", + Self::HeaderGroup => "hgroup", Self::Header1 => "h1", Self::Header2 => "h2", Self::Header3 => "h3", diff --git a/src/page.rs b/src/page.rs index f6c38db..0b5f3f4 100644 --- a/src/page.rs +++ b/src/page.rs @@ -1,7 +1,7 @@ use crate::{ auth::{self, User}, - htmx::{Build, HtmxBuilder, SwapMethod, Tag}, - settings::{ApiState, Intro, IntroFriendlyName}, + htmx::{Build, HtmxBuilder, Tag}, + settings::{ApiState, GuildSettings, Intro, IntroFriendlyName}, }; use axum::{ extract::{Path, State}, @@ -9,14 +9,64 @@ use axum::{ }; use tracing::error; -pub(crate) async fn home(user: Option) -> Redirect { - if user.is_some() { - Redirect::to("/guild/588149178912473103") +fn page_header(title: &str) -> HtmxBuilder { + HtmxBuilder::new(Tag::Html).head(|b| { + b.title(title) + .script( + "https://unpkg.com/htmx.org@1.9.3", + Some("sha384-lVb3Rd/Ca0AxaoZg5sACe8FJKF0tnUgR2Kd7ehUOG5GCcROv5uBIZsOqovBAcWua"), + ) + // Not currently using + // .script("https://unpkg.com/hyperscript.org@0.9.9", None) + .style_link("https://cdn.jsdelivr.net/npm/@picocss/pico@1/css/pico.min.css") + }) +} + +pub(crate) async fn home( + State(state): State, + user: Option, +) -> Result, Redirect> { + if let Some(user) = user { + let settings = state.settings.lock().await; + + let guild = settings + .guilds + .iter() + .filter(|(_, guild_settings)| guild_settings.users.contains_key(&user.name)); + + Ok(Html( + page_header("MemeJoin - Home") + .builder(Tag::Div, |b| { + b.attribute("class", "container") + .builder_text(Tag::Header2, "Choose a Guild") + .push_builder(guild_list(&state.origin, guild)) + }) + .build(), + )) } else { - Redirect::to("/login") + Err(Redirect::to(&format!("{}/login", state.origin))) } } +fn guild_list<'a>( + origin: &str, + guilds: impl Iterator, +) -> HtmxBuilder { + HtmxBuilder::new(Tag::Empty).ul(|b| { + let mut b = b; + for (guild_id, guild_settings) in guilds { + b = b.li(|b| { + b.link( + &guild_settings.name, + &format!("{}/guild/{}", origin, guild_id), + ) + }); + } + + b + }) +} + fn intro_list<'a>( intros: impl Iterator, label: &str, @@ -29,7 +79,7 @@ fn intro_list<'a>( .builder(Tag::FieldSet, |b| { let mut b = b .attribute("class", "container") - .attribute("style", "max-height: 20%; overflow-y: scroll"); + .attribute("style", "max-height: 50%; overflow-y: scroll"); for intro in intros { b = b.builder(Tag::Label, |b| { b.builder(Tag::Input, |b| { @@ -54,107 +104,110 @@ pub(crate) async fn guild_dashboard( let Some(guild) = settings.guilds.get(&guild_id) else { error!(%guild_id, "no such guild"); - return Err(Redirect::to("/")); + return Err(Redirect::to(&format!("{}/", state.origin))); }; let Some(guild_user) = guild.users.get(&user.name) else { error!(%guild_id, %user.name, "no user in guild"); - return Err(Redirect::to("/")); + return Err(Redirect::to(&format!("{}/", state.origin))); }; let is_moderator = guild_user.permissions.can(auth::Permission::DeleteSounds); Ok(Html( HtmxBuilder::new(Tag::Html) - .head(|b| { - b.title("MemeJoin - Dashboard") - .script( - "https://unpkg.com/htmx.org@1.9.3", - Some("sha384-lVb3Rd/Ca0AxaoZg5sACe8FJKF0tnUgR2Kd7ehUOG5GCcROv5uBIZsOqovBAcWua"), - ) - .script("https://unpkg.com/hyperscript.org@0.9.9", None) - .style_link("https://cdn.jsdelivr.net/npm/@picocss/pico@1/css/pico.min.css") - }) + .push_builder(page_header("MemeJoin - Dashboard")) .builder(Tag::Nav, |b| { - b.builder(Tag::Header1, |b| b.text("MemeJoin - A bot for user intros")) - .builder_text(Tag::Paragraph, &user.name) + b.builder(Tag::HeaderGroup, |b| { + b.attribute("class", "container") + .builder(Tag::Header1, |b| b.text("MemeJoin - A bot for user intros")) + .builder_text(Tag::Header6, &user.name) + }) }) - .builder(Tag::Main, |b| { + .builder(Tag::Empty, |b| { if is_moderator { - b.builder(Tag::Article, |b| { - b.builder_text(Tag::Header, "Wow, you're a moderator") + b.builder(Tag::Div, |b| { + b.attribute("class", "container") + .builder(Tag::Article, |b| { + b.builder_text(Tag::Header, "Wow, you're a moderator") + .push_builder(moderator_dashboard(&state)) + .builder_text(Tag::Footer, "End of super cool mod section") + }) }) } else { b } - .builder(Tag::Article, |b| { - let mut b = b.builder_text(Tag::Header, "Guild Settings"); + .builder(Tag::Div, |b| { + b.attribute("class", "container") + .builder(Tag::Article, |b| { + let mut b = b.builder_text(Tag::Header, "Guild Intros"); - for (channel_name, channel_settings) in &guild.channels { - if let Some(channel_user) = channel_settings.users.get(&user.name) { - let current_intros = - channel_user.intros.iter().filter_map(|intro_index| { - Some(( - &intro_index.index, - guild.intros.get(&intro_index.index)?, - )) - }); - let available_intros = guild.intros.iter().filter_map(|intro| { - if !channel_user - .intros - .iter() - .any(|intro_index| intro.0 == &intro_index.index) - { - Some((intro.0, intro.1)) - } else { - None + for (channel_name, channel_settings) in &guild.channels { + if let Some(channel_user) = channel_settings.users.get(&user.name) { + let current_intros = + channel_user.intros.iter().filter_map(|intro_index| { + Some(( + &intro_index.index, + guild.intros.get(&intro_index.index)?, + )) + }); + let available_intros = + guild.intros.iter().filter_map(|intro| { + if !channel_user + .intros + .iter() + .any(|intro_index| intro.0 == &intro_index.index) + { + Some((intro.0, intro.1)) + } else { + None + } + }); + b = b.builder(Tag::Article, |b| { + b.builder_text(Tag::Header, channel_name).builder( + Tag::Div, + |b| { + b.builder_text(Tag::Strong, "Your Current Intros") + .push_builder(intro_list( + current_intros, + "Remove Intro", + &format!( + "{}/v2/intros/remove/{}/{}", + state.origin, guild_id, channel_name + ), + )) + .builder_text(Tag::Strong, "Select Intros") + .push_builder(intro_list( + available_intros, + "Add Intro", + &format!( + "{}/v2/intros/add/{}/{}", + state.origin, guild_id, channel_name + ), + )) + }, + ) + }); } - }); - b = b - .builder_text(Tag::Strong, channel_name) - .builder(Tag::Div, |b| { - b.builder_text(Tag::Strong, "Your Current Intros") - .push_builder(intro_list( - current_intros, - "Remove Intro", - &format!( - "/v2/intros/remove/{}/{}", - guild_id, channel_name - ), - )) - .builder_text(Tag::Strong, "Select Intros") - .push_builder(intro_list( - available_intros, - "Add Intro", - &format!( - "/v2/intros/add/{}/{}", - guild_id, channel_name - ), - )) - }); - } - } + } - b + b + }) }) }) .build(), )) } +fn moderator_dashboard(state: &ApiState) -> HtmxBuilder { + HtmxBuilder::new(Tag::Empty).link("Go back to old UI", &format!("{}/old", state.origin)) +} + pub(crate) async fn login(State(state): State) -> Html { let authorize_uri = format!("https://discord.com/api/oauth2/authorize?client_id={}&redirect_uri={}/v2/auth&response_type=code&scope=guilds.members.read%20guilds%20identify", state.secrets.client_id, state.origin); Html( HtmxBuilder::new(Tag::Html) - .head(|b| { - b.title("MemeJoin - Login") - .script( - "https://unpkg.com/htmx.org@1.9.3", - Some("sha384-lVb3Rd/Ca0AxaoZg5sACe8FJKF0tnUgR2Kd7ehUOG5GCcROv5uBIZsOqovBAcWua"), - ) - .script("https://unpkg.com/hyperscript.org@0.9.9", None) - .style_link("https://cdn.jsdelivr.net/npm/@picocss/pico@1/css/pico.min.css") - }) + .push_builder(page_header("MemeJoin - Login")) .link("Login", &authorize_uri) .build(), ) diff --git a/src/routes.rs b/src/routes.rs index dff6894..92a22b5 100644 --- a/src/routes.rs +++ b/src/routes.rs @@ -9,7 +9,7 @@ use axum::{ }; use axum_extra::extract::{cookie::Cookie, CookieJar}; -use reqwest::{Proxy, StatusCode}; +use reqwest::{Proxy, StatusCode, Url}; use serde::{Deserialize, Serialize}; use serde_json::{json, Value}; use tracing::{error, info}; @@ -221,11 +221,13 @@ pub(crate) async fn v2_auth( ); // TODO: add permissions based on roles + let uri = Url::parse(&state.origin).expect("should be a valid url"); + let mut cookie = Cookie::new("access_token", token.clone()); - cookie.set_path("/"); + cookie.set_path(uri.path().to_string()); cookie.set_secure(true); - Ok((jar.add(cookie), Redirect::to("/"))) + Ok((jar.add(cookie), Redirect::to(&format!("{}/", state.origin)))) } pub(crate) async fn auth( @@ -240,7 +242,7 @@ pub(crate) async fn auth( let mut data = HashMap::new(); - let redirect_uri = format!("{}/auth", state.origin); + let redirect_uri = format!("{}/old/auth", state.origin); data.insert("client_id", state.secrets.client_id.as_str()); data.insert("client_secret", state.secrets.client_secret.as_str()); data.insert("grant_type", "authorization_code");