vagrant_cloud 2.0.1 → 3.0.2

data/lib/vagrant_cloud/box.rb

@@ -1,185 +1,146 @@
 module VagrantCloud
-  class Box
-    attr_accessor :account
-    attr_accessor :name
+  class Box < Data::Mutable
+    autoload :Provider, "vagrant_cloud/box/provider"
+    autoload :Version, "vagrant_cloud/box/version"
 
-    # @param [String] account
-    # @param [String] name
-    # @param [Hash] data
-    # @param [String] description
-    # @param [String] short_description
-    # @param [String] access_token
-    def initialize(account, name = nil, data = nil, short_description = nil, description = nil, access_token = nil, custom_server = nil)
-      @account = account
-      @name = name
-      @data = data
-      @description = description
-      @short_description = short_description
-      @client = Client.new(access_token, custom_server)
-    end
-
-    #--------------------
-    # Box API Helpers
-    #--------------------
+    attr_reader :organization
+    attr_required :name
+    attr_optional :created_at, :updated_at, :tag, :short_description,
+      :description_html, :description_markdown, :private, :downloads,
+      :current_version, :versions, :description, :username
 
-    # Read this box
-    # @return [Hash]
-    def data
-      @data ||= @client.request('get', box_path)
-    end
+    attr_mutable :short_description, :description, :private, :versions
 
-    # Update a box
+    # Create a new instance
     #
-    # @param [Hash] args
-    # @param [String] org - organization of the box to read
-    # @param [String] box_name - name of the box to read
-    # @return [Hash]
-    def update(args = {})
-      # hash arguments kept for backwards compatibility
-      data = @client.request('put', box_path(args[:organization], args[:name]), box: args)
-
-      # Update was called on *this* object, so update
-      # objects data locally
-      @data = data if !args[:organization] && !args[:name]
-      data
+    # @return [Box]
+    def initialize(organization:, **opts)
+      @organization = organization
+      @versions_loaded = false
+      opts[:username] = organization.username
+      super(opts)
+      if opts[:versions] && !opts[:versions].empty?
+        self.versions= Array(opts[:versions]).map do |version|
+          Box::Version.load(box: self, **version)
+        end
+      end
+      if opts[:current_version]
+        clean(data: {current_version: Box::Version.
+          load(box: self, **opts[:current_version])})
+      end
+      clean!
     end
 
-    # A generic function to read any box on Vagrant Cloud
-    # If org and box name is not supplied, it will default to
-    # reading the given Box object
+    # Delete this box
     #
-    # @param [String] org - organization of the box to read
-    # @param [String] box_name - name of the box to read
-    # @return [Hash]
-    def delete(org = nil, box_name = nil)
-      @client.request('delete', box_path(org, box_name))
+    # @return [nil]
+    # @note This will delete the box, and all versions
+    def delete
+      if exist?
+        organization.account.client.box_delete(
+          username: username,
+          name: name
+        )
+        b = organization.boxes.dup
+        b.delete(self)
+        organization.clean(data: {boxes: b})
+      end
+      nil
     end
 
-    # A generic function to read any box on Vagrant Cloud
+    # Add a new version of this box
     #
-    # @param [String] org - organization of the box to read
-    # @param [String] box_name - name of the box to read
-    # @return [Hash]
-    def read(org = nil, box_name = nil)
-      @client.request('get', box_path(org, box_name))
-    end
-
-    # @param [String] short_description
-    # @param [String] description
-    # @param [Bool] is_private
-    # @return [Hash]
-    def create(short_description = nil, description = nil, org = nil, box_name = nil, is_private = false)
-      update_data = !(org && box_name)
-
-      org ||= account.username
-      box_name ||= @name
-      short_description ||= @short_description
-      description ||= @description
-
-      params = {
-        name: box_name,
-        username: org,
-        is_private: is_private,
-        short_description: short_description,
-        description: description
-      }.delete_if { |_, v| v.nil? }
-
-      data = @client.request('post', '/boxes', box: params)
-
-      # Create was called on *this* object, so update
-      # objects data locally
-      @data = data if update_data
-      data
-    end
-
-    #--------------------
-    # Metadata Helpers
-    #--------------------
-
-    # @return [String]
-    def description
-      data['description_markdown'].to_s
+    # @param [String] version Version number
+    # @return [Version]
+    def add_version(version)
+      if versions.any? { |v| v.version == version }
+        raise Error::BoxError::VersionExistsError,
+          "Version #{version} already exists for box #{tag}"
+      end
+      v = Version.new(box: self, version: version)
+      clean(data: {versions: versions + [v]})
+      v
     end
 
