midjourney-ruby 0.1.21 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5c143a32dfee3e9931bcc253a14c6bdc96c50d3fb771dc5ea6d47a874b9c2ec3
-  data.tar.gz: 30945aa298dd6535b0edc64c368936d994e5014ed017fc7d255e84a923d1c15a
+  metadata.gz: 2e648fd2e9826785943bab6a582fca4da21ae4358bf76cd80554b84f25e6df23
+  data.tar.gz: 2e3c6ad1067d655703e40e0b30b2a73c30e0812d9408f4303275253b8a62c68d
 SHA512:
-  metadata.gz: ac86d34aa8426971743b930778b15d6d7ff00753edf987ae7ad63593d9c1cf0a3ec7dd22a174ac054a61fc8772e3438e63b66d07f2b7e1386544bf6df129ae9d
-  data.tar.gz: d7dbeecd569dcf14b7b578357eccaf0070b43bebd2db3c1feb780eb99c93372012a8d0b40df33a6224c84edc44c662c8eedb9d94fc9fcd853d31c7ae5547d821
+  metadata.gz: 487bb6c38f6ed7dc77ad490d02720b2f24d60129bc3c644a4cad178e285c12269e7fc212dac12ad6e5f9b553e0ba8fd6e99bd8b823e94df00d483c0ceb056fb7
+  data.tar.gz: 2819dadc88926ab1aad5d30bfbd331bd65fb0258103768d1dbf8f436865df0d993b94d005b43bf71c6ea7b6746a4e0ecf37ca301902ccc4168aab5f9d157a438

data/lib/midjourney/api/gateway.rb

@@ -0,0 +1,107 @@
+# frozen_string_literal: true
+
+module Midjourney
+  module API
+    #
+    # Discord API Gateway using RestClient
+    #
+    class Gateway
+      BASE_URL = 'https://discord.com/api/v9'
+
+      attr_reader :token, :channel_id
+
+      def initialize(token:, channel_id:)
+        @token = token
+        @channel_id = channel_id
+      end
+
+      def read_messages(limit = 50)
+        get_request("#{BASE_URL}/channels/#{channel_id}/messages?limit=#{limit}")
+      end
+
+      def send_message(message)
+        post_request("#{BASE_URL}/channels/#{channel_id}/messages", message_payload(message))
+      end
+
+      def send_imagine(prompt)
+        post_request("#{BASE_URL}/interactions", imagine_payload(prompt))
+      end
+
+      private
+
+      def get_request(url)
+        JSON.parse RestClient.get(url, headers).body
+      rescue RestClient::ExceptionWithResponse => e
+        puts "Error sending message: #{e.response.body}"
+      end
+
+      def post_request(url, payload)
+        RestClient.post(url, payload, headers)
+      rescue RestClient::ExceptionWithResponse => e
+        puts "Error sending message: #{e.response.body}"
+      end
+
+      def imagine_payload(prompt) # rubocop:disable Metrics/MethodLength
+        {
+          type: 2,
+          application_id: '936929561302675456',
+          guild_id: '1158081328068165704',
+          channel_id: channel_id,
+          session_id: '939307d50a643c3d8cfcf4e756063e3b',
+          data: {
+            version: '1118961510123847772',
+            id: '938956540159881230',
+            name: 'imagine',
+            type: 1,
+            options: [
+              {
+                type: 3,
+                name: 'prompt',
+                value: prompt
+              }
+            ],
+            application_command: {
+              id: '938956540159881230',
+              application_id: '936929561302675456',
+              version: '1118961510123847772',
+              default_member_permissions: nil,
+              type: 1,
+              nsfw: false,
+              name: 'imagine',
+              description: 'Create images with Midjourney',
+              dm_permission: true,
+              contexts: [0, 1, 2],
+              integration_types: [0],
+              options: [
+                {
+                  type: 3,
+                  name: 'prompt',
+                  description: 'The prompt to imagine',
+                  required: true
+                }
+              ]
+            },
+            attachments: []
+          }
+        }.to_json
+      end
+
+      def message_payload(message)
+        {
+          content: message,
+          flags: 0,
+          mobile_network_type: 'unknown',
+          tts: false
+        }.to_json
+      end
+
+      def headers
+        @headers ||= {
+          'Authorization' => token,
+          'Accept' => 'application/json',
+          'Content-Type' => 'application/json'
+        }
+      end
+    end
+  end
+end

data/lib/midjourney/api/imagine.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Midjourney
+  module API
+    #
+    # Midjourney Imagine Module
+    #
+    module Imagine
+      def imagine(prompt)
+        api.send_imagine(prompt)
+      end
+    end
+  end
+end

