sheets_db 0.11.0 → 0.12.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cdf056137eb54d33c6647240ada1498ca2527d44fe68dd8f170882ecff233725
-  data.tar.gz: 61d76eff207e4d88bbad700ea9be592dbdbac4a2d709a1883c53dbeae055790e
+  metadata.gz: 8186e2c37fc147e8bcc4fed8c4bf288b05dde2466c9145e24882d7e1c41d3d35
+  data.tar.gz: cdaf52978086b08fed7cd9635aed44ae2f06bfb756058f1eb4d4fc0bf73e8945
 SHA512:
-  metadata.gz: a589ac3ae9d86b63b65691d12ef212323f1ea9266bb8e2230b67972781ba0ca5ec812f3befc0341f67261e16e70fb1511bce37103583ee452869fef1df7bc24b
-  data.tar.gz: a71d6cf51c4795e6d4185e27c37a0dadf51ea2ec8fdf55607586e510e3a437723f508a20ed7c52fb50373c98c08e8c0cfa74289b5b3da40414241a0079c520f0
+  metadata.gz: 89fd0245435dcfe13010330ca1220bc1d1ec1082333583021cb65f1d74857f6886eebf4325d7f1107ccdc3a406eb58877c914ef3bbdab8b9c4f1e98ffedb7e70
+  data.tar.gz: 2575c45b0f15e24cdbe8296d79a0429eb8120a41414cf112328106b650c70142df3312de2e1d86ffe63ae7c65e32f839bf413ee94c3ea647deae2d2eb692d77a

data/lib/sheets_db/collection.rb

@@ -2,17 +2,40 @@ module SheetsDB
   class Collection < Resource
     set_resource_type GoogleDrive::Collection
 
-    def self.has_many(resource, class_name:, resource_type: :subcollections)
-      unless [:subcollections, :spreadsheets].include?(resource_type)
-        raise ArgumentError, "resource_type must be :subcollections or :spreadsheets"
+    class << self
+      def has_many(resource, class_name:, resource_type: :subcollections)
+        unless [:subcollections, :spreadsheets].include?(resource_type)
+          raise ArgumentError, "resource_type must be :subcollections or :spreadsheets"
+        end
+        register_association(resource, class_name: class_name, resource_type: resource_type)
+        define_method(resource) do
+          @associated_resources ||= {}
+          @associated_resources[resource] ||= google_drive_resource.send(resource_type).map { |raw|
+            Support.constantize(class_name).new(raw)
+          }
+        end
       end
-      register_association(resource, class_name: class_name, resource_type: resource_type)
-      define_method(resource) do
-        result = instance_variable_get(:"@#{resource}")
-        result || instance_variable_set(:"@#{resource}",
-          google_drive_resource.send(resource_type).map { |raw| Support.constantize(class_name).new(raw) }
-        )
+    end
+
+    %i[
+      spreadsheet
+      subcollection
+    ].each do |child_resource_type|
+      define_method :"google_drive_#{child_resource_type}_by_title" do |title, **kwargs|
+        find_child_google_drive_resource_by(type: child_resource_type, title: title, **kwargs)
       end
     end
+
+    def spreadsheets
+      @anonymous_resources ||= {}
+      @anonymous_resources[:spreadsheets] ||=
+        google_drive_resource.spreadsheets.map { |raw| Spreadsheet.new(raw) }
+    end
+
+    def collections
+      @anonymous_resources ||= {}
+      @anonymous_resources[:collections] ||=
+        google_drive_resource.subcollections.map { |raw| Collection.new(raw) }
+    end
   end
-end
+end

data/lib/sheets_db/resource.rb

@@ -2,6 +2,7 @@ module SheetsDB
   class Resource
     class ResourceTypeMismatchError < StandardError; end
     class CollectionTypeAlreadyRegisteredError < StandardError; end
+    class ChildResourceNotFoundError < StandardError; end
 
     class << self
       attr_reader :resource_type
@@ -17,8 +18,8 @@ module SheetsDB
 
       def find_by_id(id, session: SheetsDB::Session.default)
         google_drive_resource = session.raw_file_by_id(id)
