midjourney-ruby 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 727ce777e01b7a609b901f0eeb3b27c2f0568eb9f27c126a5e31391202afd8c4
-  data.tar.gz: dd12224e8cd69ceaf548221db086f3b009456aa594a30fe438786f111e0a0c42
+  metadata.gz: 2e648fd2e9826785943bab6a582fca4da21ae4358bf76cd80554b84f25e6df23
+  data.tar.gz: 2e3c6ad1067d655703e40e0b30b2a73c30e0812d9408f4303275253b8a62c68d
 SHA512:
-  metadata.gz: 93b06c02f6ece38ef9fa4847d89b540ea8a08c5e519867349be534028ec725442b39608a975b387d713d2a6d21dfc73e9ecd36139563f6b265e8189d7d12c5ae
-  data.tar.gz: 9047f70dd745f3a996ff9d7b74a8f4b6e470f7da664e97c83ab53765042d014ca0fda4964360c56f667e0c34d005ba8bf68923ae08492822063c145cbc94be93
+  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

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

data/lib/midjourney.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+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.2.0'
+
+  class Error < StandardError; end
+
+  extend API::Imagine
+  extend API::Messages
+
+  def self.configuration
+    @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

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+namespace :midjourney do
+  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 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')
+
+    if File.exist?(initializer_path)
+      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! 🎉"
+    end
+  end
+end

metadata

@@ -1,29 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: midjourney-ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Leonardo Momente
 autorequire:
-bindir: exe
+bindir: bin
 cert_chain: []
-date: 2023-09-29 00:00:00.000000000 Z
-dependencies:
-- !ruby/object:Gem::Dependency
-  name: rspec
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '3.2'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '3.2'
+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.
 email:
@@ -32,18 +18,17 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".rspec"
-- Gemfile
-- LICENSE.txt
-- README.md
-- Rakefile
-- lib/midjourney/ruby.rb
-- lib/midjourney/ruby/version.rb
-- sig/midjourney/ruby.rbs
+- 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
 licenses:
 - MIT
 metadata:
+  allowed_push_host: https://rubygems.org
   homepage_uri: https://github.com/leom806/midjourney-ruby
   source_code_uri: https://github.com/leom806/midjourney-ruby
   changelog_uri: https://github.com/leom806/midjourney-ruby/blob/main/CHANGELOG.md
@@ -65,5 +50,5 @@ requirements: []
 rubygems_version: 3.4.10
 signing_key:
 specification_version: 4
-summary: Ruby client for Midjourney.
+summary: Ruby client for Midjourney
 test_files: []

data/.rspec

@@ -1,3 +0,0 @@
---format documentation
---color
---require spec_helper

data/Gemfile

@@ -1,10 +0,0 @@
-# frozen_string_literal: true
-
-source "https://rubygems.org"
-
-# Specify your gem's dependencies in midjourney-ruby.gemspec
-gemspec
-
-gem "rake", "~> 13.0"
-
-gem "rspec", "~> 3.0"

data/LICENSE.txt

@@ -1,21 +0,0 @@
-The MIT License (MIT)
-
-Copyright (c) 2023 Leonardo Momente
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in
-all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-THE SOFTWARE.

data/README.md

@@ -1,37 +0,0 @@
-# Midjourney Ruby
-
-Midjourney is an awesome tool that brings great value to products.
-This gem was created to facilitate Ruby and also Rails applications to use all services of Midjourney directly.
-
-Keep in mind that you need to have a Midjourney account to use this gem, checkout [Midjourney](https://midjourney.com) for more information.
-
-## Installation
-
-Add the following to your Gemfile:
-
-```shell
-gem "midjourney-ruby"
-```
-
-Run `bundle install`.
-
-That's it. You are now ready to go!
-
-## Usage
-
-### Imagine
-
-## Development
-
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
-
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
-
-## Contributing
-
-Bug reports and pull requests are welcome on GitHub at [Open an Issue](https://github.com/leom806/midjourney-ruby).
-
-## License
-
-The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
-Feel free to use it and contribute.

data/Rakefile

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-
-RSpec::Core::RakeTask.new(:spec)
-
-task default: :spec

data/lib/midjourney/ruby/version.rb

@@ -1,7 +0,0 @@
-# frozen_string_literal: true
-
-module Midjourney
-  module Ruby
-    VERSION = "0.1.1"
-  end
-end

data/lib/midjourney/ruby.rb

@@ -1,10 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "ruby/version"
-
-module Midjourney
-  module Ruby
-    class Error < StandardError; end
-    # Your code goes here...
-  end
-end

data/sig/midjourney/ruby.rbs

@@ -1,6 +0,0 @@
-module Midjourney
-  module Ruby
-    VERSION: String
-    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
-  end
-end