skylab_studio 1.0.5 → 1.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 74d45652e950eb110f6d634681b5caad2830017e50da39a7610a3c2e148f5c8f
-  data.tar.gz: 71f434342a60a6e1604b5d2258c00f2766d571a4d81959a494c6d9bc7e07cacd
+  metadata.gz: 549e0ebc3cedf74a848a299346f93ce8b77c6d8813383f7993c82e755fbf4d88
+  data.tar.gz: 32825398b0428f26bf8548b40a4eae238d24b97709ce408b021c343901a99835
 SHA512:
-  metadata.gz: a2be823c425b7c9754d3c92e0777429fd5ad32bc795ac5dd1e1964f3299913fe527a805c856f6bab2eba787aafc5df26869126e95113a8b65b5acfd6320413d1
-  data.tar.gz: 7223682dfcb1368c03e493b57b53377ff7f9085d9ba0a71e7f40a85894e0a9d8e3efe916eb289ae9f89adc7b0949c88ac4d6ce23e09aa7855f9477d81c43f460
+  metadata.gz: af7021375c6030fcf977ff50fd85f1a55947ff7d6a0f4c122b0d8dd55ed4265dc5cccfcb020f04636cda76310910fd09fc1dbcda2b0477937b761185d36f7e63
+  data.tar.gz: c5cac2e61a900dd0460e477c6c8ecf5a83d2ae3e69fb0503b780e71550f3c79f25e67e83cc49fdfd6dd7663ac4e558da0521eea7e2bd9d9b17f52aa8f92baf5f

data/lib/skylab_studio/client.rb

@@ -1,4 +1,13 @@
+# frozen_string_literal: true
+
 require_relative 'error'
+require 'fileutils'
+require 'digest'
+require 'open-uri'
+require 'base64'
+require 'net/http'
+require 'vips'
+require 'concurrent'
 
 module SkylabStudio
   class ClientNilArgument < Error; end
@@ -25,109 +34,364 @@ module SkylabStudio
       @configuration = SkylabStudio::Config.new(settings)
     end
 
-    def list_jobs(options = {})
-      SkylabStudio::Request.new(@configuration).get(:jobs, payload: options)
+    def list_jobs()
+      SkylabStudio::Request.new(@configuration).get(:jobs)
     end
 
     def create_job(options = {})
-      validate_argument_presence options, :job
+      validate_argument_presence options, :name
+      validate_argument_presence options, :profile_id
 
-      SkylabStudio::Request.new(@configuration).post(:jobs, payload: options)
+      SkylabStudio::Request.new(@configuration).post(:jobs, options)
     end
 
-    def get_job(options = {})
-      validate_argument_presence options, :id
+    def get_job(job_id)
+      validate_argument_presence nil, :job_id
 
-      SkylabStudio::Request.new(@configuration).get("jobs/#{options[:id]}", payload: options)
+      SkylabStudio::Request.new(@configuration).get("jobs/#{job_id}")
     end
 
-    def update_job(options = {})
-      validate_argument_presence options, :id
-      validate_argument_presence options, :job
+    def get_job_by_name(options = {})
+      validate_argument_presence options, :name
 
-      SkylabStudio::Request.new(@configuration).patch("jobs/#{options[:id]}", payload: options)
+      SkylabStudio::Request.new(@configuration).get('jobs/find_by_name', options)
     end
 
-    def delete_job(options = {})
-      validate_argument_presence options, :id
+    def update_job(job_id, options = {})
+      validate_argument_presence nil, :job_id
 
-      SkylabStudio::Request.new(@configuration).delete("jobs/#{options[:id]}")
+      SkylabStudio::Request.new(@configuration).patch("jobs/#{job_id}", options)
     end
 
-    def process_job(options = {})
+    def queue_job(options = {})
       validate_argument_presence options, :id
 
-      SkylabStudio::Request.new(@configuration).post("jobs/#{options[:id]}/process", payload: options)
+      SkylabStudio::Request.new(@configuration).post("jobs/#{options[:id]}/queue", options)
     end
 
-    def cancel_job(options = {})
-      validate_argument_presence options, :id
+    def fetch_jobs_in_front(job_id)
+      SkylabStudio::Request.new(@configuration).get("jobs/#{job_id}/jobs_in_front")
+    end
+
+    def delete_job(job_id)
+      validate_argument_presence nil, :job_id
 
