From 52171ac2788183816cd5e8b145ab4408a7310978 Mon Sep 17 00:00:00 2001
From: George Claghorn <george.claghorn@gmail.com>
Date: Mon, 3 Jul 2017 17:54:57 -0400
Subject: Add a Google Cloud Storage site

---
 lib/active_file/site.rb           |  1 +
 lib/active_file/sites/gcs_site.rb | 43 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+)
 create mode 100644 lib/active_file/sites/gcs_site.rb

(limited to 'lib')

diff --git a/lib/active_file/site.rb b/lib/active_file/site.rb
index e44c0145a9..3340c16620 100644
--- a/lib/active_file/site.rb
+++ b/lib/active_file/site.rb
@@ -41,4 +41,5 @@ module ActiveFile::Sites
 end
 
 require "active_file/sites/disk_site"
+require "active_file/sites/gcs_site"
 require "active_file/sites/s3_site"
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
-- 
cgit v1.2.3