fixes for sm cmd

src/api/cr.ins/main.ts

@@ -0,0 +1,3 @@
+import { Server, ServerManagement } from '../../class';
+
+export default (management: ServerManagement) => new Server(management, 3891, `${__dirname}/routes`);

src/api/cr.ins/routes/index.ts

@@ -0,0 +1 @@
+export { default as root } from './root';

src/api/cr.ins/routes/root.ts

@@ -0,0 +1,32 @@
+import { LocalStorage, Route, Server } from '../../../class';
+
+export default class Root extends Route {
+  constructor(server: Server) {
+    super(server);
+    this.conf = {
+      path: '/',
+    };
+  }
+
+  public bind() {
+    this.router.get('/m/:id', async (req, res) => {
+      try {
+        const id = req.params.id.split('.')[0];
+        const file = await this.server.client.db.File.findOne({ identifier: id });
+        if (!file) return res.status(404).json({ code: this.constants.codes.NOT_FOUND, message: this.constants.messages.NOT_FOUND });
+        if (file.downloaded >= file.maxDownloads) return res.status(404).json({ code: this.constants.codes.NOT_FOUND, message: this.constants.messages.NOT_FOUND });
+        if (req.query.d === '1') {
+          res.contentType('text/html');
+          const decomp = await LocalStorage.decompress(file.data);
+          res.status(200).send(decomp);
+        } else {
+          res.contentType(file.mimeType);
+          res.status(200).send(file.data);
+        }
+        return await file.updateOne({ $inc: { downloaded: 1 } });
+      } catch (err) {
+        return this.handleError(err, res);
+      }
+    });
+  }
+}

src/api/index.ts

@@ -1,5 +1,7 @@
 import locsh from './loc.sh/main';
+import crins from './cr.ins/main';
 
 export default {
   'loc.sh': locsh,
+  'cr.ins': crins,
 };

src/api/loc.sh/routes/root.ts

@@ -23,18 +23,5 @@ export default class Root extends Route {
         return res.status(500).json({ code: this.constants.codes.SERVER_ERROR, message: this.constants.messages.SERVER_ERROR });
       }
     });
-
-    this.router.get('/m/:id', async (req, res) => {
-      try {
-        const id = req.params.id.split('.')[0];
-        const file = await this.server.client.db.File.findOne({ identifier: id });
-        if (file.downloaded >= file.maxDownloads) return res.status(404).json({ code: this.constants.codes.NOT_FOUND, message: this.constants.messages.NOT_FOUND });
-        res.contentType(file.mimeType);
-        res.status(200).send(file.data);
-        return await file.updateOne({ $inc: { downloaded: 1 } });
-      } catch (err) {
-        return this.handleError(err, res);
-      }
-    });
   }
 }

src/commands/storemessages.ts

@@ -25,12 +25,12 @@ export default class StoreMessages extends Command {
       if (args[1]) {
         messages = messages.filter((m) => m.author.id === args[1]);
       }
-      let html = `<h3>Library of Code sp-us</h3><strong>Channel:</strong> ${chan.name} (${chan.id})<br><strong>Generated by:</strong> ${message.author.username}#${message.author.discriminator}<br><strong>Generated at:</strong> ${new Date().toLocaleString('en-us')}<br><br>`;
+      let html = `<strong><i>CLASSIFIED RESOURCE: CL-GEN/AD</i></strong><br><h3>Library of Code sp-us</h3><strong>Channel:</strong> ${chan.name} (${chan.id})<br><strong>Generated by:</strong> ${message.author.username}#${message.author.discriminator}<br><strong>Generated at:</strong> ${new Date().toLocaleString('en-us')}<br><br>`;
       for (const msg of messages) {
         html += `(<i>${new Date(msg.timestamp).toLocaleString('en-us')}</i>) [<strong>${msg.author.username}#${msg.author.discriminator} - ${msg.author.id}</strong>]: ${msg.cleanContent}<br>`;
       }
       message.delete();
-      const identifier = randomBytes(10).toString('hex');
+      const identifier = randomBytes(20).toString('hex');
 
       const comp = await LocalStorage.compress(html);
       const file = new this.client.db.File({
@@ -39,11 +39,11 @@ export default class StoreMessages extends Command {
         mimeType: 'application/gzip',
         data: comp,
         downloaded: 0,
-        maxDownloads: 1,
+        maxDownloads: 20,
       });
       await file.save();
       loadingMessage.delete();
-      this.client.getDMChannel(message.author.id).then((c) => c.createMessage(`https://loc.sh/m/${identifier}.html.gz`)).catch(() => this.error(message.channel, 'Could not send a DM to you.'));
+      this.client.getDMChannel(message.author.id).then((c) => c.createMessage(`https://cr.ins/m/${identifier}.html.gz || https://cr.ins/m/${identifier}.html?d=1`)).catch(() => this.error(message.channel, 'Could not send a DM to you.'));
       return this.success(message.channel, `Fetched messages for <#${chan.id}>. Check your DMs for link to access.`);
     } catch (err) {
       return this.client.util.handleError(err, message, this);