-      SkylabStudio::Request.new(@configuration).post("jobs/#{options[:id]}/cancel", payload: options)
+      SkylabStudio::Request.new(@configuration).delete("jobs/#{job_id}")
+    end
+
+    def cancel_job(job_id)
+      validate_argument_presence nil, :job_id
+
+      SkylabStudio::Request.new(@configuration).post("jobs/#{job_id}/cancel", nil)
     end
 
     def list_profiles(options = {})
-      SkylabStudio::Request.new(@configuration).get(:profiles, payload: options)
+      SkylabStudio::Request.new(@configuration).get(:profiles, options)
     end
 
     def create_profile(options = {})
-      validate_argument_presence options, :profile
+      validate_argument_presence options, :name
 
-      SkylabStudio::Request.new(@configuration).post(:profiles, payload: options)
+      SkylabStudio::Request.new(@configuration).post(:profiles, options)
     end
 
-    def get_profile(options = {})
-      validate_argument_presence options, :id
+    def get_profile(profile_id)
+      validate_argument_presence nil, :profile_id
 
-      SkylabStudio::Request.new(@configuration).get("profiles/#{options[:id]}", payload: options)
+      SkylabStudio::Request.new(@configuration).get("profiles/#{profile_id}")
     end
 
-    def update_profile(options = {})
-      validate_argument_presence options, :id
-      validate_argument_presence options, :profile
+    def update_profile(profile_id, options = {})
+      validate_argument_presence nil, :profile_id
 
-      SkylabStudio::Request.new(@configuration).patch("profiles/#{options[:id]}", payload: options)
+      SkylabStudio::Request.new(@configuration).patch("profiles/#{profile_id}", options)
     end
 
-    def delete_profile(options = {})
-      validate_argument_presence options, :id
+    def upload_job_photo(photo_path = nil, job_id = nil)
+      validate_argument_presence nil, job_id
+      validate_argument_presence nil, photo_path
 
-      SkylabStudio::Request.new(@configuration).delete("profiles/#{options[:id]}")
+      upload_photo(photo_path, job_id, 'job')
     end
 
-    def list_photos(options = {})
-      SkylabStudio::Request.new(@configuration).get(:photos, payload: options)
+    def upload_profile_photo(photo_path = nil, profile_id = nil)
+      validate_argument_presence nil, :photo_path
+      validate_argument_presence nil, :profile_id
+
+      upload_photo(photo_path, profile_id, 'profile')
     end
 
-    def create_photo(options = {})
-      validate_argument_presence options, :photo
+    def get_photo(photo_id)
+      validate_argument_presence nil, :photo_id
 
-      SkylabStudio::Request.new(@configuration).post(:photos, payload: options)
+      SkylabStudio::Request.new(@configuration).get("photos/#{photo_id}")
     end
 
-    def get_photo(options = {})
-      validate_argument_presence options, :id
-
-      SkylabStudio::Request.new(@configuration).get("photos/#{options[:id]}", payload: options)
+    def get_job_photos(job_id)
+      SkylabStudio::Request.new(@configuration).get('photos/list_for_job', { job_id: job_id })
     end
 
-    def update_photo(options = {})
-      validate_argument_presence options, :id
-      validate_argument_presence options, :photo
+    def delete_photo(photo_id)
+      validate_argument_presence nil, :photo_id
 
-      SkylabStudio::Request.new(@configuration).patch("photos/#{options[:id]}", payload: options)
+      SkylabStudio::Request.new(@configuration).delete("photos/#{photo_id}")
     end
 
-    def delete_photo(options = {})
-      validate_argument_presence options, :id
+    def download_photo(photo_id, output_path, profile: nil, options: {})
+      file_name = ''
+
+      unless Dir.exist?(output_path)
+        # Must be a file path - separate output_path and file_name
+        file_name = File.basename(output_path)
+        output_path = File.dirname(output_path) || ''
+      end
+
+      begin
+        photo = get_photo(photo_id)
+        profile_id = photo['job']['profileId']
+
+        file_name = photo['name'] if file_name.empty?
+
+        profile ||= get_profile(profile_id)
+
+        is_extract = profile['enableExtract'].to_s == 'true'
+        replace_background = profile['replaceBackground'].to_s == 'true'
+        is_dual_file_output = profile['dualFileOutput'].to_s == 'true'
+        profile['enableStripPngMetadata'].to_s
+        bgs = options[:bgs]
+
+        # Load output image
+        image_buffer = download_image_async(photo['retouchedUrl'])
+        image = Vips::Image.new_from_buffer(image_buffer, '')
+
+        if is_extract # Output extract image
+          png_file_name = "#{File.basename(file_name, '.*')}.png"
 