-        if @resource_type && !google_drive_resource.is_a?(@resource_type)
-          fail(ResourceTypeMismatchError, "The file with id #{id} is not a #{@resource_type}")
+        if resource_type && !google_drive_resource.is_a?(resource_type)
+          fail(ResourceTypeMismatchError, "The file with id #{id} is not a #{resource_type}")
         end
         new(google_drive_resource)
       end
@@ -26,10 +27,10 @@ module SheetsDB
       def belongs_to_many(resource, class_name:)
         register_association(resource, class_name: class_name, resource_type: :parents)
         define_method(resource) do
-          result = instance_variable_get(:"@#{resource}")
-          result || instance_variable_set(:"@#{resource}",
-            google_drive_resource.parents.map { |id| Support.constantize(class_name).find_by_id(id) }
-          )
+          @associated_resources ||= {}
+          @associated_resources[resource] ||= google_drive_resource.parents.map { |id|
+            Support.constantize(class_name).find_by_id(id)
+          }
         end
       end
 
@@ -43,6 +44,15 @@ module SheetsDB
           class_name: class_name
         }
       end
+
+      def association_methods_for_type(type)
+        raise ArgumentError unless %i[spreadsheet worksheet subcollection].include?(type)
+
+        google_type = type == :spreadsheet ? :file : type
+        create_prefix = type == :worksheet ? :add : :create
+
+        OpenStruct.new(find: :"#{google_type}_by_title", create: :"#{create_prefix}_#{type}")
+      end
     end
 
     extend Forwardable
@@ -56,6 +66,16 @@ module SheetsDB
       @google_drive_resource = google_drive_resource
     end
 
+    def reload!
+      @associated_resources = nil
+      @anonymous_resources = nil
+      google_drive_resource.reload_metadata
+    end
+
+    def delete!
+      google_drive_resource.delete
+    end
+
     def ==(other)
       other.is_a?(self.class) &&
         other.google_drive_resource == google_drive_resource
@@ -75,5 +95,14 @@ module SheetsDB
         updated_at: updated_at
       }
     end
+
+    def find_child_google_drive_resource_by(type:, title:, create: false)
+      association_methods = self.class.association_methods_for_type(type)
+      child_resource = google_drive_resource.public_send(association_methods.find, title)
+      child_resource ||= google_drive_resource.public_send(association_methods.create, title) if create
+      raise ChildResourceNotFoundError, [self, type, title] if child_resource.nil?
+
+      child_resource
+    end
   end
 end

data/lib/sheets_db/spreadsheet.rb

@@ -5,33 +5,26 @@ module SheetsDB
 
     set_resource_type GoogleDrive::Spreadsheet
 
-    def self.has_many(resource, worksheet_name:, class_name:)
-      register_worksheet_association(resource, worksheet_name: worksheet_name, class_name: class_name)
-      create_worksheet_association(resource, worksheet_name: worksheet_name, class_name: class_name)
-    end
+    class << self
+      def has_many(resource, worksheet_name:, class_name:)
+        register_worksheet_association(resource, worksheet_name: worksheet_name, class_name: class_name)
+        create_worksheet_association(resource, worksheet_name: worksheet_name, class_name: class_name)
+      end
 
-    def self.register_worksheet_association(resource, worksheet_name:, class_name:)
-      @associations ||= {}
-      if @associations.fetch(resource, nil)
-        raise WorksheetAssociationAlreadyRegisteredError
+      def register_worksheet_association(resource, worksheet_name:, class_name:)
+        @associations ||= {}
+        if @associations.fetch(resource, nil)
+          raise WorksheetAssociationAlreadyRegisteredError
+        end
+        @associations[resource] = {
+          worksheet_name: worksheet_name,
+          class_name: class_name
+        }
       end
-      @associations[resource] = {
-        worksheet_name: worksheet_name,
-        class_name: class_name
-      }
-    end
 