-    # @return [String]
-    def description_short
-      data['short_description'].to_s
+    # Check if this instance is dirty
+    #
+    # @param [Boolean] deep Check nested instances
+    # @return [Boolean] instance is dirty
+    def dirty?(key=nil, deep: false)
+      if key
+        super(key)
+      else
+        d = super() || !exist?
+        if deep && !d
+          d = Array(plain_versions).any? { |v| v.dirty?(deep: true) }
+        end
+        d
+      end
     end
 
-    # @return [TrueClass, FalseClass]
-    def private
-      !!data['private']
+    # @return [Boolean] box exists remotely
+    def exist?
+      !!created_at
     end
 
     # @return [Array<Version>]
-    def versions
-      version_list = data['versions'].map { |data| VagrantCloud::Version.new(self, data['number'], data) }
-      version_list.sort_by { |version| Gem::Version.new(version.number) }
-    end
-
-    #------------------------
-    # Old Version API Helpers
-    #------------------------
-
-    # @param [Integer] number
-    # @param [Hash] data
-    # @return [Version]
-    def get_version(number, data = nil)
-      VagrantCloud::Version.new(self, number, data)
-    end
-
-    # @param [String] name
-    # @param [String] description
-    # @return [Version]
-    def create_version(name, description = nil)
-      params = { version: name }
-      params[:description] = description if description
-      data = @client.request('post', "#{box_path}/versions", version: params)
-      get_version(data['number'], data)
-    end
-
-    # @param [String] name
-    # @param [String] description
-    # @return [Version]
-    def ensure_version(name, description = nil)
-      version = versions.select { |v| v.version == name }.first
-      version ||= create_version(name, description)
-      if description && (description != version.description)
-        version.update(description)
+    # @note This is used to allow versions information to be loaded
+    # only when requested
+    def versions_on_demand
+      if !@versions_loaded
+        if exist?
+          r = self.organization.account.client.box_get(username: username, name: name)
+          v = Array(r[:versions]).map do |version|
+            Box::Version.load(box: self, **version)
+          end
+          clean(data: {versions: v + Array(plain_versions)})
+        else
+          clean(data: {versions: []})
+        end
+        @versions_loaded = true
       end
-      version
+      plain_versions
     end
+    alias_method :plain_versions, :versions
+    alias_method :versions, :versions_on_demand
 
-    # @param [Symbol]
-    # @return [String]
-    def param_name(param)
-      # This needs to return strings, otherwise it won't match the JSON that
-      # Vagrant Cloud returns.
-      ATTR_MAP.fetch(param, param.to_s)
+    # Save the box if any changes have been made
+    #
+    # @return [self]
+    def save
+      save_versions if dirty?(deep: true)
+      save_box if dirty?
+      self
     end
 
-    private
+    protected
 
-    # Constructs the box path based on an account and box name.
-    # If no params are given, it constructs a path for *this* Box object,
-    # but if both params are given it will construct a path for a one-off request
+    # Save the box
     #
-    # @param [String] - username
-    # @param [String] - box_name
-    # @return [String] - API path to box
-    def box_path(username = nil, box_name = nil)
-      if username && box_name
-        "/box/#{username}/#{box_name}"
+    # @return [self]
+    def save_box
+      req_args = {
+        username: username,
+        name: name,
+        short_description: short_description,
+        description: description,
+        is_private: self.private
+      }
+      if exist?
+        result = organization.account.client.box_update(**req_args)
       else
-        "/box/#{account.username}/#{name}"
+        result = organization.account.client.box_create(**req_args)
       end
+      clean(data: result, ignores: [:current_version, :versions])
+      self
     end
 
-    # Vagrant Cloud returns keys different from what you set for some params.
-    # Values in this map should be strings.
-    ATTR_MAP = {
-      is_private: 'private',
-      description: 'description_markdown'
-    }.freeze
+    # Save the versions if any require saving
+    #
+    # @return [self]
+    def save_versions
+      versions.map(&:save)
+      self
+    end
   end
 end

data/lib/vagrant_cloud/box/provider.rb

