Add a Google Cloud Storage site

1 files changed, 43 insertions, 0 deletions

diff --git a/lib/active_file/sites/gcs_site.rb b/lib/active_file/sites/gcs_site.rb
new file mode 100644
index 0000000000..d9164621c2
--- /dev/null
+++ b/lib/active_file/sites/gcs_site.rb
@@ -0,0 +1,43 @@
+require "google/cloud/storage"
+
+class ActiveFile::Sites::GCSSite < ActiveFile::Site
+  attr_reader :client, :bucket
+
+  def initialize(project:, keyfile:, bucket:)
+    @client = Google::Cloud::Storage.new(project: project, keyfile: keyfile)
+    @bucket = @client.bucket(bucket)
+  end
+
+  def upload(key, data)
+    bucket.create_file(data, key)
+  end
+
+  def download(key)
+    io = file_for(key).download
+    io.rewind
+    io.read
+  end
+
+  def delete(key)
+    file_for(key).try(:delete)
+  end
+
+  def exist?(key)
+    file_for(key).present?
+  end
+
+
+  def byte_size(key)
+    file_for(key).size
+  end
+
+  def checksum(key)
+    file_for(key).md5.unpack("m0").first.unpack("H*").first
+  end
+
+
+  private
+    def file_for(key)
+      bucket.file(key)
+    end
+end