Stephen
5 months ago
15 changed files with 643 additions and 303 deletions
Split View
Diff Options
-
+1 -0.env
-
+75 -0Cargo.lock
-
+5 -3Cargo.toml
-
+5 -0diesel.toml
-
+0 -0migrations/.gitkeep
-
+6 -0migrations/00000000000000_diesel_initial_setup/down.sql
-
+36 -0migrations/00000000000000_diesel_initial_setup/up.sql
-
+1 -0migrations/2020-11-21-032604_create_user_programs/down.sql
-
+11 -0migrations/2020-11-21-032604_create_user_programs/up.sql
-
+43 -0src/framebuffer.rs
-
+291 -0src/handler.rs
-
+20 -300src/main.rs
-
+18 -0src/models.rs
-
+123 -0src/program.rs
-
+8 -0src/schema.rs
+ 1
- 0
.env
View File
| @ -0,0 +1 @@ | |||
| DATABASE_URL=postgres://stephen@localhost/cat_disruptor_6500 | |||
+ 75
- 0
Cargo.lock
View File
+ 5
- 3
Cargo.toml
View File
+ 5
- 0
diesel.toml
View File
| @ -0,0 +1,5 @@ | |||
| # For documentation on how to configure this file, | |||
| # see diesel.rs/guides/configuring-diesel-cli | |||
| [print_schema] | |||
| file = "src/schema.rs" | |||
+ 0
- 0
migrations/.gitkeep
View File
+ 6
- 0
migrations/00000000000000_diesel_initial_setup/down.sql
View File
| @ -0,0 +1,6 @@ | |||
| -- This file was automatically created by Diesel to setup helper functions | |||
| -- and other internal bookkeeping. This file is safe to edit, any future | |||
| -- changes will be added to existing projects as new migrations. | |||
| DROP FUNCTION IF EXISTS diesel_manage_updated_at(_tbl regclass); | |||
| DROP FUNCTION IF EXISTS diesel_set_updated_at(); | |||
+ 36
- 0
migrations/00000000000000_diesel_initial_setup/up.sql
View File
| @ -0,0 +1,36 @@ | |||
| -- This file was automatically created by Diesel to setup helper functions | |||
| -- and other internal bookkeeping. This file is safe to edit, any future | |||
| -- changes will be added to existing projects as new migrations. | |||
| -- Sets up a trigger for the given table to automatically set a column called | |||
| -- `updated_at` whenever the row is modified (unless `updated_at` was included | |||
| -- in the modified columns) | |||
| -- | |||
| -- # Example | |||
| -- | |||
| -- ```sql | |||
| -- CREATE TABLE users (id SERIAL PRIMARY KEY, updated_at TIMESTAMP NOT NULL DEFAULT NOW()); | |||
| -- | |||
| -- SELECT diesel_manage_updated_at('users'); | |||
| -- ``` | |||
| CREATE OR REPLACE FUNCTION diesel_manage_updated_at(_tbl regclass) RETURNS VOID AS $$ | |||
| BEGIN | |||
| EXECUTE format('CREATE TRIGGER set_updated_at BEFORE UPDATE ON %s | |||
| FOR EACH ROW EXECUTE PROCEDURE diesel_set_updated_at()', _tbl); | |||
| END; | |||
| $$ LANGUAGE plpgsql; | |||
| CREATE OR REPLACE FUNCTION diesel_set_updated_at() RETURNS trigger AS $$ | |||
| BEGIN | |||
| IF ( | |||
| NEW IS DISTINCT FROM OLD AND | |||
| NEW.updated_at IS NOT DISTINCT FROM OLD.updated_at | |||
| ) THEN | |||
| NEW.updated_at := current_timestamp; | |||
| END IF; | |||
| RETURN NEW; | |||
| END; | |||
| $$ LANGUAGE plpgsql; | |||
+ 1
- 0
migrations/2020-11-21-032604_create_user_programs/down.sql
View File
| @ -0,0 +1 @@ | |||
| DROP TABLE user_programs | |||
+ 11
- 0
migrations/2020-11-21-032604_create_user_programs/up.sql
View File
| @ -0,0 +1,11 @@ | |||
| CREATE TABLE user_programs ( | |||
| id SERIAL PRIMARY KEY, | |||
| discord_user_id NUMERIC NOT NULL, | |||
| name VARCHAR NOT NULL, | |||
| code TEXT NOT NULL | |||
| ); | |||
| CREATE UNIQUE INDEX user_programs_name_id_index ON user_programs ( | |||
| discord_user_id, | |||
| name | |||
| ); | |||
+ 43
- 0
src/framebuffer.rs
View File
| @ -0,0 +1,43 @@ | |||
| #[derive(Clone)] | |||
| pub(crate) struct FrameBuffer { | |||
| width: u32, | |||
| height: u32, | |||
| buffer: Vec<u8>, | |||
| } | |||
| impl FrameBuffer { | |||
| pub fn new(width: u32, height: u32) -> Self { | |||
| Self { | |||
| width, | |||
| height, | |||
| buffer: vec![0; width as usize * height as usize * 4], | |||
| } | |||
| } | |||
| pub fn set_pixel(&mut self, x: u32, y: u32, red: u8, green: u8, blue: u8, alpha: u8) { | |||
| if x >= self.width || y >= self.height { | |||
| return; | |||
| } | |||
| let x = x as usize; | |||
| let y = y as usize; | |||
| self.buffer[(x + self.width as usize * y) * 4] = red; | |||
| self.buffer[(x + self.width as usize * y) * 4 + 1] = green; | |||
| self.buffer[(x + self.width as usize * y) * 4 + 2] = blue; | |||
| self.buffer[(x + self.width as usize * y) * 4 + 3] = alpha; | |||
| } | |||
| pub fn as_png_vec(&self) -> Vec<u8> { | |||
| let mut buffer: Vec<u8> = Vec::new(); | |||
| { | |||
| let mut encoder = png::Encoder::new(&mut buffer, self.width, self.height); | |||
| encoder.set_color(png::ColorType::RGBA); | |||
| encoder.set_depth(png::BitDepth::Eight); | |||
| let mut writer = encoder.write_header().unwrap(); | |||
| writer.write_image_data(&self.buffer).unwrap(); | |||
| } | |||
| buffer | |||
| } | |||
| } | |||
+ 291
- 0
src/handler.rs
View File
| @ -0,0 +1,291 @@ | |||
| use crate::framebuffer::FrameBuffer; | |||
| use crate::program::Program; | |||
| use diesel::PgConnection; | |||
| use phf::phf_map; | |||
| use serenity::async_trait; | |||
| use serenity::http::AttachmentType; | |||
| use serenity::model::channel::{Message, ReactionType}; | |||
| use serenity::model::id::UserId; | |||
| use serenity::model::prelude::Ready; | |||
| use serenity::prelude::*; | |||
| use std::borrow::Cow; | |||
| use std::collections::HashMap; | |||
| use std::str::FromStr; | |||
| use std::sync::{Arc, Mutex}; | |||
| use xbasic::basic_io::BasicIO; | |||
| use xbasic::expr::ExprValue; | |||
| use xbasic::xbasic::XBasicBuilder; | |||
| static EMOJI_MAP: phf::Map<&'static str, &'static str> = phf_map! { | |||
| "cat" => "🐈", | |||
| "chicken" => "🐔", | |||
| "spaghetti" => "🍝", | |||
| "dog" => "🐕", | |||
| "bot" => "🤖", | |||
| "mango" => "🥭", | |||
| "banana" => "🍌", | |||
| "bee" => "🐝" | |||
| }; | |||
| struct DiscordIO { | |||
| s: String, | |||
| frame: Option<FrameBuffer>, | |||
| } | |||
| impl DiscordIO { | |||
| fn new() -> Self { | |||
| Self { | |||
| s: String::new(), | |||
| frame: None, | |||
| } | |||
| } | |||
| } | |||
| impl BasicIO for DiscordIO { | |||
| fn read_line(&mut self) -> String { | |||
| unimplemented!() | |||
| } | |||
| fn write_line(&mut self, line: String) { | |||
| self.s += &*(line + "\r\n"); | |||
| } | |||
| } | |||
| pub(crate) struct Handler { | |||
| programs: Arc<Mutex<HashMap<UserId, Program>>>, | |||
| conn: Arc<Mutex<PgConnection>>, | |||
| } | |||
| impl Handler { | |||
| pub fn new(conn: PgConnection) -> Self { | |||
| Self { | |||
| programs: Arc::new(Mutex::new(HashMap::new())), | |||
| conn: Arc::new(Mutex::new(conn)), | |||
| } | |||
| } | |||
| } | |||
| #[async_trait] | |||
| impl EventHandler for Handler { | |||
| async fn message(&self, ctx: Context, msg: Message) { | |||
| for (key, value) in EMOJI_MAP.entries() { | |||
| let msg_lower = format!(" {} ", msg.content.to_lowercase()); | |||
| if msg_lower.contains(&format!(" {} ", key)) | |||
| || msg_lower.contains(&format!(" {}s ", key)) | |||
| { | |||
| let reaction_type = match ReactionType::from_str(value) { | |||
| Ok(x) => x, | |||
| Err(x) => { | |||
| println!("Could not react: {}", x); | |||
| return; | |||
| } | |||
| }; | |||
| if let Err(e) = msg.react(&ctx, reaction_type).await { | |||
| println!("Error reacting: {}", e); | |||
| } | |||
| } | |||
| } | |||
| for line in msg.content.split('\n') { | |||
| if self.programs.lock().unwrap().contains_key(&msg.author.id) { | |||
| match line { | |||
| "!STOP" => { | |||
| self.programs | |||
| .lock() | |||
| .unwrap() | |||
| .remove(&msg.author.id) | |||
| .unwrap(); | |||
| } | |||
| "RUN" => { | |||
| let code = self.programs.lock().unwrap()[&msg.author.id].stringify(); | |||
| let io = DiscordIO::new(); | |||
| let (output, fb, errors) = { | |||
| let mut xbb = XBasicBuilder::new(io); | |||
| xbb.compute_limit(1000000000); | |||
| xbb.define_function("setframe".to_owned(), 2, |args, io| { | |||
| let w = args[0].clone().into_decimal() as u32; | |||
| let h = args[1].clone().into_decimal() as u32; | |||
| io.frame = Some(FrameBuffer::new(w, h)); | |||
| ExprValue::Decimal(0.0) | |||
| }) | |||
| .unwrap(); | |||
| xbb.define_function("setpixel".to_owned(), 5, |args, io| { | |||
| let x = args[0].clone().into_decimal() as u32; | |||
| let y = args[1].clone().into_decimal() as u32; | |||
| let red = args[2].clone().into_decimal() as u8; | |||
| let green = args[3].clone().into_decimal() as u8; | |||
| let blue = args[4].clone().into_decimal() as u8; | |||
| match &mut io.frame { | |||
| Some(fb) => { | |||
| fb.set_pixel(x, y, red, green, blue, 255); | |||
| } | |||
| None => {} | |||
| } | |||
| ExprValue::Decimal(0.0) | |||
| }) | |||
| .unwrap(); | |||
| let mut xb = xbb.build(); | |||
| let _ = xb.run(&format!("{}\n", code)); | |||
| let errors = if xb.error_handler.had_errors | |||
| || xb.error_handler.had_runtime_error | |||
| { | |||
| Some(xb.error_handler.errors.join("\n")) | |||
| } else { | |||
| None | |||
| }; | |||
| (xb.get_io().s.clone(), xb.get_io().frame.clone(), errors) | |||
| }; | |||
| if !output.is_empty() { | |||
| msg.channel_id.say(&ctx, output).await.unwrap(); | |||
| } | |||
| if let Some(fb) = &fb { | |||
| let buf = fb.as_png_vec(); | |||
| msg.channel_id | |||
| .send_message(&ctx, |e| { | |||
| e.add_file(AttachmentType::Bytes { | |||
| data: Cow::Borrowed(&buf), | |||
| filename: "output.png".to_string(), | |||
| }); | |||
| e | |||
| }) | |||
| .await | |||
| .unwrap(); | |||
| } | |||
| if let Some(e) = errors { | |||
| msg.channel_id.say(&ctx, e).await.unwrap(); | |||
| } | |||
| } | |||
| "LIST" => { | |||
| msg.channel_id | |||
| .say( | |||
| &ctx, | |||
| format!( | |||
| "```\n{}\n```", | |||
| self.programs.lock().unwrap()[&msg.author.id] | |||
| .stringy_line_nums() | |||
| ), | |||
| ) | |||
| .await | |||
| .unwrap(); | |||
| } | |||
| _ => { | |||
| if let Some(name) = line.strip_prefix("SAVE ") { | |||
| let result = self | |||
| .programs | |||
| .lock() | |||
| .unwrap() | |||
| .get_mut(&msg.author.id) | |||
| .unwrap() | |||
| .save_program(&self.conn.lock().unwrap(), msg.author.id, name); | |||
| match result { | |||
| Some(_) => { | |||
| msg.channel_id | |||
| .say(&ctx, format!("Saved as {}", name)) | |||
| .await | |||
| .unwrap(); | |||
| } | |||
| None => { | |||
| msg.channel_id | |||
| .say(&ctx, "Could not save program.") | |||
| .await | |||
| .unwrap(); | |||
| } | |||
| } | |||
| return; | |||
| } | |||
| if let Some(name) = line.strip_prefix("LOAD ") { | |||
| let result = self | |||
| .programs | |||
| .lock() | |||
| .unwrap() | |||
| .get_mut(&msg.author.id) | |||
| .unwrap() | |||
| .load_program(&self.conn.lock().unwrap(), msg.author.id, name); | |||
| match result { | |||
| Some(_) => { | |||
| msg.channel_id | |||
| .say(&ctx, format!("Loaded {} into memory.", name)) | |||
| .await | |||
| .unwrap(); | |||
| } | |||
| None => { | |||
| msg.channel_id | |||
| .say(&ctx, "Could not load program into memory.") | |||
| .await | |||
| .unwrap(); | |||
| } | |||
| } | |||
| return; | |||
| } | |||
| let mut split = line.splitn(2, ' '); | |||
| let first = split.next().unwrap(); | |||
| if let Ok(num) = first.parse::<u32>() { | |||
| match split.next() { | |||
| Some(x) => { | |||
| if x.is_empty() { | |||
| let _ = self | |||
| .programs | |||
| .lock() | |||
| .unwrap() | |||
| .get_mut(&msg.author.id) | |||
| .unwrap() | |||
| .code | |||
| .remove(&num); | |||
| return; | |||
| } | |||
| self.programs | |||
| .lock() | |||
| .unwrap() | |||
| .get_mut(&msg.author.id) | |||
| .unwrap() | |||
| .code | |||
| .insert(num, x.to_owned()); | |||
| } | |||
| None => { | |||
| let _ = self | |||
| .programs | |||
| .lock() | |||
| .unwrap() | |||
| .get_mut(&msg.author.id) | |||
| .unwrap() | |||
| .code | |||
| .remove(&num); | |||
| } | |||
| } | |||
| } | |||
| } | |||
| } | |||
| } | |||
| if line == "!START" { | |||
| self.programs | |||
| .lock() | |||
| .unwrap() | |||
| .insert(msg.author.id, Program::new()); | |||
| } | |||
| } | |||
| } | |||
| async fn ready(&self, _: Context, ready: Ready) { | |||
| println!("{} is connected", ready.user.name); | |||
| } | |||
| } | |||
+ 20
- 300
src/main.rs
View File
+ 18
- 0
src/models.rs
View File
| @ -0,0 +1,18 @@ | |||
| use crate::schema::user_programs; | |||
| use bigdecimal::BigDecimal; | |||
| #[derive(Queryable)] | |||
| pub struct UserProgram { | |||
| pub id: i32, | |||
| pub discord_user_id: u64, | |||
| pub name: String, | |||
| pub code: String, | |||
| } | |||
| #[derive(Insertable)] | |||
| #[table_name = "user_programs"] | |||
| pub struct NewUserProgram<'a> { | |||
| pub discord_user_id: BigDecimal, | |||
| pub name: &'a str, | |||
| pub code: &'a str, | |||
| } | |||
+ 123
- 0
src/program.rs
View File
| @ -0,0 +1,123 @@ | |||
| use crate::models::NewUserProgram; | |||
| use crate::schema::user_programs::columns; | |||
| use crate::schema::user_programs::columns::discord_user_id; | |||
| use crate::schema::user_programs::dsl::user_programs; | |||
| use bigdecimal::{BigDecimal, FromPrimitive}; | |||
| use diesel::{BoolExpressionMethods, ExpressionMethods, PgConnection, QueryDsl, RunQueryDsl}; | |||
| use serenity::model::id::UserId; | |||
| use std::collections::HashMap; | |||
| pub(crate) struct Program { | |||
| pub(crate) code: HashMap<u32, String>, | |||
| } | |||
| impl Program { | |||
| pub fn new() -> Self { | |||
| Self { | |||
| code: HashMap::new(), | |||
| } | |||
| } | |||
| pub fn stringify(&self) -> String { | |||
| let mut code: Vec<(u32, String)> = | |||
| self.code.iter().map(|(a, b)| (*a, b.to_owned())).collect(); | |||
| code.sort_by(|a, b| a.0.cmp(&b.0)); | |||
| code.into_iter() | |||
| .map(|a| a.1) | |||
| .collect::<Vec<String>>() | |||
| .join("\n") | |||
| } | |||
| pub fn stringy_line_nums(&self) -> String { | |||
| let mut code: Vec<(u32, String)> = | |||
| self.code.iter().map(|(a, b)| (*a, b.to_owned())).collect(); | |||
| code.sort_by(|a, b| a.0.cmp(&b.0)); | |||
| code.into_iter() | |||
| .map(|a| format!("{}\t{}", a.0, a.1)) | |||
| .collect::<Vec<String>>() | |||
| .join("\n") | |||
| } | |||
| pub fn save_program(&self, conn: &PgConnection, user_id: UserId, name: &str) -> Option<()> { | |||
| let code = self.stringy_line_nums(); | |||
| let new_program = NewUserProgram { | |||
| discord_user_id: BigDecimal::from_u64(*user_id.as_u64()).unwrap(), | |||
| name, | |||
| code: code.as_str(), | |||
| }; | |||
| diesel::insert_into(user_programs) | |||
| .values(&new_program) | |||
| .on_conflict((discord_user_id, columns::name)) | |||
| .do_update() | |||
| .set(columns::code.eq(&code)) | |||
| .execute(conn) | |||
| .ok()?; | |||
| Some(()) | |||
| } | |||
| pub fn load_program(&mut self, conn: &PgConnection, user_id: UserId, name: &str) -> Option<()> { | |||
| let code: Vec<String> = user_programs | |||
| .filter( | |||
| columns::discord_user_id | |||
| .eq(BigDecimal::from_u64(*user_id.as_u64()).unwrap()) | |||
| .and(columns::name.eq(name)), | |||
| ) | |||
| .limit(1) | |||
| .select(columns::code) | |||
| .get_results(conn) | |||
| .ok()?; | |||
| if code.is_empty() { | |||
| return None; | |||
| } | |||
| let code = &code[0]; | |||
| self.parse_string(code)?; | |||
| Some(()) | |||
| } | |||
| fn parse_string(&mut self, code: &str) -> Option<()> { | |||
| let mut valid = true; | |||
| let code: HashMap<u32, String> = code | |||
| .split('\n') | |||
| .map(|line| { | |||
| let mut iter = line.splitn(2, &[' ', '\t'][..]); | |||
| // This unwrap_or_else thing is pretty ugly | |||
| // Is there a better way? | |||
| let line_num: u32 = iter | |||
| .next() | |||
| .unwrap_or_else(|| { | |||
| valid = false; | |||
| "0" | |||
| }) | |||
| .parse() | |||
| .unwrap_or_else(|_| { | |||
| valid = false; | |||
| 0 | |||
| }); | |||
| let line_code = iter | |||
| .next() | |||
| .unwrap_or_else(|| { | |||
| valid = false; | |||
| "" | |||
| }) | |||
| .to_owned(); | |||
| (line_num, line_code) | |||
| }) | |||
| .collect(); | |||
| if valid { | |||
| self.code = code; | |||
| Some(()) | |||
| } else { | |||
| None | |||
| } | |||
| } | |||
| } | |||
+ 8
- 0
src/schema.rs
View File
| @ -0,0 +1,8 @@ | |||
| table! { | |||
| user_programs (id) { | |||
| id -> Int4, | |||
| discord_user_id -> Numeric, | |||
| name -> Varchar, | |||
| code -> Text, | |||
| } | |||
| } | |||