-      SkylabStudio::Request.new(@configuration).delete("photos/#{options[:id]}")
+          # Dual File Output will provide an image in the format specified in the outputFileType field
+          # and an extracted image as a PNG.
+          image.write_to_file(File.join(output_path, png_file_name)) if is_dual_file_output
+
+          download_replaced_background_image(file_name, image, output_path, profile: profile, bgs: bgs) if replace_background
+
+          # Regular Extract output
+          image.write_to_file(File.join(output_path, png_file_name)) unless is_dual_file_output || replace_background
+        else # Non-extracted regular image output
+          image.write_to_file(File.join(output_path, file_name))
+        end
+
+        puts "Successfully downloaded: #{file_name}"
+        [file_name, true]
+      rescue StandardError => e
+        error_msg = "Failed to download photo id: #{photo_id} - #{e}"
+        raise error_msg if options[:return_on_error].nil?
+
+        [file_name, false]
+      end
+    end
+
+    def download_all_photos(photos_list, profile, output_path)
+      raise 'Invalid output path' unless Dir.exist?(output_path)
+
+      success_photos = []
+      errored_photos = []
+      bgs = []
+
+      begin
+        profile = get_profile(profile['id'])
+        bgs = download_bg_images(profile) if profile && profile['photos']&.any?
+
+        photo_ids = photos_list.map { |photo| photo['id'].to_s }.compact
+
+        pool = Concurrent::FixedThreadPool.new(@configuration.settings[:max_download_concurrency])
+        download_tasks = []
+        photo_options = { return_on_error: true, bgs: bgs }
+        photo_ids.each do |photo_id|
+          download_tasks << Concurrent::Future.execute(executor: pool) do
+            download_photo(photo_id.to_i, output_path, profile: profile, options: photo_options)
+          end
+        end
+
+        # Wait for all download tasks to complete
+        results = download_tasks.map(&:value)
+        results.each do |result|
+          if result[1]
+            success_photos << result[0]
+          else
+            errored_photos << result[0]
+          end
+        end
+
+        { success_photos: success_photos, errored_photos: errored_photos }
+      rescue StandardError => e
+        warn e
+
+        { success_photos: success_photos, errored_photos: errored_photos }
+      end
     end
 
     private
 
