storazzo 0.7.0 → 0.7.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cd4df2335167b54bd0c1ef58279131c00d554e75a04411bea14b3ebb80417b93
-  data.tar.gz: 1aa197bfff96028744fa4e3f1d1dc48998c0b970b1413bb8fbfcdb3912e71ec9
+  metadata.gz: cf4a2d3f1f19159e98d91c9ffa590733482e5024d2fc773ddac89be26b7c3384
+  data.tar.gz: 81679032b930b55b6077ee2f4557ad1b00dab5684c702f25b79b8fb8131efd3f
 SHA512:
-  metadata.gz: 3aa715f7601839c5f1149b45932cbfc3fedddfbc664501ac8ed3a32f86f7a227a702a9a4825882b310772a20e4bf6503fddea9f6a47253a7906e91d2fa03773b
-  data.tar.gz: d4475efdfdb0f25e979d58140c573bf3bb347271d7df07d11a2cffe88a39d4dfda388532d03ef7d656d50a5b45418853c14c4ca80352b903149125cd6cd5e448
+  metadata.gz: 9ecccb88f2966188b3a54b53de353338810a7e578ab4ea73de4fa3279d3c6575aa05639e6fbbc70db27a6bb0501c76ab06e6748292e339ae8c0d7da60385d780
+  data.tar.gz: 75254ee445f4c73341361e627d75fdc284e0169571c3cff3a3dd6860be45857ee8af6482c512f42614b7d74373250b71a9d64427458a2de4fd4aac3f109275cc

data/Gemfile

@@ -21,4 +21,7 @@ group :test do
   gem 'minitest'
   gem 'minitest-reporters'
   gem 'mocha'
+  gem 'rspec'
+  gem 'cucumber'
+  gem 'aruba' # For testing CLI applications
 end

data/README.md

@@ -3,35 +3,33 @@
 
 📦 Storazzo 💎 gem - a Gem to automatically parse your FS for mounts (💽 💾 💿 ) and compute MD5 (🤐) of all files therein and then collect in central DB 🔋 through 📦 StorazzoApp📦 (TM).
 