data/lib/midjourney/api/messages.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Midjourney
+  module API
+    #
+    # Midjourney Read Messages Module
+    #
+    module Messages
+      def read_messages(limit = 50)
+        api.read_messages(limit)
+      end
+
+      def send_message(message)
+        api.send_message(message)
+      end
+    end
+  end
+end

data/lib/midjourney/configuration.rb

@@ -1,9 +1,13 @@
 module Midjourney
+  #
+  # This class is used to store the configuration of the gem.
+  #
   class Configuration
-    attr_accessor :api_key
+    attr_accessor :discord_user_token, :discord_channel_id
 
     def initialize
-      @api_key = nil
+      @discord_user_token  = nil
+      @discord_channel_id = nil
     end
   end
 end

data/lib/midjourney.rb

@@ -1,17 +1,44 @@
 # frozen_string_literal: true
 
-require_relative "midjourney/configuration"
+require 'json'
+require 'pry-byebug'
+require 'rainbow'
+require 'rake'
+require 'rest-client'
 
+require_relative 'midjourney/configuration'
+require_relative 'midjourney/api/gateway'
+require_relative 'midjourney/api/imagine'
+require_relative 'midjourney/api/messages'
+
+#
+# Midjourney Ruby Client
+#
 module Midjourney
-  VERSION = "0.1.21"
+  VERSION = '0.2.0'
 
   class Error < StandardError; end
 
+  extend API::Imagine
+  extend API::Messages
+
   def self.configuration
-    @@configuration ||= Configuration.new
+    @configuration ||= Configuration.new
   end
 
   def self.config
     yield(configuration)
   end
+
+  def self.api
+    @api ||= API::Gateway.new(token: discord_user_token, channel_id: discord_channel_id)
+  end
+
+  def self.discord_user_token
+    configuration.discord_user_token
+  end
+
+  def self.discord_channel_id
+    configuration.discord_channel_id
+  end
 end

data/lib/tasks/install.rb

@@ -1,26 +1,32 @@
+# frozen_string_literal: true
+
 namespace :midjourney do
-  desc "Install Midjourney Ruby Client"
+  desc 'Install Midjourney Ruby Client'
   task install: :environment do
     return :ok unless defined?(Rails)
 
     initializer_content = <<~RUBY
       # frozen_string_literal: true
 
+      #
+      # Take a look at https://github.com/leom806/midjourney-ruby for more information.
+      #
       Midjourney.config do |config|
-        # Fetch from an environment variable
-        config.api_key = ENV["MIDJOURNEY_API_KEY"]
-        # Or fetch from Rails credentials if you're using Rails 5.2+ and have credentials set up
-        config.api_key = Rails.application.credentials.midjourney_api_key # or whatever you named your key
+        # Fetch from an environment variable or fetch from Rails credentials if
+        # you're using Rails 5.2+ and have credentials set up.
+
+        config.discord_user_token  = ENV['DISCORD_USER_TOKEN']
+        config.discord_channel_id = ENV['DISCORD_CHANNEL_ID']
       end
     RUBY
 
-    initializer_path = Rails.root.join("config", "initializers", "midjourney.rb")
+    initializer_path = Rails.root.join('config', 'initializers', 'midjourney.rb')
 
     if File.exist?(initializer_path)
-      puts "\n  " + Rainbow("Midjourney Ruby").purple + " is already installed! 🎉"
+      puts "\n #{Rainbow('Midjourney Ruby').purple} is already installed."
     else
-      File.open(initializer_path, "w") { |f| f.write(initializer_content) }
-      puts "\n  " + Rainbow("Midjourney Ruby").purple + " has been installed! 🎉"
+      File.open(initializer_path, 'w') { |f| f.write(initializer_content) }
+      puts "\n #{Rainbow('Midjourney Ruby').purple} has been installed! 🎉"
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: midjourney-ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.21
+  version: 0.2.0
 platform: ruby
 authors:
 - Leonardo Momente
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-09-29 00:00:00.000000000 Z
+date: 2023-10-01 00:00:00.000000000 Z
 dependencies: []
 description: This is a Ruby client for Midjourney that allows you to use its main
   products via their Discord API.
@@ -19,6 +19,9 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/midjourney.rb
+- lib/midjourney/api/gateway.rb
+- lib/midjourney/api/imagine.rb
+- lib/midjourney/api/messages.rb
 - lib/midjourney/configuration.rb
 - lib/tasks/install.rb
 homepage: https://github.com/leom806/midjourney-ruby