gitlab-cloud-connector 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 17fe80dc098578ea54c50cf02e06e83bb516a4b18d6fbb649d9fd7d593e853c8
-  data.tar.gz: 89cbb45796d3eaa078a36e084b872f7b820f6fbf85fb745490cb164c52a66447
+  metadata.gz: 4d3a9c6cb24eae104dda40308b879b159d1e1583da6c39292dc864b3f9306746
+  data.tar.gz: 474c1fa8efc2fb0480818d74ac587152aa3ab907a5c340e3d5de7524c95fb3e2
 SHA512:
-  metadata.gz: 52f2b3ac0efb7d0aff0a360bf2272cdaf3151a8b912ce3d60503ac0cb2297f72809e879254751afeec188d7976c80b94f5f1fcd318792559f3460a7f32de8331
-  data.tar.gz: a4f104be9c1c99486d2b44a5183aba2c71d2f398cb4454557e27684e2669ac7c95394c6abe4321fb3f2e928ed59617a917886431fdf87c1e136255270bef8fa5
+  metadata.gz: f502784866d6d67fdbccbf262591ccc14d0a96fd44c6e38ab0de5273e224ce85258e0e504ded2d90f8d00b6f7a48662afae650a3dc8afe6de39ad8adcba37b26
+  data.tar.gz: ce85d65412bf2722bc68c5dcc2c8f16fa5da8c80147abef96c606dce03367aaaf053396671cf88281530aba9cce406440ef6b4ce11ce35a0379e3966c989edd9

data/README.md

@@ -10,8 +10,6 @@ We expect Bundler is used to manage dependencies. To add the dependency, add it
 gem "gitlab-cloud-connector", "~> 1.0.0", require: 'gitlab/cloud_connector'
 ```
 
-## Usage
-
 ### Data Storage
 
 Cloud Connector uses YAML files stored in the project's `/config` directory. The data is organized by model type:

data/lib/gitlab/cloud_connector/configuration.rb

@@ -1,21 +1,48 @@
 # frozen_string_literal: true
 
+require_relative 'data_model/yaml_data_loader'
+require_relative 'data_model/abstract_data_loader'
+
 module Gitlab
   module CloudConnector
     module Configuration
+      DEFAULT_DATA_LOADER_CLASS = Gitlab::CloudConnector::DataModel::YamlDataLoader
+      INVALID_LOADER_MESSAGE = "Data loader must inherit from Gitlab::CloudConnector::DataModel::AbstractDataLoader"
+
+      InvalidDataLoaderError = Class.new(StandardError)
+
       class << self
-        attr_writer :config_dir
+        attr_writer :config_dir, :memoize_data
 
         def config_dir
           @config_dir ||= File.expand_path("../../../config", __dir__).freeze
         end
 
+        def data_loader_class
+          @data_loader_class ||= DEFAULT_DATA_LOADER_CLASS
+        end
+
+        def data_loader_class=(klass)
+          validate_data_loader!(klass)
+          @data_loader_class = klass
+        end
+
+        def memoize_data?
+          return @memoize_data unless @memoize_data.nil?
+
+          (data_loader_class == DEFAULT_DATA_LOADER_CLASS)
+        end
+
         def configure
           yield self if block_given?
         end
 
-        def reset!
-          @config_dir = nil
+        private
+
+        def validate_data_loader!(klass)
+          return if klass && klass < Gitlab::CloudConnector::DataModel::AbstractDataLoader
+
+          raise InvalidDataLoaderError, INVALID_LOADER_MESSAGE
         end
       end
     end

data/lib/gitlab/cloud_connector/data_model/abstract_data_loader.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module CloudConnector
+    module DataModel
+      class AbstractDataLoader
+        def initialize(model_class)
+          @model_class = model_class
+        end
+
+        def load!
+          raise NotImplementedError, "#{self.class} must implement #load!"
+        end
+
+        protected
+
+        attr_reader :model_class
+      end
+    end
+  end
+end

data/lib/gitlab/cloud_connector/data_model/base.rb

@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 
 require_relative 'associations'
-require_relative 'data_loader'
 
 module Gitlab
   module CloudConnector
@@ -24,13 +23,17 @@ module Gitlab
           end
 
           def all
-            @all ||= data_loader.load!
+            if Configuration.memoize_data?
+              @all ||= data_loader.load!
+            else
+              data_loader.load!
+            end
           end
 
           private
 
           def data_loader
-            @data_loader ||= DataLoader.new(self)
+            Configuration.data_loader_class.new(self)
           end
         end
 
@@ -43,6 +46,13 @@ module Gitlab
         def [](name)
           instance_variable_get(:"@#{name}")
         end
+
+        def to_hash
+          instance_variables.each_with_object({}) do |var, hash|
+            key = var.to_s.delete('@').to_sym
+            hash[key] = instance_variable_get(var)
+          end
+        end
       end
     end
   end

data/lib/gitlab/cloud_connector/data_model/yaml_data_loader.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'yaml'
+require_relative 'abstract_data_loader'
+
+module Gitlab
+  module CloudConnector
+    module DataModel
+      class YamlDataLoader < AbstractDataLoader
+        def load!
+          Dir.glob(data_files_path).map { |file| load_model_from_file(file) }
+        end
+
+        private
+
+        def data_files_path
+          File.join(CloudConnector::Configuration.config_dir, model_class.model_name.tableize, '*.yml')
+        end
+
+        def load_model_from_file(file_path)
+          data = YAML.safe_load(
+            File.read(file_path),
+            permitted_classes: [Time],
+            symbolize_names: true
+          ).deep_symbolize_keys!
+
+          model_class.new(**data)
+        rescue StandardError => e
+          raise "Error loading file #{file_path}: #{e.message}"
+        end
+      end
+    end
+  end
+end

data/lib/gitlab/cloud_connector/version.rb

@@ -2,6 +2,6 @@
 
 module Gitlab
   module CloudConnector
-    VERSION = '1.0.0'
+    VERSION = '1.1.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gitlab-cloud-connector
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Nikola Milojevic
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-01-20 00:00:00.000000000 Z
+date: 2025-03-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -163,9 +163,11 @@ files:
 - lib/gitlab/cloud_connector.rb
 - lib/gitlab/cloud_connector/available_services_generator.rb
 - lib/gitlab/cloud_connector/configuration.rb
+- lib/gitlab/cloud_connector/data_model/abstract_data_loader.rb
 - lib/gitlab/cloud_connector/data_model/associations.rb
 - lib/gitlab/cloud_connector/data_model/base.rb
 - lib/gitlab/cloud_connector/data_model/data_loader.rb
+- lib/gitlab/cloud_connector/data_model/yaml_data_loader.rb
 - lib/gitlab/cloud_connector/data_models.rb
 - lib/gitlab/cloud_connector/json_web_token.rb
 - lib/gitlab/cloud_connector/version.rb