Add noteedit

Author: Tibowl

src/commands/misc/noteadd.ts

@@ -11,9 +11,11 @@ export default class NoteAdd extends Command {
         super({
             name,
             category: "Misc",
-            help: `Create a new note, notes are per category or user.
+            help: `Create a new note.
 
-Requires "Manage Messages" permissions to add, but anyone can see.`,
+Notes are either per category (when used in a guild) or per user (when used in DM's).
+
+Requires "Manage Messages" permissions to add/remove/edit, but anyone can see.`,
             usage: "noteadd <name>",
             aliases: ["na", "createnote", "addnote", "newnote", "an"],
             options: [{

src/commands/misc/noteedit.ts

@@ -0,0 +1,85 @@
+import { CommandInteraction, Message, MessageEmbed } from "discord.js"
+import client from "../../main"
+import Command from "../../utils/Command"
+import { CommandSource, SendMessage } from "../../utils/Types"
+import { Colors, getCategory, getUser, sendMessage } from "../../utils/Utils"
+
+
+export default class NoteEdit extends Command {
+    constructor(name: string) {
+        super({
+            name,
+            category: "Misc",
+            help: `Edits a note by ID.
+
+Notes are either per category (when used in a guild) or per user (when used in DM's).
+
+Requires "Manage Messages" permissions to add/remove/edit, but anyone can see.`,
+            usage: "noteedit <id>",
+            aliases: ["editn", "editnote", "enote"],
+            options: [{
+                name: "id",
+                description: "ID to edit",
+                type: "INTEGER",
+                required: true
+            }, {
+                name: "name",
+                description: "New name",
+                type: "STRING",
+                required: true
+            }]
+        })
+    }
+
+    async runInteraction(source: CommandInteraction): Promise<SendMessage | undefined> {
+        const { options } = source
+
+        const id = options.getInteger("id", true)
+        const name = options.getString("name", true)
+
+        return this.run(source, id, name)
+    }
+
+    async runMessage(source: Message|CommandInteraction, args: string[]): Promise<SendMessage | undefined> {
+        if (args.length < 1) return this.sendHelp(source)
+
+        const id = parseInt(args[0])
+        if (isNaN(id)) return this.sendHelp(source)
+
+        const name = args.slice(1).join(" ")
+
+        return this.run(source, id, name)
+    }
+
+    async run(source: CommandSource, id: number, newSubject: string): Promise<SendMessage | undefined> {
+        const { notesManager } = client
+        const user = getUser(source)
+
+        const guild = source.guild
+        const category = getCategory(source)
+        const name = category?.name ?? guild?.name ?? user.username
+
+        const guildID = guild?.id ?? "user"
+        const categoryID = category?.id ?? guild?.id ?? user.id
+
+        if (source.member == undefined || typeof source.member.permissions == "string")
+            return sendMessage(source, "Unable to check permissions.", undefined, true)
+        else if (!source.member.permissions.has("MANAGE_MESSAGES"))
+            return sendMessage(source, "You do not have have \"Manage Messages\" permissions in here.", undefined, true)
+
+        if (newSubject.length > 256) return sendMessage(source, "Note name too long")
+
+        const note = notesManager.getNoteById(guildID, categoryID, id)
+        if (note == undefined) return sendMessage(source, `Could not find note #${id} in ${name}`)
+
+        notesManager.editNote(guildID, categoryID, id, newSubject, user.id)
+        const reply = sendMessage(source, new MessageEmbed()
+            .setTitle(`Edited note #${id} in ${name}`)
+            .setColor(Colors.ORANGE)
+            .setDescription(`Original: \`${note.subject}\`
+New: \`${newSubject}\``)
+        )
+
+        return reply
+    }
+}

src/commands/misc/noteremove.ts

@@ -10,9 +10,11 @@ export default class NoteRemove extends Command {
         super({
             name,
             category: "Misc",
-            help: `Removes a new note, notes are per category or user.
+            help: `Removes a note by ID.
 
-Requires "Manage Messages" permissions to add, but anyone can see.`,
+Notes are either per category (when used in a guild) or per user (when used in DM's).
+
+Requires "Manage Messages" permissions to add/remove/edit, but anyone can see.`,
             usage: "noteremove <id>",
             aliases: ["rn", "dn", "removenote", "deletenote", "delnote", "dnote", "rnote"],
             options: [{

src/commands/misc/notes.ts

@@ -11,7 +11,9 @@ export default class Notes extends Command {
         super({
             name,
             category: "Misc",
-            help: `View notes, notes are per category or user.
+            help: `List of notes.
+
+Notes are either per category (when used in a guild) or per user (when used in DM's).
 
 Requires "Manage Messages" permissions to add, but anyone can see.`,
             usage: "notes",

src/utils/NotesManager.ts

@@ -24,6 +24,7 @@ export default class NotesManager {
         this.getNoteByCategoryStatement = this.sql.prepare("SELECT * FROM notes WHERE guild_id = @guild_id AND category_id = @category_id")
         this.getNoteByIdStatement = this.sql.prepare("SELECT * FROM notes WHERE guild_id = @guild_id AND category_id = @category_id AND id = @id")
         this.deleteNoteStatement = this.sql.prepare("DELETE FROM notes WHERE guild_id = @guild_id AND category_id = @category_id AND id = @id")
+        this.editNoteStatement = this.sql.prepare("UPDATE notes SET subject = @subject WHERE guild_id = @guild_id AND category_id = @category_id AND id = @id")
     }
 
     private addNotesStatement: SQLite.Statement
@@ -68,4 +69,15 @@ export default class NotesManager {
         })
         Logger.info(`Deleted note by ${userId} in ${guildId} / ${categoryId} #${id}`)
     }
+
+    private editNoteStatement: SQLite.Statement
+    editNote(guildId: string, categoryId: string, id: number, newSubject: string, userId: string): void {
+        this.editNoteStatement.run({
+            guild_id: guildId,
+            category_id: categoryId,
+            id,
+            subject: newSubject
+        })
+        Logger.info(`Edited note by ${userId} in ${guildId} / ${categoryId} #${id}: ${newSubject}`)
+    }
 }