+![Storazzo logo](assets/storazzo-logo.png)
+
 # INSTALL
 
 `gem install storazzo`
 
 (Latest version is hosted in https://rubygems.org/gems/storazzo)
 
-# Tests
-
-I still struggle to enforce the include of LOCAL unchecked code rather than latest required system gem (cmon Ruby!)
-but I found loads of interesting ways to test my code by googling and StoackOverflowing:
-
-* `rake test TEST="test/sum_test.rb"`
-* test-gcs-bucket: `ruby -I test test/test_gcs_bucket.rb` (meh - see below)
-* test-media-subfolder: `rake test TEST="test/media/*.rb"`
-
-Single test in single file:
+# Development & Testing
 
-* `rake test TEST="test/sum_test.rb" TESTOPTS="--name=test_returns_two"` (sample)
-* `rake test TEST="test/media/test_local_folder.rb" TESTOPTS="--name=test_1_first_directory_parsing_actually_works"`
-* `ruby -I test test/test_local_folder.rb -n test_first_directory_parsing_actually_works` (note this includes `storazzo` latest gem
-    and doesnt benefit from LATEST code so its NOT good for testing: use RAKE for that).
+To run the tools locally without installing the gem, use `just` (recommended) or standard Ruby flags.
 
-**Testing binary files** is hard: by default they 'require storazzo' and use the GEM INSTALLed version which is a few versions away, usually.
-So while developing you might want to include the lib/ folder, like this:
+### Using `just` (Recommended)
+This repository includes a `justfile` with common development commands:
+* `just setup` - Install dependencies
+* `just test` - Run all tests
+* `just run scan /path` - Run the local `storazzo` CLI
+* `just hello` - Run the local `hello-storazzo` script
 
-* Use local gem (super latest) for checking latest code: `ruby -Ilib bin/hello-storazzo`
-* This will use the gem installed a few days ago, likely so wont do you any good to test latest code: `bin/hello-storazzo`
+### Manual Local Execution
+If you don't have `just` installed, you can use `bundle exec` and the `-Ilib` flag:
+* **Run CLI**: `bundle exec ruby -Ilib bin/storazzo scan /path`
+* **Run Tests**: `bundle exec rake test`
+* **Single Test**: `bundle exec ruby -Ilib:test test/media/test_local_folder.rb`
 
-Now to toggle verbosity I believe I need to go into Rakefile (bummer)
+### Why `bundle exec` and `-Ilib`?
+Using `bundle exec` ensures all dependencies from the `Gemfile` are available. The `-Ilib` flag adds the local `lib/` directory to the Ruby load path, ensuring your local changes are used instead of any installed gem version.
 
 # Thanks
 

data/VERSION

@@ -1 +1 @@
-0.7.0
+0.7.4

data/bin/hello-storazzo

@@ -1,6 +1,8 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
+require 'bundler/setup'
+$LOAD_PATH.unshift(File.expand_path('../lib', __dir__))
 require 'storazzo'
 
 msg = ARGV.empty? ? 'HelloStorazzo' : ARGV.join(' ')

data/bin/ricdisk-magic

@@ -1,21 +1,15 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-############# ############# ############# ############# ############# ############# ############# #############
-# placeholder until i make it work..
-# https://guides.rubygems.org/make-your-own-gem/#adding-an-executable
-# require 'storazzo'
-# require_relative '../lib/storazzo'
+require 'bundler/setup'
+$LOAD_PATH.unshift(File.expand_path('../lib', __dir__))
+require 'storazzo'
 require 'fileutils'
 require 'yaml'
 require 'socket'
-require 'optparse' # http://ruby.about.com/od/advancedruby/a/optionparser.htm
+require 'optparse'
 
-# Including local gem.
-# puts File.expand_path(FileUtils.pwd, "/../")
-# local_gem_path = File.expand_path(FileUtils.pwd, "/../")
-# gem 'storazzo', path: local_gem_path
-require 'storazzo'
+# require 'storazzo'
 # include Storazzo
 # include Storazzo::Colors
 extend Storazzo::Colors
@@ -32,8 +26,8 @@ include Storazzo::Common # instead
 # include 'lib/ric_disk'
 # extend Storazzo::Colors
 
-if RUBY_VERSION.split('.')[0] == 1
-  puts 'Refusing to launch a script form Ruby 1. Sorry Ric, its 2020 damn it!'
+if RUBY_VERSION.split('.')[0].to_i < 3
+  puts 'Refusing to launch a script form Ruby 1 or 2. Sorry Ric, its 2020 damn it!'
   exit 2020
 end
 

data/bin/stats-with-md5

@@ -7,7 +7,7 @@ require 'optparse' # http://ruby.about.com/od/advancedruby/a/optionparser.htm
 require 'date' # for DateTime
 require 'tempfile'
 
-if RUBY_VERSION.split('.')[0] == 1
+if RUBY_VERSION.split('.')[0].to_i == 1
   puts 'Refusing to launch a script form Ruby 1. Sorry Ric, its 2020 damn it!'
   exit 2020
 end

data/bin/storazzo

@@ -1,6 +1,8 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
+require 'bundler/setup'
+$LOAD_PATH.unshift(File.expand_path('../lib', __dir__))
 require 'storazzo'
 require 'thor'
 

data/lib/storazzo/common.rb

@@ -28,7 +28,12 @@
 # 👶 🧒 👦 👧 👩 👨 🧑 🧔 🧓 👴 👵 👤 👥 👪 👫 👬 👭 👱 👳 👲 🧕 👸 🤴 🎅 🤶 disabled
 # 🧏 🦻 🦮 🦯 🦺 🦼 🦽 🦾 🦿 🤵 👮 👷 💁 💂 🕴 🕵 🦸 🦹 🧙 🧚 🧜 🧝 🧞 🧛 🧟 👼 👿 👻 👹 👺 👽 👾 🛸 💀 ☠ 🕱 🧠 🦴 👁 👀 👂 👃 👄 🗢 👅 🦷 🦵 🦶 💭 🗬 🗭 💬 🗨 🗩 🗪 🗫 🗰 🗱 🗮 🗯 🗣 🗤 🗥 🗦 🗧 💦 💧 💢 💫 💤 💨 💥 💪 🗲 🔥 💡 💩 💯 🔟 🔰 🕲
 require_relative 'colors'
-require 'pry'
+# require 'pry'
+begin
+  require 'pry'
+rescue LoadError
+  # Pry not available, no worries
+end
 
 module Storazzo
   module Common
@@ -52,8 +57,8 @@ module Storazzo
       puts "[W⚠️RN] #{azure(s)}"
     end
 
-    def err(_str)
-      puts "[ERR👎] #{red(s)}" # ⛔
+    def err(str)
+      puts "[ERR👎] #{red(str)}" # ⛔
     end
 
     def bug(s)
@@ -68,8 +73,12 @@ module Storazzo
     end
 
     def ppp(complex_object_to_colorize)
-      # TODO: i need to learn to return without printing..
-      Pry::ColorPrinter.pp(complex_object_to_colorize)
+      if defined?(Pry::ColorPrinter)
+        Pry::ColorPrinter.pp(complex_object_to_colorize)
+      else
+        require 'pp'
+        pp complex_object_to_colorize
+      end
     end
 
     def fatal(s)
@@ -84,6 +93,10 @@ module Storazzo
       `uname`.chomp == 'Linux'
     end
 
+    def slugify(string)
+      string.to_s.downcase.strip.gsub(/\s+/, '_').gsub(/[^\w-]/, '')
+    end
+
     private
 
     def _debug_true

data/lib/storazzo/gcs/client.rb

@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+require 'google/cloud/storage'
+require 'storazzo/common'
+
+module Storazzo
+  module GCS
+    class Client
+      include Storazzo::Common
+
+      attr_reader :storage, :project_id, :bucket_name
+
+      def initialize(project_id = nil, bucket_name = nil)
+        @project_id = project_id || autodetect_project_id
+        @bucket_name = bucket_name || ENV['GCS_BUCKET'] || "#{@project_id}-storazzo"
+        deb "GCS Client initialized for project: #{@project_id}, bucket: #{@bucket_name}"
+      end
+
+      def storage
+        @storage ||= Google::Cloud::Storage.new(project_id: @project_id)
+      end
+
+      def ensure_bucket_exists
+        return true if bucket_exists?(@bucket_name)
+
+        deb "Bucket #{@bucket_name} not found. Creating it in project #{@project_id}..."
+        storage.create_bucket(@bucket_name)
+        true
+      rescue StandardError => e
+        err "Failed to ensure/create bucket #{@bucket_name}: #{e.message}"
+        false
+      end
+
+      def autodetect_project_id
+        # Priority 1: Environment Variable
+        return ENV['GOOGLE_CLOUD_PROJECT'] if ENV['GOOGLE_CLOUD_PROJECT']
+        
+        # Priority 2: Config file
+        config = Storazzo::RicDiskConfig.instance
+        config.load
+        return config.project_id if config.project_id && config.project_id != 'YOUR-PROJECT-ID'
+
+        # Priority 3: GCloud CLI default
+        begin
+          `gcloud config get-value project 2>/dev/null`.strip
+        rescue StandardError
+          nil
+        end
+      end
+
+      def bucket_exists?(bucket_name)
+        bucket_name = bucket_name.gsub('gs://', '').split('/').first
+        !storage.bucket(bucket_name).nil?
+      rescue StandardError => e
+        warn "Error checking bucket #{bucket_name}: #{e.message}"
+        false
+      end
+
+      def list_buckets
+        storage.buckets.map(&:name)
+      end
+
+      def upload_file(local_path, bucket_name, remote_path)
+        bucket_name = bucket_name.gsub('gs://', '').split('/').first
+        bucket = storage.bucket(bucket_name)
+        raise "Bucket #{bucket_name} not found!" unless bucket
+
+        deb "Uploading #{local_path} to gs://#{bucket_name}/#{remote_path}..."
+        bucket.create_file(local_path, remote_path)
+      end
+
+      def download_file(bucket_name, remote_path, local_path)
+        bucket_name = bucket_name.gsub('gs://', '').split('/').first
+        bucket = storage.bucket(bucket_name)
+        raise "Bucket #{bucket_name} not found!" unless bucket
+
+        file = bucket.file(remote_path)
+        raise "Remote file #{remote_path} not found in bucket #{bucket_name}!" unless file
+
+        deb "Downloading gs://#{bucket_name}/#{remote_path} to #{local_path}..."
+        file.download(local_path)
+      end
+    end
+  end
+end

data/lib/storazzo/ric_disk.rb

@@ -4,6 +4,7 @@
 # it's considered interesting if there's a ".ricdisk/.ricdisk"
 
 require 'digest'
+require 'securerandom'
 
 module Storazzo
   class RicDisk
@@ -34,7 +35,7 @@ module Storazzo
       #     # todo substitute with protobuf..
       attr_accessor :name, :description, :ricdisk_file, :ricdisk_file_full, :local_mountpoint, :wr, :path,
                     :ricdisk_file_empty, :size, :active_dirs, :ricdisk_version,
-                    :unique_hash # new 202207
+                    :unique_hash, :disk_uuid, :llm_description, :llm_storage_type # new 202207
 
       ################################
       ## INSTANCE methods
@@ -53,7 +54,6 @@ module Storazzo
         # ok back to business, now path is a String :)
         path = ric_disk_object.path
         deb "RicDisk initialize.. path=#{path}"
-        deb "RicDisk initialize.. path=#{path}"
         @local_mountpoint = File.expand_path(path)
         @ard = ric_disk_object # AbstractRicDiskObject
         @description = "This is an automated RicDisk description from v.#{RicdiskVersion}. Created on #{Time.now}'"
@@ -62,21 +62,34 @@ module Storazzo
         @ricdisk_file_full = "#{@local_mountpoint}/#{@ricdisk_file}"
         @label = path.split('/').last
         @name = path.split('/').last
-        # @wr = File.writable?("#{path}/#{ricdisk_file}" ) # .writeable?
-        # @wr = writeable?
-        @tags = %w[ricdisk storazzo]
+        
+        load_existing_config
+
+        @tags ||= %w[ricdisk storazzo]
         @size = RicDisk._compute_size_could_take_long(path)
-        @unique_hash = "MD5::#{Digest::MD5.hexdigest(File.expand_path(path))}" #   hash = Digest::MD5.hexdigest(File.expand_path(get_local_mountpoint))
+        @unique_hash = "MD5::#{Digest::MD5.hexdigest(File.expand_path(path))}"
+        @disk_uuid ||= SecureRandom.uuid if defined?(SecureRandom)
         @computation_hostname = Socket.gethostname
-        @created_at = Time.now
+        @created_at ||= Time.now
 
         @ricdisk_file_empty = ricdisk_file_empty?
 
-        # @config = RicDiskConfig.instance.get_config
-        # #puts @config if @config
-        # find_info_from_mount(path)
         deb "RicDisk initialize. to_s: #{self}"
-        # find_info_from_df()
+      end
+
+      def load_existing_config
+        return unless File.exist?(@ricdisk_file_full) && !File.empty?(@ricdisk_file_full)
+        
+        begin
+          config = YAML.safe_load(File.read(@ricdisk_file_full))
+          @disk_uuid = config['disk_uuid']
+          @llm_description = config['llm_description']
+          @llm_storage_type = config['llm_storage_type']
+          @tags = config['tags']
+          @name = config['name'] || @name
+        rescue StandardError => e
+          warn "Error loading existing config from #{@ricdisk_file_full}: #{e.message}"
+        end
       end
 
       def ricdisk_file_empty?

data/lib/storazzo/search_engine.rb

@@ -3,9 +3,11 @@
 require 'sqlite3'
 require 'fileutils'
 require 'google/cloud/storage'
+require 'storazzo/common'
 
 module Storazzo
   class SearchEngine
+    include Storazzo::Common
     DB_PATH = File.expand_path("~/.storazzo_index.db")
 
     def initialize
@@ -15,50 +17,146 @@ module Storazzo
     end
 
     def create_tables
+      @db.execute <<-SQL
+        CREATE TABLE IF NOT EXISTS disks (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          name TEXT,
+          slug TEXT UNIQUE,
+          type TEXT,
+          uuid TEXT,
+          llm_description TEXT,
+          llm_storage_type TEXT,
+          last_scanned_at DATETIME
+        );
+      SQL
+
       @db.execute <<-SQL
         CREATE TABLE IF NOT EXISTS files (
           id INTEGER PRIMARY KEY AUTOINCREMENT,
           md5 VARCHAR(32),
           size INTEGER,
           path TEXT,
-          disk TEXT,
-          UNIQUE(disk, path)
+          disk_id INTEGER,
+          file_mtime DATETIME,
+          content_type TEXT,
+          ingested_at DATETIME,
+          FOREIGN KEY(disk_id) REFERENCES disks(id),
+          UNIQUE(disk_id, path)
         );
       SQL
     end
 
-    def sync_from_gcs
-      puts "Syncing metadata from GCS... (Stub)"
-      # Here we would use Google::Cloud::Storage to download `.rds` files
-      # from the designated GCS metadata bucket and then call `ingest_stats_file`.
+    def sync_all_from_gcs
+      client = Storazzo::GCS::Client.new
+      config = Storazzo::RicDiskConfig.instance
+      config.load
+      
+      buckets = config.get_bucket_paths
+      puts "Syncing metadata from #{buckets.size} buckets..."
+      
+      buckets.each do |bucket_url|
+        bucket_name = bucket_url.gsub('gs://', '').split('/').first
+        # For now, we search in the standard 'backup/ricdisk-magic/' path
+        prefix = "backup/ricdisk-magic/"
+        
+        begin
+          bucket = client.storage.bucket(bucket_name)
+          next unless bucket
+          
+          files = bucket.files(prefix: prefix)
+          rds_files = files.select { |f| f.name.end_with?('.rds') }
+          
+          puts "--- Bucket: gs://#{bucket_name} (#{rds_files.size} catalogs found) ---"
+          
+          rds_files.each do |remote_file|
+            # 1. Download to local tmp
+            local_tmp_path = File.join(Dir.tmpdir, File.basename(remote_file.name))
+            puts "Downloading #{remote_file.name}..."
+            remote_file.download(local_tmp_path)
+            
+            # 2. Ingest into SQLite
+            disk_name = File.basename(remote_file.name, '.rds').gsub('-ricdisk_stats_v11', '')
+            ingest_stats_file(local_tmp_path, disk_name)
+            
+            # 3. Cleanup
+            FileUtils.rm(local_tmp_path)
+          end
+        rescue => e
+          warn "Error syncing from gs://#{bucket_name}: #{e.message}"
+        end
+      end
     end
 
     def query(string)
-      @db.execute("SELECT * FROM files WHERE path LIKE ? OR disk LIKE ?", ["%#{string}%", "%#{string}%"])
+      @db.execute <<-SQL, ["%#{string}%", "%#{string}%", "%#{string}%"]
+        SELECT f.*, d.name as disk_name, d.slug as disk_slug 
+        FROM files f 
+        JOIN disks d ON f.disk_id = d.id
+        WHERE f.path LIKE ? OR d.name LIKE ? OR d.slug LIKE ?
+      SQL
     end
     
+    def find_or_create_disk(disk_name, opts = {})
+      slug = slugify(disk_name)
+      type = opts[:type] || 'local'
+      uuid = opts[:uuid]
+      llm_desc = opts[:llm_description]
+      llm_storage = opts[:llm_storage_type]
+      
+      @db.execute <<-SQL, [disk_name, slug, type, uuid, llm_desc, llm_storage]
+        INSERT OR IGNORE INTO disks (name, slug, type, uuid, llm_description, llm_storage_type) 
+        VALUES (?, ?, ?, ?, ?, ?)
+      SQL
+      
+      # Update existing record if new info is provided
+      if uuid || llm_desc || llm_storage
+        @db.execute <<-SQL, [uuid, llm_desc, llm_storage, slug]
+          UPDATE disks SET uuid = COALESCE(?, uuid), 
+                           llm_description = COALESCE(?, llm_description),
+                           llm_storage_type = COALESCE(?, llm_storage_type)
+          WHERE slug = ?
+        SQL
+      end
+
+      @db.get_first_value("SELECT id FROM disks WHERE slug = ?", [slug])
+    end
+
     def ingest_stats_file(file_path, disk_name)
-      File.readlines(file_path).each do |line|
-        next if line.start_with?('#') || line.strip.empty?
-        
-        # Example format:
-        # [file_v1.2] md5 mode type datetime size [content_type] filename
-        parts = line.split(' ')
-        
-        # very basic extraction trying to find the `[content_type]` bracket to locate the filename
-        content_type_idx = parts.find_index { |p| p.start_with?('[') && p.end_with?(']') && p != parts.first }
-        next unless content_type_idx
-        
-        md5 = parts[1]
-        size = parts[content_type_idx - 1].to_i
-        path = parts[(content_type_idx + 1)..-1].join(' ')
-        
-        begin
-          @db.execute("INSERT OR REPLACE INTO files (md5, size, path, disk) VALUES (?, ?, ?, ?)", [md5, size, path, disk_name])
-        rescue SQLite3::Exception => e
-          puts "Error inserting #{path}: #{e.message}"
+      disk_id = find_or_create_disk(disk_name)
+      ingested_at = Time.now.iso8601
+      
+      puts "Ingesting #{file_path} (disk_id: #{disk_id})..."
+      
+      @db.transaction do
+        File.readlines(file_path).each do |line|
+          next if line.start_with?('#') || line.strip.empty?
+          
+          # Example format:
+          # [file_v1.2] md5 mode type datetime size [content_type] filename
+          parts = line.split(' ')
+          
+          # Locating the `[content_type]` bracket
+          content_type_idx = parts.find_index { |p| p.start_with?('[') && p.end_with?(']') && p != parts.first }
+          next unless content_type_idx
+          
+          md5 = parts[1]
+          file_mtime = parts[4] # Standardized creation/mod time
+          size = parts[content_type_idx - 1].to_i
+          content_type = parts[content_type_idx].gsub(/[\[\]]/, '')
+          path = parts[(content_type_idx + 1)..-1].join(' ')
+          
+          begin
+            @db.execute <<-SQL, [md5, size, path, disk_id, file_mtime, content_type, ingested_at]
+              INSERT OR REPLACE INTO files (md5, size, path, disk_id, file_mtime, content_type, ingested_at) 
+              VALUES (?, ?, ?, ?, ?, ?, ?)
+            SQL
+          rescue SQLite3::Exception => e
+            puts "Error inserting #{path}: #{e.message}"
+          end
         end
       end
+      
+      @db.execute("UPDATE disks SET last_scanned_at = ? WHERE id = ?", [ingested_at, disk_id])
     end
   end
 end

data/lib/storazzo.rb

@@ -41,6 +41,7 @@ require 'storazzo/media/abstract_ric_disk'
 require 'storazzo/media/gcs_bucket'
 require 'storazzo/media/local_folder'
 require 'storazzo/media/mount_point'
+require 'storazzo/gcs/client'
 require 'storazzo/ric_disk_ugly'     # OLD and 90% working
 require 'storazzo/ric_disk_config' # => RicDiskConfif
 require 'storazzo/ric_disk_sample_config'   # => NOTHING!!

data/storazzo.gemspec

@@ -27,6 +27,10 @@ Gem::Specification.new do |s|
   #   "storazzo",
   #   "hello-storazzo",
   # ]
+  s.metadata = {
+    "github_repo" => "https://github.com/palladius/storazzo"
+  }
+
   s.homepage = 'https://rubygems.org/gems/storazzo' # maybe https://github.com/palladius/storazzo
   s.license = 'MIT'
 

data/test/gcs/test_client.rb

@@ -0,0 +1,42 @@
+require 'minitest/autorun'
+require 'mocha/minitest'
+require 'storazzo'
+require 'storazzo/gcs/client'
+
+class GcsClientTest < Minitest::Test
+  def setup
+    ENV['GOOGLE_CLOUD_PROJECT'] = 'test-project'
+    ENV['GCS_BUCKET'] = nil
+    @mock_storage = mock()
+    Google::Cloud::Storage.stubs(:new).returns(@mock_storage)
+  end
+
+  def test_default_bucket_name
+    client = Storazzo::GCS::Client.new('test-project')
+    assert_equal 'test-project-storazzo', client.bucket_name
+  end
+
+  def test_custom_bucket_name_from_env
+    ENV['GCS_BUCKET'] = 'my-custom-bucket'
+    client = Storazzo::GCS::Client.new('test-project')
+    assert_equal 'my-custom-bucket', client.bucket_name
+  end
+
+  def test_ensure_bucket_exists_when_it_does
+    client = Storazzo::GCS::Client.new('test-project')
+    mock_bucket = mock()
+    @mock_storage.expects(:bucket).with('test-project-storazzo').returns(mock_bucket)
+    
+    assert client.ensure_bucket_exists
+  end
+
+  def test_ensure_bucket_exists_when_it_does_not_creates_it
+    client = Storazzo::GCS::Client.new('test-project')
+    
+    # First check returns nil, then create_bucket is called
+    @mock_storage.expects(:bucket).with('test-project-storazzo').returns(nil)
+    @mock_storage.expects(:create_bucket).with('test-project-storazzo').returns(true)
+    
+    assert client.ensure_bucket_exists
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: storazzo
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.7.4
 platform: ruby
 authors:
 - Riccardo Carlesso
@@ -80,6 +80,7 @@ files:
 - lib/storazzo/colors.rb
 - lib/storazzo/common.rb
 - lib/storazzo/debug.rb
+- lib/storazzo/gcs/client.rb
 - lib/storazzo/hashify.rb
 - lib/storazzo/main.rb
 - lib/storazzo/media/README.md
@@ -100,6 +101,7 @@ files:
 - test/benchmark/test_hashing_functions-speed.rb
 - test/bin/new-idea.rb
 - test/bin/storazzo.rb
+- test/gcs/test_client.rb
 - test/media/test_abstract_ric_disk.rb
 - test/media/test_gcs_bucket.rb
 - test/media/test_local_folder.rb
@@ -118,7 +120,8 @@ files:
 homepage: https://rubygems.org/gems/storazzo
 licenses:
 - MIT
-metadata: {}
+metadata:
+  github_repo: https://github.com/palladius/storazzo
 rdoc_options: []
 require_paths:
 - lib
@@ -141,6 +144,7 @@ test_files:
 - test/benchmark/test_hashing_functions-speed.rb
 - test/bin/new-idea.rb
 - test/bin/storazzo.rb
+- test/gcs/test_client.rb
 - test/media/test_abstract_ric_disk.rb
 - test/media/test_gcs_bucket.rb
 - test/media/test_local_folder.rb