+    def get_upload_url(options = { use_cache_upload: false })
+      SkylabStudio::Request.new(@configuration).get('photos/upload_url', options)
+    end
+
+    def create_photo(options = {})
+      SkylabStudio::Request.new(@configuration).post(:photos, options)
+    end
+
+    def upload_photo(photo_path, id, model = 'job')
+      valid_exts_to_check = %w[.jpg .jpeg .png .webp]
+
+      raise 'Invalid file type: must be of type jpg/jpeg/png/webp' unless valid_exts_to_check.any? { |ext| photo_path.downcase.end_with?(ext) }
+
+      file_size = File.size(photo_path)
+
+      raise 'Invalid file size: must be no larger than 27MB' if file_size > 27 * 1024 * 1024
+
+      photo_name = File.basename(photo_path)
+      photo_ext = File.extname(photo_name)
+      headers = {}
+      md5hash = ''
+
+      # Read file contents to binary
+      data = nil
+      File.open(photo_path, 'rb') do |file|
+        data = file.read
+        md5hash = Digest::MD5.hexdigest(data)
+      end
+
+      # model - either job or profile (job_id/profile_id)
+      photo_data = { "#{model}_id": id, name: "#{photo_name}#{photo_ext}", path: photo_path, 'use_cache_upload': false }
+
+      if model == 'job'
+        job_type = get_job(id)['type']
+
+        headers = { 'X-Amz-Tagging' => 'job=photo&api=true' } if job_type == 'regular'
+      end
+
+      # Ask studio to create the photo record
+      photo_response_json = create_photo(photo_data)
+
+      unless photo_response_json
+        raise 'Unable to create the photo object, if creating profile photo, ensure enable_extract and replace_background is set to: True'
+      end
+
+      photo_id = photo_response_json['id']
+
+      b64md5 = Base64.strict_encode64([md5hash].pack('H*'))
+      payload = {
+        'use_cache_upload' => false,
+        'photo_id' => photo_id,
+        'content_md5' => b64md5
+      }
+
+      # Ask studio for a presigned url
+      upload_url_resp = get_upload_url(payload)
+      upload_url = upload_url_resp['url']
+
+      # PUT request to presigned url with image data
+      headers['Content-MD5'] = b64md5
+
+      begin
+        uri = URI(upload_url)
+        request = Net::HTTP::Put.new(uri, headers)
+        request.body = data
+        upload_photo_resp = Net::HTTP.start(uri.hostname) { |http| http.request(request) }
+
+        unless upload_photo_resp
+          puts 'First upload attempt failed, retrying...'
+          retry_count = 0
+          # Retry upload
+
+          while retry_count < 3
+            upload_photo_resp = Net::HTTP.start(uri.hostname) { |http| http.request(request) }
+            if upload_photo_resp
+              break # Upload was successful, exit the loop
+            elsif retry_count == 2 # Check if retry count is 2 (0-based indexing)
+              raise 'Unable to upload to the bucket after retrying.'
+            else
+              sleep(1) # Wait for a moment before retrying
+              retry_count += 1
+            end
+          end
+        end
+      rescue StandardError => e
+        puts "An exception of type #{e.class} occurred: #{e.message}"
+        puts 'Deleting created, but unuploaded photo...'
+        delete_photo({ id: photo_id }) if photo_id
+      end
+
+      photo_response_json
+    end
+
+    def download_image_async(image_url)
+      raise "Invalid retouchedUrl: \"#{image_url}\" - Please ensure the job is complete" unless image_url.start_with?('http', 'https')
+
+      begin
+        uri = URI(image_url)
+        response = Net::HTTP.get_response(uri)
+
+        if response.is_a?(Net::HTTPSuccess)
+          response.body
+
+        else
+          puts "Error downloading image: #{response.message}"
+          nil
+        end
+      rescue StandardError => e
+        puts "Error downloading image: #{e.message}"
+        nil
+      end
+    end
+
+    def download_bg_images(profile)
+      temp_bgs = []
+
+      bg_photos = profile['photos'].select { |photo| photo['jobId'].nil? }
+
+      bg_photos.each do |bg|
+        bg_buffer = download_image_async(bg['originalUrl'])
+        bg_image = Vips::Image.new_from_buffer(bg_buffer, '')
+        temp_bgs << bg_image
+      end
+
+      temp_bgs
+    end
+
+    def download_replaced_background_image(file_name, input_image, output_path, profile: nil, bgs: nil)
+      output_file_type = profile&.[]('outputFileType') || 'png'
+
+      bgs = download_bg_images(profile) if bgs.nil? && !profile.nil? && profile&.[]('photos')&.any?
+
+      alpha_channel = input_image[3]
+      rgb_channel = input_image[0..2]
+      rgb_cutout = rgb_channel.bandjoin(alpha_channel)
+
+      if bgs&.any?
+        bgs.each_with_index do |bg_image, i|
+          new_file_name = if i.zero?
+                            "#{File.basename(file_name,
+                                             '.*')}.#{output_file_type}"
+                          else
+                            "#{File.basename(file_name,
+                                             '.*')} (#{i + 1}).#{output_file_type}"
+                          end
+          resized_bg_image = bg_image.thumbnail_image(input_image.width, height: input_image.height, crop: :centre)
+          result_image = resized_bg_image.composite2(rgb_cutout, :over)
+          result_image.write_to_file(File.join(output_path, new_file_name))
+        end
+      end
+
+      true
+    rescue StandardError => e
+      error_msg = "Error downloading background image: #{e.message}"
+      raise error_msg
+    end
+
     def validate_argument_presence(options, key)
-      raise SkylabStudio::ClientNilArgument, "#{key} cannot be nil" if options[key].nil?
+      raise SkylabStudio::ClientNilArgument, "#{key} cannot be nil" if options.is_a?(Hash) && options[key].nil?
+
+      raise SkylabStudio::ClientNilArgument, "#{key} cannot be nil" if key.nil?
     end
   end
 end

data/lib/skylab_studio/config.rb

@@ -1,10 +1,12 @@
+# frozen_string_literal: true
+
 require_relative 'version'
 
 module SkylabStudio
   class Config
     attr_accessor :settings
 
-    DEFAULT_URL = 'https://studio.skylabtech.ai'.freeze
+    DEFAULT_URL = 'https://studio.skylabtech.ai'
 
     def self.defaults
       source = URI.parse(DEFAULT_URL)
@@ -17,7 +19,8 @@ module SkylabStudio
         port: source.port,
         api_version: 'v1',
         debug: true,
-        client_stub: "ruby-#{VERSION}"
+        client_stub: "ruby-#{VERSION}",
+        max_download_concurrency: 5
       }
     end
 