@@ -0,0 +1,175 @@
+module VagrantCloud
+  class Box
+    class Provider < Data::Mutable
+
+      # Result for upload requests to upload directly to the
+      # storage backend.
+      #
+      # @param [String] upload_url URL for uploading file asset
+      # @param [String] callback_url URL callback to PUT after successful upload
+      # @param [Proc] callback Callable proc to perform callback via configured client
+      DirectUpload = Struct.new(:upload_url, :callback_url, :callback, keyword_init: true)
+
+      attr_reader :version
+      attr_required :name
+      attr_optional :hosted, :created_at, :updated_at,
+        :checksum, :checksum_type, :original_url, :download_url,
+        :url
+
+      attr_mutable :url, :checksum, :checksum_type
+
+      def initialize(version:, **opts)
+        if !version.is_a?(Version)
+          raise TypeError, "Expecting type `#{Version.name}` but received `#{version.class.name}`"
+        end
+        @version = version
+        super(**opts)
+      end
+
+      # Delete this provider
+      #
+      # @return [nil]
+      def delete
+        if exist?
+          version.box.organization.account.client.box_version_provider_delete(
+            username: version.box.username,
+            name: version.box.name,
+            version: version.version,
+            provider: name
+          )
+          pv = version.providers.dup
+          pv.delete(self)
+          version.clean(data: {providers: pv})
+        end
+        nil
+      end
+
+      # Upload box file to be hosted on VagrantCloud. This
+      # method provides different behaviors based on the
+      # parameters passed. When the `direct` option is enabled
+      # the upload target will be directly to the backend
+      # storage. However, when the `direct` option is used the
+      # upload process becomes a two steps where a callback
+      # must be called after the upload is complete.
+      #
+      # If the path is provided, the file will be uploaded
+      # and the callback will be requested if the `direct`
+      # option is enabled.
+      #
+      # If a block is provided, the upload URL will be yielded
+      # to the block. If the `direct` option is set, the callback
+      # will be automatically requested after the block execution
+      # has completed.
+      #
+      # If no path or block is provided, the upload URL will
+      # be returned. If the `direct` option is set, the
+      # `DirectUpload` instance will be yielded and it is
+      # the caller's responsibility to issue the callback
+      #
+      # @param [String] path Path to asset
+      # @param [Boolean] direct Upload directly to backend storage
+      # @yieldparam [String] url URL to upload asset
+      # @return [self, Object, String, DirectUpload] self when path provided, result of yield when block provided, URL otherwise
+      # @note The callback request uses PUT request method
+      def upload(path: nil, direct: false)
+        if !exist?
+          raise Error::BoxError::ProviderNotFoundError,
+            "Provider #{name} not found for box #{version.box.tag} version #{version.version}"
+        end
+        if path && block_given?
+          raise ArgumentError,
+            "Only path or block may be provided, not both"
+        end
+        if path && !File.exist?(path)
+          raise Errno::ENOENT, path
+        end
+        req_args = {
+          username: version.box.username,
+          name: version.box.name,
+          version: version.version,
+          provider: name
+        }
+        if direct
+          r = version.box.organization.account.client.box_version_provider_upload_direct(**req_args)
+        else
+          r = version.box.organization.account.client.box_version_provider_upload(**req_args)
+        end
+        result = DirectUpload.new(
+          upload_url: r[:upload_path],
+          callback_url: r[:callback],
+          callback: proc {
+            if r[:callback]
+              version.box.organization.account.client.
+                request(method: :put, path: URI.parse(r[:callback]).path)
+            end
+          }
+        )
+        if block_given?
+          block_r = yield result.upload_url
+          result[:callback].call
+          block_r
+        elsif path
+          File.open(path, "rb") do |file|
+            chunks = lambda { file.read(Excon.defaults[:chunk_size]).to_s }
+            Excon.put(result.upload_url, request_block: chunks)
+          end
+          result[:callback].call
+          self
+        else
+          # When returning upload information for requester to complete,
+          # return upload URL when `direct` option is false, otherwise
+          # return the `DirectUpload` instance
+          direct ? result : result.upload_url
+        end
+      end
+
+      # @return [Boolean] provider exists remotely
+      def exist?
+        !!created_at
+      end
+
+      # Check if this instance is dirty
+      #
+      # @param [Boolean] deep Check nested instances
+      # @return [Boolean] instance is dirty
+      def dirty?(key=nil, **args)
+        if key
+          super(key)
+        else
+          super || !exist?
+        end
+      end
+
+      # Save the provider if any changes have been made
+      #
+      # @return [self]
+      def save
+        save_provider if dirty?
+        self
+      end
+
+      protected
+
+      # Save the provider
+      #
+      # @return [self]
+      def save_provider
+        req_args = {
+          username: version.box.username,
+          name: version.box.name,
+          version: version.version,
+          provider: name,
+          checksum: checksum,
+          checksum_type: checksum_type
+        }
+        if exist?
+          result = version.box.organization.account.client.box_version_provider_update(**req_args)
+        else
+          result = version.box.organization.account.client.box_version_provider_create(**req_args)
+        end
+        clean(data: result)
+        self
+      end
+    end
+  end
+end