-    def self.create_worksheet_association(resource, worksheet_name:, class_name:)
-      define_method(resource) do
-        @worksheets ||= {}
-        @worksheets[resource] ||= begin
-          google_drive_worksheet = google_drive_resource.worksheet_by_title(worksheet_name)
-          raise WorksheetNotFoundError, worksheet_name if google_drive_worksheet.nil?
-          Worksheet.new(
-            spreadsheet: self,
-            google_drive_resource: google_drive_worksheet,
-            type: Support.constantize(class_name)
-          )
+      def create_worksheet_association(resource, **kwargs)
+        define_method(resource) do
+          worksheet_association(resource, **kwargs)
         end
       end
     end
@@ -51,5 +44,35 @@ module SheetsDB
     def select_from_association(association_name, &block)
       send(association_name).select(&block)
     end
+
+    def worksheet_association(association_name, worksheet_name:, class_name:)
+      @associated_resources ||= {}
+      @associated_resources[association_name] ||=
+        find_or_create_worksheet!(title: worksheet_name, type: Support.constantize(class_name))
+    end
+
+    def worksheets
+      @anonymous_resources ||= {}
+      @anonymous_resources[:worksheets] ||= google_drive_resource.worksheets.map { |raw|
+        set_up_worksheet!(google_drive_resource: raw)
+      }
+    end
+
+    def set_up_worksheet!(google_drive_resource:, type: nil)
+      Worksheet.new(
+        google_drive_resource: google_drive_resource,
+        spreadsheet: self,
+        type: type
+      ).set_up!
+    end
+
+    def find_or_create_worksheet!(title:, type: nil)
+      resource = google_drive_worksheet_by_title(title, create: true)
+      set_up_worksheet!(google_drive_resource: resource, type: type)
+    end
+
+    def google_drive_worksheet_by_title(title, **kwargs)
+      find_child_google_drive_resource_by(type: :worksheet, title: title, **kwargs)
+    end
   end
 end

data/lib/sheets_db/version.rb

@@ -1,3 +1,3 @@
 module SheetsDB
-  VERSION = "0.11.0"
+  VERSION = "0.12.0"
 end

data/lib/sheets_db/worksheet/row.rb

@@ -172,6 +172,7 @@ module SheetsDB
       def reload!
         worksheet.reload!
         reset_attributes_and_associations_cache
+        self
       end
 
       def save!
@@ -179,6 +180,7 @@ module SheetsDB
         worksheet.update_attributes_at_row_position(staged_attributes, row_position: row_position)
         save_changed_foreign_items!
         reset_attributes_and_associations_cache
+        self
       end
 
       def assign_next_row_position_if_not_set
@@ -192,6 +194,7 @@ module SheetsDB
       def reset_attributes_and_associations_cache
         @loaded_attributes = {}
         @loaded_associations = {}
+        self
       end
 
       def staged_attributes

data/lib/sheets_db/worksheet.rb

@@ -7,15 +7,34 @@ module SheetsDB
 
     include Enumerable
 
-    attr_reader :spreadsheet, :google_drive_resource, :type, :synchronizing
+    attr_reader :spreadsheet, :worksheet_title, :google_drive_resource, :synchronizing
 
-    def initialize(spreadsheet:, google_drive_resource:, type:)
+    def initialize(spreadsheet:, google_drive_resource:, type: nil)
       @spreadsheet = spreadsheet
       @google_drive_resource = google_drive_resource
+      @worksheet_title = google_drive_resource.title
       @type = type
       @synchronizing = true
     end
 
+    def set_up!
+      if google_drive_resource.num_rows == 0
+        registered_column_names = attribute_definitions.map { |name, definition| definition[:column_name] }
+        write_matrix!([registered_column_names])
+      end
+      self
+    end
+
+    def type
+      @type || @generated_type ||= begin
+        Class.new(Row).tap { |generated_type_class|
+          column_names.each do |name|
+            generated_type_class.attribute name.to_sym
+          end
+        }
+      end
+    end
+
     def ==(other)
       other.is_a?(self.class) &&
         other.google_drive_resource == google_drive_resource &&