data/lib/skylab_studio/error.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module SkylabStudio
   class Error < StandardError; end
 end

data/lib/skylab_studio/request.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require_relative 'error'
 
 module SkylabStudio
@@ -37,7 +39,7 @@ module SkylabStudio
     private
 
     def request_path(end_point)
-      "/api/#{@configuration.api_version}/#{end_point}"
+      "/api/public/#{@configuration.api_version}/#{end_point}"
     end
 
     def use_ssl?
@@ -64,19 +66,19 @@ module SkylabStudio
     end
 
     def handle_response(response)
-      case @response
-      when Net::HTTPUnauthorized then
+      case response
+      when Net::HTTPUnauthorized
         raise SkylabStudio::ClientInvalidKey, 'Invalid api key'
-      when Net::HTTPForbidden then
+      when Net::HTTPForbidden
         raise SkylabStudio::ClientInvalidKey, 'Invalid api key'
-      when Net::HTTPNotFound then
+      when Net::HTTPNotFound
         raise SkylabStudio::ClientInvalidEndpoint, 'Resource not found'
-      when Net::HTTPBadRequest then
+      when Net::HTTPBadRequest
         raise SkylabStudio::ClientBadRequest, 'There was an error processing your request'
-      when Net::HTTPTooManyRequests then
+      when Net::HTTPTooManyRequests
         raise SkylabStudio::ClientBadRequest, 'The rate limit has been met'
       when Net::HTTPSuccess
-        response
+        JSON.parse(response.body)
       else
         raise SkylabStudio::ClientUnknownError, 'An error has occurred'
       end

data/lib/skylab_studio/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module SkylabStudio
-  VERSION = '1.0.5'.freeze
+  VERSION = '1.0.7'
 end

data/lib/skylab_studio.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module SkylabStudio
 end
 
@@ -7,7 +9,7 @@ require 'net/https'
 require 'uri'
 require 'json'
 
-require 'skylab_studio/request'
-require 'skylab_studio/client'
-require 'skylab_studio/config'
-require 'skylab_studio/version' unless defined?(Skylab::VERSION)
+require_relative 'skylab_studio/request'
+require_relative 'skylab_studio/client'
+require_relative 'skylab_studio/config'
+require_relative 'skylab_studio/version' unless defined?(Skylab::VERSION)

data/spec/lib/skylab_studio/client_spec.rb

@@ -1,11 +1,11 @@
+# frozen_string_literal: true
+
 require File.expand_path('../../../lib/skylab_studio.rb', __dir__)
 
 RSpec.describe SkylabStudio::Client do
-  before(:each) do
-    @client = SkylabStudio::Client.new
-  end
+  let(:client) { SkylabStudio::Client.new }
 
-  subject { @client }
+  subject { client }
 
   it { should respond_to(:list_jobs) }
   it { should respond_to(:get_job) }
@@ -34,7 +34,7 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:get).and_return(true)
 
-      @client.list_jobs.should eq(true)
+      subject.list_jobs.should eq(true)
     end
   end
 
@@ -42,23 +42,19 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:post).and_return(true)
 
-      @client.create_job(
-        job: {
-          profile_id: 1
-        }
-      ).should eq(true)
+      subject.create_job({ name: 'test-job', profile_id: 1 }).should eq(true)
     end
   end
 
   describe '#get_job' do
     it 'should raise error with no id' do
-      expect { @client.get_job }.to raise_error(SkylabStudio::ClientNilArgument)
+      expect { subject.get_job }.to raise_error(ArgumentError)
     end
 
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:get).and_return(true)
 
-      @client.get_job(id: 123).should eq(true)
+      subject.get_job(123).should eq(true)
     end
   end
 
@@ -66,12 +62,7 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:patch).and_return(true)
 
-      @client.update_job(
-        id: 1,
-        job: {
-          profile_id: 2
-        }
-      ).should eq(true)
+      subject.update_job(1, { profile_id: 2 }).should eq(true)
     end
   end
 
@@ -79,17 +70,15 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:delete).and_return(true)
 
-      @client.delete_job(
-        id: 1
-      ).should eq(true)
+      subject.delete_job(1).should eq(true)
     end
   end
 
-  describe '#process_job' do
+  describe '#queue_job' do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:post).and_return(true)
 
-      @client.process_job(
+      subject.queue_job(
         id: 1
       ).should eq(true)
     end
@@ -99,9 +88,7 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:post).and_return(true)
 
-      @client.cancel_job(
-        id: 1
-      ).should eq(true)
+      subject.cancel_job(1).should eq(true)
     end
   end
 
@@ -109,7 +96,7 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:get).and_return(true)
 
-      @client.list_profiles.should eq(true)
+      subject.list_profiles.should eq(true)
     end
   end
 
@@ -117,8 +104,8 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:post).and_return(true)
 
-      @client.create_profile(
-        profile: {
+      subject.create_profile(
+        {
           name: 'Foo'
         }
       ).should eq(true)
@@ -127,13 +114,13 @@ RSpec.describe SkylabStudio::Client do
 
   describe '#get_profile' do
     it 'should raise error with no id' do
-      expect { @client.get_profile }.to raise_error(SkylabStudio::ClientNilArgument)
+      expect { subject.get_profile }.to raise_error(ArgumentError)
     end
 
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:get).and_return(true)
 
-      @client.get_profile(id: 123).should eq(true)
+      subject.get_profile(123).should eq(true)
     end
   end
 
@@ -141,67 +128,80 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:patch).and_return(true)
 
-      @client.update_profile(
-        id: 1,
-        profile: {
-          name: 'Bar'
-        }
-      ).should eq(true)
+      subject.update_profile(1, { name: 'Bar' } ).should eq(true)
     end
   end
 
-  describe '#delete_profile' do
-    it 'should return response' do
-      SkylabStudio::Request.any_instance.stub(:delete).and_return(true)
+  describe '#upload_job_photo' do
+    before do
+      stub_request(:post, 'https://studio.skylabtech.ai/api/public/v1/photos')
+        .to_return(status: 200, body: { id: 1, photo: {} }.to_json, headers: {})
 
-      @client.delete_profile(
-        id: 1
-      ).should eq(true)
+      stub_request(:get, 'https://studio.skylabtech.ai/api/public/v1/photos/upload_url')
+        .to_return(status: 200, body: { url: 'http://test.test/' }.to_json, headers: {})
+
+      stub_request(:get, 'https://studio.skylabtech.ai/api/public/v1/jobs/1')
+        .to_return(status: 200, body: {}.to_json, headers: {})
+
+      stub_request(:delete, 'https://studio.skylabtech.ai/api/public/v1/photos/1')
+        .to_return(status: 200, body: { id: 1 }.to_json, headers: {})
+
+      stub_request(:put, /.*/)
+        .to_return(status: 200, body: {}.to_json, headers: {})
+
+      # allow_any_instance_of(Net::HTTPSuccess).to receive(:body) { { id: 1 }.to_json }
     end
-  end
 
-  describe '#list_photos' do
     it 'should return response' do
-      SkylabStudio::Request.any_instance.stub(:get).and_return(true)
+      photo_path = "#{File.expand_path('../../', File.dirname(__FILE__))}/test-portrait-1.JPG"
+      id = 1
+
+      expected_response = { id: id, photo: {} }.to_json
 
-      @client.list_photos.should eq(true)
+      expect(subject.upload_job_photo(photo_path, id).to_json).to eq(expected_response)
     end
   end
 
-  describe '#create_photo' do
+  describe '#upload_profile_photo' do
+    before do
+      stub_request(:post, 'https://studio.skylabtech.ai/api/public/v1/photos')
+        .to_return(status: 200, body: { id: 1, photo: {} }.to_json, headers: {})
+
+      stub_request(:get, 'https://studio.skylabtech.ai/api/public/v1/photos/upload_url')
+        .to_return(status: 200, body: { url: 'http://test.test/' }.to_json, headers: {})
+
+      stub_request(:get, 'https://studio.skylabtech.ai/api/public/v1/profiles/upload_url')
+        .to_return(status: 200, body: { url: 'http://test.test/' }.to_json, headers: {})
+
+      stub_request(:get, 'https://studio.skylabtech.ai/api/public/v1/jobs/1')
+        .to_return(status: 200, body: {}.to_json, headers: {})
+
+      stub_request(:delete, 'https://studio.skylabtech.ai/api/public/v1/photos/1')
+        .to_return(status: 200, body: { id: 1 }.to_json, headers: {})
+
+      stub_request(:put, /.*/)
+        .to_return(status: 200, body: {}.to_json, headers: {})
+    end
+
     it 'should return response' do
-      SkylabStudio::Request.any_instance.stub(:post).and_return(true)
+      photo_path = "#{File.expand_path('../../', File.dirname(__FILE__))}/test-portrait-1.JPG"
+      id = 1
 
-      @client.create_photo(
-        photo: {
-          name: 'Foo'
-        }
-      ).should eq(true)
+      expected_response = { id: id, photo: {} }.to_json
+
+      expect(subject.upload_profile_photo(photo_path, 1).to_json).to eq(expected_response)
     end
   end
 
   describe '#get_photo' do
     it 'should raise error with no id' do
-      expect { @client.get_photo }.to raise_error(SkylabStudio::ClientNilArgument)
+      expect { subject.get_photo }.to raise_error(ArgumentError)
     end
 
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:get).and_return(true)
 
-      @client.get_photo(id: 123).should eq(true)
-    end
-  end
-
-  describe '#update_photo' do
-    it 'should return response' do
-      SkylabStudio::Request.any_instance.stub(:patch).and_return(true)
-
-      @client.update_photo(
-        id: 1,
-        photo: {
-          name: 'Bar'
-        }
-      ).should eq(true)
+      subject.get_photo(123).should eq(true)
     end
   end
 
@@ -209,9 +209,7 @@ RSpec.describe SkylabStudio::Client do
     it 'should return response' do
       SkylabStudio::Request.any_instance.stub(:delete).and_return(true)
 
-      @client.delete_photo(
-        id: 1
-      ).should eq(true)
+      subject.delete_photo(1).should eq(true)
     end
   end
 end

data/spec/lib/skylab_studio/config_spec.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require File.expand_path('../../../lib/skylab_studio.rb', __dir__)
 
 RSpec.describe SkylabStudio::Config do

data/spec/lib/skylab_studio/request_spec.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require File.expand_path('../../../lib/skylab_studio.rb', __dir__)
 
 RSpec.describe SkylabStudio::Request do
@@ -12,9 +14,10 @@ RSpec.describe SkylabStudio::Request do
 
   describe '#post' do
     it 'should return success' do
-      Net::HTTP.any_instance.stub(:request).and_return(Net::HTTPSuccess.new(1.0, 200, 'OK'))
+      allow_any_instance_of(Net::HTTP).to receive(:request) { Net::HTTPSuccess.new(1.0, 200, 'OK') }
+      allow_any_instance_of(Net::HTTPSuccess).to receive(:body) { {}.to_json }
 
-      @request.post(:jobs, {}).should be_instance_of(Net::HTTPSuccess)
+      @request.post(:jobs, {}).should be_instance_of(Hash)
     end
 
     it 'should raise error on 401' do
@@ -62,9 +65,10 @@ RSpec.describe SkylabStudio::Request do
 
   describe '#get' do
     it 'should return success' do
-      Net::HTTP.any_instance.stub(:request).and_return(Net::HTTPSuccess.new(1.0, 200, 'OK'))
+      allow_any_instance_of(Net::HTTP).to receive(:request) { Net::HTTPSuccess.new(1.0, 200, 'OK') }
+      allow_any_instance_of(Net::HTTPSuccess).to receive(:body) { {}.to_json }
 
-      @request.get(:jobs, {}).should be_instance_of(Net::HTTPSuccess)
+      @request.get(:jobs, {}).should be_instance_of(Hash)
     end
 
     it 'should raise error on 401' do
@@ -112,9 +116,10 @@ RSpec.describe SkylabStudio::Request do
 
   describe '#delete' do
     it 'should return success' do
-      Net::HTTP.any_instance.stub(:request).and_return(Net::HTTPSuccess.new(1.0, 200, 'OK'))
+      allow_any_instance_of(Net::HTTP).to receive(:request) { Net::HTTPSuccess.new(1.0, 200, 'OK') }
+      allow_any_instance_of(Net::HTTPSuccess).to receive(:body) { {}.to_json }
 
-      @request.delete(:jobs).should be_instance_of(Net::HTTPSuccess)
+      @request.delete(:jobs).should be_instance_of(Hash)
     end
 
     it 'should raise error on 401' do
@@ -162,9 +167,10 @@ RSpec.describe SkylabStudio::Request do
 
   describe '#put' do
     it 'should return success' do
-      Net::HTTP.any_instance.stub(:request).and_return(Net::HTTPSuccess.new(1.0, 200, 'OK'))
+      allow_any_instance_of(Net::HTTP).to receive(:request) { Net::HTTPSuccess.new(1.0, 200, 'OK') }
+      allow_any_instance_of(Net::HTTPSuccess).to receive(:body) { {}.to_json }
 
-      @request.put(:jobs, {}).should be_instance_of(Net::HTTPSuccess)
+      @request.put(:jobs, {}).should be_instance_of(Hash)
     end
 
     it 'should raise error on 401' do
@@ -212,9 +218,10 @@ RSpec.describe SkylabStudio::Request do
 
   describe '#patch' do
     it 'should return success' do
-      Net::HTTP.any_instance.stub(:request).and_return(Net::HTTPSuccess.new(1.0, 200, 'OK'))
+      allow_any_instance_of(Net::HTTP).to receive(:request) { Net::HTTPSuccess.new(1.0, 200, 'OK') }
+      allow_any_instance_of(Net::HTTPSuccess).to receive(:body) { {}.to_json }
 
-      @request.patch(:jobs, {}).should be_instance_of(Net::HTTPSuccess)
+      @request.patch(:jobs, {}).should be_instance_of(Hash)
     end
 
     it 'should raise error on 401' do

data/spec/lib/skylab_studio/version_spec.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 RSpec.describe SkylabStudio::VERSION do
   it 'provides the current version number' do
     SkylabStudio::VERSION.nil?.should eq(false)

data/spec/spec_helper.rb

@@ -1,4 +1,7 @@
+# frozen_string_literal: true
+
 require 'simplecov'
+require 'webmock/rspec'
 
 SimpleCov.start
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: skylab_studio
 version: !ruby/object:Gem::Version
-  version: 1.0.5
+  version: 1.0.7
 platform: ruby
 authors:
 - Paul Lam
-autorequire:
+autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-05-18 00:00:00.000000000 Z
+date: 2024-03-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -16,28 +16,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.8'
+        version: 3.12.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.8'
+        version: 3.12.0
 - !ruby/object:Gem::Dependency
   name: shoulda-matchers
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '='
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.1.2
+        version: 5.3.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '='
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.1.2
+        version: 5.3.0
 description: studio.skylabtech.ai ruby client
 email:
 - info@skylabtech.ai
@@ -60,7 +60,7 @@ homepage: https://github.com/skylab-tech/studio_client_ruby
 licenses:
 - Apache-2.0
 metadata: {}
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -68,7 +68,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.7.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -76,12 +76,12 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubygems_version: 3.1.6
-signing_key:
+signing_key: 
 specification_version: 4
 summary: A HTTP client for accessing studio.skylabtech.ai services.
 test_files:
 - spec/spec_helper.rb
-- spec/lib/skylab_studio/config_spec.rb
+- spec/lib/skylab_studio/request_spec.rb
 - spec/lib/skylab_studio/client_spec.rb
+- spec/lib/skylab_studio/config_spec.rb
 - spec/lib/skylab_studio/version_spec.rb
-- spec/lib/skylab_studio/request_spec.rb