[cmd] art

commands/message/core/help.js

@@ -28,11 +28,14 @@ function createCommandPage(context, prefix, command){
     value: command.aliases.join(', '),
     inline: true
   })
+
+  // TODO: maybe try building a little parser that highlights things via ansi
   if(command.metadata.usage) page.fields.push({
     name: `${icon("util")} Usage`,
-    value: codeblock("ansi", [prefix + command.metadata.usage]),
+    value: codeblock("py", [prefix + command.metadata.usage]),
     inline: true
   })
+  
   if(command.metadata.examples){
     let ex = []
     for(const e of command.metadata.examples) ex.push(prefix + e)

commands/message/fun/art.js

@@ -0,0 +1,108 @@
+const { createEmbed } = require('../../../labscore/utils/embed')
+const { format } = require('../../../labscore/utils/ansi')
+const { editOrReply } = require('../../../labscore/utils/message')
+const { STATICS } = require('../../../labscore/utils/statics')
+
+const { paginator } = require('../../../labscore/client');
+
+const superagent = require('superagent');
+const { icon, highlight, codeblock } = require('../../../labscore/utils/markdown');
+
+const SIZES = Object.freeze({
+  "wallpaper": { x: 1920, y: 1080},
+  "phone": { x: 1170, y: 2353},
+  "avatar": { x: 512, y: 512}
+})
+
+function validateNumber(input, low, high){
+  if(input == "rand") return true;
+  if(isNaN(input)) return true;
+  return (parseInt(input) <= high && parseInt(input) >= low);
+}
+
+module.exports = {
+  name: 'art',
+  aliases: ['wallpaper'],
+  label: 'text',
+  metadata: {
+    description: 'crazy ai art',
+    examples: ['art -type wallpaper -seed 839648 -variance 8866 -rotate 1'],
+    category: 'fun',
+    usage: `art [-type <${Object.keys(SIZES).join('|')}>] [-seed <10000-999999>] [-variance <1000-9999>] [-rotate <0-360>]`
+  },
+  args: [
+    { name: 'type', default: 'wallpaper', required: false },
+    { name: 'seed', default: 'rand', required: false },
+    { name: 'variance', default: 'rand', required: false },
+    { name: 'rotate', default: 'rand', required: false }
+  ],
+  run: async (context, args) => {
+    context.triggerTyping();
+    let response = await editOrReply(context, createEmbed("loading", context, `Generating image...`))
+    try{
+      let seed = Math.floor(Math.random() * 999999) + 100000,
+        variance = Math.floor(Math.random() * 9999) + 1000,
+        rotate = Math.floor(Math.random() * 360)
+
+      if(!validateNumber(args.seed, 10000, 999999)) return await editOrReply(context, createEmbed("warning", context, "Invalid Seed (must be between 10000 and 999999)"))
+      if(args.seed !== "rand") seed = parseInt(args.seed);
+
+      if(!validateNumber(args.variance, 1000, 9999)) return await editOrReply(context, createEmbed("warning", context, "Invalid Variance (must be between 1000 and 9999)"))
+      if(args.variance !== "rand") variance = parseInt(args.variance);
+
+      if(!validateNumber(args.rotate, 0, 360)) return await editOrReply(context, createEmbed("warning", context, "Invalid Rotation (must be between 0 and 360)"))
+      if(args.rotate !== "rand") rotate = parseInt(args.rotate);
+
+      if(!SIZES[args.type.toLowerCase()]) return await editOrReply(context, createEmbed("warning", context, `Invalid Type (must be one of '${Object.keys(SIZES).join(`', '`)}')`))
+      let sizeX = SIZES[args.type.toLowerCase()].x,
+        sizeY = SIZES[args.type.toLowerCase()].y
+
+      let timings = Date.now();
+      let res = await superagent.get(`https://limb.us-east1-gke.intellij.net/generate_art_json`)
+        .query({
+          "seed": seed,
+          "x_resolution": sizeX,
+          "y_resolution": sizeY,
+          "direction": "X",
+          "index": "4",
+          "variance": variance,
+          "architecture": "densenet",
+          "activation": "softsign",
+          "width": "3",
+          "depth": "5",
+          "alpha": "1.8097",
+          "beta": "-0.08713800000000001",
+          "antialiasing": "true",
+          "antialiasing_factor": "2",
+          "noise_factor": "0",
+          "color_space": "rgb",
+          "scale": "2.811",
+          "rotation": rotate,
+          "offset_x": "1202",
+          "offset_y": "257",
+          "function_": "quadratic",
+          "custom_function": "None"
+        })
+      
+      res = JSON.parse(res.text)
+
+      await response.edit({
+        embeds: [
+          createEmbed("default", context, {
+            description: `${codeblock(`py`, [`lc.art -type ${args.type.toLowerCase()} -seed ${seed} -variance ${variance} -rotate ${rotate}`])}`,
+            image: {
+              url: res.image_link
+            },
+            footer: {
+              iconUrl: `https://cdn.discordapp.com/avatars/${context.application.id}/${context.application.icon}.png?size=256`,
+              text: `labsCore • Took ${((Date.now() - timings) / 1000).toFixed(2)}s`
+            }
+          })
+        ]
+      })
+    }catch(e){
+      console.log(e)
+      return editOrReply(context, {embeds:[createEmbed("error", context, `Unable to generate image.`)]})
+    }
+  }
+};

labscore/paginator/structures/BasePaginator.js

@@ -126,6 +126,10 @@ module.exports = class BasePaginator extends EventEmitter {
     return this.commandMessage;
   }
 
+  async appendPage(page) {
+    this.pages.push(page)
+  }
+
   stop(timeout = false) {
     this.emit("stop", this, timeout);
     this.removeAllListeners();

labscore/utils/message.js

@@ -2,6 +2,6 @@ module.exports.editOrReply = function(context, message, disableReference = false
   // Apply message_reference
   if(!message.message_reference && !disableReference) message.reference = true
   if(!message.allowedMentions) message.allowedMentions = {parse: [], repliedUser: false}
-  if(!message.content && !message.embed && !message.embeds && !message.components) return context.editOrReply({embeds: [message]})
+  if(!message.content && !message.embed && !message.embeds && !message.components && !message.files) return context.editOrReply({embeds: [message]})
   return context.editOrReply(message)
 }