@ -1,20 +1,20 @@
use crate ::utils ::* ;
use serde_json ::Map ;
use serenity ::framework ::standard ::{ macros ::command , Args , CommandResult } ;
use serenity ::http ::typing ::Typing ;
use serenity ::model ::prelude ::* ;
use serenity ::prelude ::* ;
use crate ::serenity ;
use crate ::Error ;
use crate ::{ Context , PrefixContext } ;
use poise ::command ;
use slug ::slugify ;
use std ::env ;
use std ::fs ;
use std ::fs ::File ;
use std ::io ::Write ;
use std ::path ::Path ;
#[ command ]
async fn challenge ( ctx : & Context , msg : & Message ) -> CommandResult {
msg . reply (
& ctx . http ,
#[ command(prefix_command, slash_command, description_localized( " en-US " , " View the latest handwriting challenge info. " )) ]
pub async fn challenge ( ctx : Context < ' _ > ) -> Result < ( ) , Error > {
ctx . say (
format! (
"Tegaki Tuesday #{n}: <https://tegakituesday.com/{n}>" ,
n = get_challenge_number ( )
@ -24,27 +24,26 @@ async fn challenge(ctx: &Context, msg: &Message) -> CommandResult {
Ok ( ( ) )
}
#[ command ]
async fn submit ( ctx : & Context , msg : & Message ) -> CommandResult {
#[ command (prefix_command, broadcast_typing, description_localized(" en-US " , " Submit to the latest handwriting challenge. " )) ]
pub async fn submit ( ctx : PrefixContext < ' _ > ) -> Result < ( ) , Error > {
// TODO: The code for this command needs to be refactored,
// there are large duplicated sections that need to be merged somehow.
let guild_data = get_guild_data ( ) ;
let guild = . guild_id . unwrap ( ) . as_u64 ( ) . to_string ( ) ;
let guild = ctx. msg. guild_id . unwrap ( ) . as_u64 ( ) . to_string ( ) ;
if ! guild_data . contains_key ( & guild )
| | ! & guild_data [ & guild ]
. as_object ( )
. unwrap ( )
. contains_key ( "submissionChannel" )
{
msg . reply ( & ctx . http , "Submissions aren't enabled for this server yet." )
. await ? ;
ctx . msg . reply ( & ctx . discord . http , "Submissions aren't enabled for this server yet." ) . await ? ;
return Ok ( ( ) ) ;
}
let current_guild_data = & guild_data [ & guild ] . as_object ( ) . unwrap ( ) ;
let submission_channel = current_guild_data [ "submissionChannel" ] . as_str ( ) . unwrap ( ) ;
if submission_channel ! = & msg . channel_id . as_u64 ( ) . to_string ( ) {
. reply (
& ctx . ,
if submission_channel ! = ctx . msg . channel_id . as_u64 ( ) . to_string ( ) {
ctx. msg. reply (
& ctx . discord. http,
format! (
"Sorry, submissions aren't permitted here. Please go to <#{}>. Thanks!" ,
guild
@ -53,12 +52,10 @@ async fn submit(ctx: &Context, msg: &Message) -> CommandResult {
. await ? ;
return Ok ( ( ) ) ;
}
if msg . attachments . len ( ) = = 0 {
msg . reply ( & ctx . http , "Please attach at least one image." )
. await ? ;
if ctx . msg . attachments . len ( ) = = 0 {
ctx . msg . reply ( & ctx . discord . http , "Please attach at least one image." ) . await ? ;
return Ok ( ( ) ) ;
}
let typing = Typing ::start ( ctx . http . clone ( ) , * msg . channel_id . as_u64 ( ) ) . unwrap ( ) ;
let challenge_number = get_challenge_number ( ) ;
let submission_images_dir = get_submission_images_dir ( ) ;
@ -71,10 +68,10 @@ async fn submit(ctx: &Context, msg: &Message) -> CommandResult {
let mut invalid_types = false ;
let mut requires_rebuild = false ;
for ( i , submission ) in submission_data . iter_mut ( ) . enumerate ( ) {
if is_matching_submission ( & submission , & ) {
if is_matching_submission ( & submission , & ctx. msg. author ) {
existing_submitter = true ;
let mut images = submission [ "images" ] . as_array_mut ( ) . unwrap ( ) . clone ( ) ;
for attachment in . attachments . iter ( ) {
for attachment in ctx. msg. attachments . iter ( ) {
let extension ;
if let Some ( content_type ) = & attachment . content_type {
if content_type = = "image/png" {
@ -93,8 +90,8 @@ async fn submit(ctx: &Context, msg: &Message) -> CommandResult {
let file_name = format! (
"{}-{}-{}-{}.{}" ,
i + 1 ,
slugify ( & . author . name ) ,
. author . discriminator ,
slugify ( & ctx. msg. author . name ) ,
ctx. msg. author . discriminator ,
images . len ( ) + 1 ,
extension
) ;
@ -113,10 +110,10 @@ async fn submit(ctx: &Context, msg: &Message) -> CommandResult {
let mut submitter_data = Map ::new ( ) ;
submitter_data . insert (
String ::from ( "username" ) ,
format! ( "{}#{}" , . author . name , msg . author . discriminator ) . into ( ) ,
format! ( "{}#{}" , ctx. msg. author . name , ctx . msg . author . discriminator ) . into ( ) ,
) ;
let mut images : Vec < String > = Vec ::new ( ) ;
for attachment in . attachments . iter ( ) {
for attachment in ctx. msg. attachments . iter ( ) {
let extension ;
if let Some ( content_type ) = & attachment . content_type {
if content_type = = "image/png" {
@ -135,8 +132,8 @@ async fn submit(ctx: &Context, msg: &Message) -> CommandResult {
let file_name = format! (
"{}-{}-{}{}.{}" ,
submission_data . len ( ) + 1 ,
slugify ( & . author . name ) ,
. author . discriminator ,
slugify ( & ctx. msg. author . name ) ,
ctx. msg. author . discriminator ,
if images . len ( ) = = 0 {
String ::from ( "" )
} else {
@ -153,7 +150,7 @@ async fn submit(ctx: &Context, msg: &Message) -> CommandResult {
submitter_data . insert ( String ::from ( "images" ) , images . into ( ) ) ;
submitter_data . insert (
String ::from ( "id" ) ,
. author . id . as_u64 ( ) . to_string ( ) . into ( ) ,
ctx. msg. author . id . as_u64 ( ) . to_string ( ) . into ( ) ,
) ;
submission_data . push ( submitter_data . into ( ) ) ;
}
@ -169,18 +166,17 @@ async fn submit(ctx: &Context, msg: &Message) -> CommandResult {
} else if invalid_types {
message . push_str ( "Sorry, your submission could not be uploaded; only **.png**, **.jpg**, and **.jpeg** files are permitted." ) ;
}
let _ = typing . stop ( ) ;
msg . reply ( & ctx . http , message ) . await ? ;
ctx . msg . reply ( & ctx . discord . http , message ) . await ? ;
Ok ( ( ) )
}
#[ command ]
async fn images ( ctx : & Context , msg : & Message ) -> CommandResult {
#[ command (prefix_command, slash_command, description_localized(" en-US " , " List images in your current submission, if available. " )) ]
pub async fn images ( ctx : Context < ' _ > ) -> Result < ( ) , Error > {
let submission_data = get_current_submission_data ( ) ;
let images : Vec < String > = {
let mut images = Vec ::new ( ) ;
for submission in submission_data . iter ( ) {
if is_matching_submission ( & submission , & msg ) {
if is_matching_submission ( & submission , & ctx. author ( ) ) {
for image in submission [ "images" ] . as_array ( ) . unwrap ( ) . iter ( ) {
images . push ( String ::from ( image . as_str ( ) . unwrap ( ) ) ) ;
}
@ -191,8 +187,7 @@ async fn images(ctx: &Context, msg: &Message) -> CommandResult {
} ;
let challenge_number = get_challenge_number ( ) ;
if images . len ( ) = = 0 {
msg . reply (
& ctx . http ,
ctx . say (
format! (
"You haven't submitted anything for Tegaki Tuesday #{}." ,
challenge_number
@ -213,41 +208,29 @@ async fn images(ctx: &Context, msg: &Message) -> CommandResult {
image
) ) ;
}
msg. reply ( & ctx . http , message ) . await ? ;
ctx. say ( message ) . await ? ;
Ok ( ( ) )
}
#[ command ]
// imageDelete instead of image_delete to keep command naming from Python version
#[ allow(non_snake_case) ]
async fn imageDelete ( ctx : & Context , msg : & Message , mut args : Args ) -> CommandResult {
let number ;
match args . single ::< i32 > ( ) {
Ok ( value ) = > number = value ,
Err ( _ ) = > {
msg . reply ( & ctx . http , format! ( "Please provide the image number you want to delete. You can get a list of your submitted images using `{}images`." , env ::var ( "PREFIX" ) . unwrap ( ) ) ) . await ? ;
return Ok ( ( ) ) ;
}
}
#[ command(prefix_command, slash_command, description_localized( " en-US " , " Delete images from your current submission using image numbers from the images command. " )) ]
pub async fn imagedelete (
ctx : Context < ' _ > ,
number : i32 ,
) -> Result < ( ) , Error > {
if number < 1 {
msg . reply (
& ctx . http ,
"That isn't a valid image number. Image numbers start at 1." ,
)
. await ? ;
ctx . say ( "That isn't a valid image number. Image numbers start at 1." ) . await ? ;
return Ok ( ( ) ) ;
}
let challenge_number = get_challenge_number ( ) ;
let mut submission_data = get_current_submission_data ( ) ;
for ( i , submission ) in submission_data . iter_mut ( ) . enumerate ( ) {
if ! is_matching_submission ( & submission , & msg ) {
if ! is_matching_submission ( & submission , & ctx. author ( ) ) {
continue ;
}
let mut images = submission [ "images" ] . as_array ( ) . unwrap ( ) . clone ( ) ;
let image_count = images . len ( ) ;
if image_count < number . try_into ( ) . unwrap ( ) {
msg . reply (
& ctx . http ,
ctx . say (
if image_count = = 0 {
// This is an edge case that should never happen.
// In this scenario, there is submission data with an empty image list.
@ -296,11 +279,12 @@ async fn imageDelete(ctx: &Context, msg: &Message, mut args: Args) -> CommandRes
for ( j , image ) in images . iter_mut ( ) . enumerate ( ) {
let old = image . as_str ( ) . unwrap ( ) ;
let from = format! ( "{}/{}" , submission_images_dir , old ) ;
let author = ctx . author ( ) ;
let new = format! (
"{}-{}-{}{}.{}" ,
i + 1 ,
slugify ( & msg. author. name ) ,
msg. author. discriminator ,
slugify ( & . name ) ,
. discriminator ,
if j = = 0 {
String ::from ( "" )
} else {
@ -317,11 +301,10 @@ async fn imageDelete(ctx: &Context, msg: &Message, mut args: Args) -> CommandRes
}
set_submission_data ( submission_data ) ;
rebuild_site ( ) ;
msg. reply ( & ctx . http , message ) . await ? ;
ctx. say ( message ) . await ? ;
return Ok ( ( ) ) ;
}
msg . reply (
& ctx . http ,
ctx . say (
format! (
"You haven't submitted anything for Tegaki Tuesday #{}." ,
challenge_number
@ -331,11 +314,15 @@ async fn imageDelete(ctx: &Context, msg: &Message, mut args: Args) -> CommandRes
Ok ( ( ) )
}
#[ command ]
#[ allow(non_snake_case) ]
async fn suggest ( ctx : & Context , msg : & Message , args : Args ) -> CommandResult {
// TODO: make also slash command
#[ command(prefix_command, description_localized( " en-US " , " Make a suggestion for future challenge prompts! " )) ]
pub async fn suggest (
ctx : PrefixContext < ' _ > ,
#[ description = " Suggestion text. Please include passage and source. " ]
suggestion : String ,
) -> Result < ( ) , Error > {
let guild_data = get_guild_data ( ) ;
let channel = ChannelId (
let channel = serenity:: ChannelId(
guild_data [ "suggestionChannel" ]
. as_str ( )
. unwrap ( )
@ -344,33 +331,32 @@ async fn suggest(ctx: &Context, msg: &Message, args: Args) -> CommandResult {
) ;
// User::accent_colour is only available via the REST API
// If we just do msg.author.accent_colour here, we will get None
let a ccent_color = ctx
. http
. get_user ( * msg. author. id . as_u64 ( ) )
let a uthor = & ctx . msg . author ;
let accent_color = ctx . discord . http
. get_user ( * . id . as_u64 ( ) )
. await
. unwrap ( )
. accent_colour ;
channel
. send_message ( & ctx . , | m | {
. send_message ( & ctx . discord. http, | m | {
m . allowed_mentions ( | am | {
am . empty_parse ( ) ;
am
} ) ;
m . embed ( | e | {
let username = format! ( "{}#{}" , msg. author. name , msg . author . discriminator ) ;
let username = format! ( "{}#{}" , . name , author . discriminator ) ;
e . title ( "New suggestion" ) ;
e . description ( format! (
"{}\n\n[See original message]({}) ({})" ,
args . rest ( ) ,
msg . link ( ) ,
msg . guild ( & ctx ) . unwrap ( ) . name
"{}\n\n[See original message]({})" ,
suggestion ,
ctx . msg . link ( ) ,
) ) ;
let mut = serenity ::builder ::CreateEmbedAuthor ::default ( ) ;
. icon_url ( get_avatar ( & msg. author) )
let mut embed_ author = serenity ::builder ::CreateEmbedAuthor ::default ( ) ;
embed_ author
. icon_url ( get_avatar ( & ) )
. name ( username )
. url ( format! ( "https://discord.com/users/{}" , msg. author. id ) ) ;
e . set_author ( ) ;
. url ( format! ( "https://discord.com/users/{}" , . id ) ) ;
e . set_author ( embed_ author) ;
if let Some ( accent_color ) = accent_color {
e . color ( accent_color ) ;
}
@ -380,6 +366,6 @@ async fn suggest(ctx: &Context, msg: &Message, args: Args) -> CommandResult {
} )
. await
. unwrap ( ) ;
. reply ( & ctx . http , "Suggestion sent! Thank you for making a suggestion. If it is chosen to be used in a future challenge, you will be mentioned in the challenge description!" ) . await ? ;
ctx. msg. reply ( & ctx . discord . http , "Suggestion sent! Thank you for making a suggestion. If it is chosen to be used in a future challenge, you will be mentioned in the challenge description!" ) . await ? ;
Ok ( ( ) )
}