@@ -31,7 +50,8 @@ module SheetsDB
     def columns
       @columns ||= begin
         {}.tap { |directory|
-          google_drive_resource.rows.first.each_with_index do |name, i|
+          header_row = google_drive_resource.rows.first || []
+          header_row.each_with_index do |name, i|
             unless name == ""
               directory[name] = Column.new(name: name, column_position: i + 1)
             end
@@ -98,12 +118,20 @@ module SheetsDB
 
     def update_attributes_at_row_position(staged_attributes, row_position:)
       staged_attributes.each do |attribute_name, value|
-        attribute_definition, column = get_definition_and_column(attribute_name)
-        raise ColumnNotFoundError, column unless column
-        assignment_value = attribute_definition[:multiple] ? value.join(",") : value
-        google_drive_resource[row_position, column.column_position] = assignment_value
+        update_attribute_at_row_position(
+          attribute_name: attribute_name,
+          value: value,
+          row_position: row_position
+        )
       end
-      google_drive_resource.synchronize if synchronizing
+      synchronize! if synchronizing
+    end
+
+    def update_attribute_at_row_position(attribute_name:, value:, row_position:)
+      attribute_definition, column = get_definition_and_column(attribute_name)
+      raise ColumnNotFoundError, column unless column
+      assignment_value = attribute_definition[:multiple] ? value.join(",") : value
+      google_drive_resource[row_position, column.column_position] = assignment_value
     end
 
     def next_available_row_position
@@ -163,7 +191,11 @@ module SheetsDB
     end
 
     def reload!
+      @columns = nil
+      @existing_raw_data = nil
+      @generated_type = nil
       google_drive_resource.reload
+      self
     end
 
     def convert_to_boolean(raw_value)
@@ -205,9 +237,64 @@ module SheetsDB
     def transaction
       disable_synchronization!
       yield
-      google_drive_resource.synchronize
+      synchronize!
     ensure
       enable_synchronization!
     end
+
+    def delete_google_drive_resource!
+      google_drive_resource.delete
+      @google_drive_resource = nil
+      spreadsheet.reload!
+    end
+
+    def truncate!
+      clear_google_drive_resource!
+      set_up!
+      reload!
+    end
+
+    def write_matrix(matrix, rewrite: false, save: false)
+      clear_google_drive_resource if rewrite
+      google_drive_resource.update_cells(1, 1, matrix)
+      synchronize! if save
+      self
+    end
+
+    def write_matrix!(matrix, **options)
+      write_matrix(matrix, **options.merge(save: true))
+    end
+
+    def write_raw_data(data, **options)
+      raise ArgumentError, "mismatched keys" if data.map(&:keys).uniq.count > 1
+
+      write_matrix(data.map(&:values).prepend(data.first.keys), **options)
+    end
+
+    def write_raw_data!(data, **options)
+      write_raw_data(data, **options.merge(save: true))
+    end
+
+    def clear_google_drive_resource(save: false)
+      empty_matrix = Array.new(google_drive_resource.num_rows, Array.new(google_drive_resource.num_cols))
+      write_matrix(empty_matrix, save: save)
+    end
+
+    def clear_google_drive_resource!
+      clear_google_drive_resource(save: true)
+    end
+
+    def synchronize!
+      google_drive_resource.synchronize
+      @existing_raw_data = nil
+      self
+    end
+
+    def existing_raw_data
+      @existing_raw_data ||= begin
+        rows = google_drive_resource.rows
+        rows.drop(1).map { |row| Hash[rows.first.zip(row)] }
+      end
+    end
   end
 end

data/sheets_db.gemspec

@@ -22,6 +22,6 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler", "~> 2.1"
   spec.add_development_dependency "rake", "~> 13"
-  spec.add_development_dependency "rspec", "~> 3.10"
+  spec.add_development_dependency "rspec", "~> 3.11"
   spec.add_development_dependency "simplecov"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sheets_db
 version: !ruby/object:Gem::Version
-  version: 0.11.0
+  version: 0.12.0
 platform: ruby
 authors:
 - Ravi Gadad
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-01-17 00:00:00.000000000 Z
+date: 2023-02-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: google_drive
@@ -58,14 +58,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.10'
+        version: '3.11'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.10'
+        version: '3.11'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement