init project

lib/educoder/ufile.rb

@@ -0,0 +1,130 @@
+#coding=utf-8
+#
+# ucloud 文件上传
+#
+
+require 'base64'
+require 'openssl'
+require 'faraday'
+
+module Educoder
+  class Ufile
+      PATH_PREFIX = %r{^/}
+
+      def initialize(uploader={})
+        @ucloud_public_key             = uploader[:ucloud_public_key]
+        @ucloud_private_key            = uploader[:ucloud_private_key]
+        @ucloud_public_read            = uploader[:ucloud_public_read]
+        @ucloud_bucket                 = @ucloud_public_read ? uploader[:ucloud_public_bucket] : uploader[:ucloud_private_bucket]
+        @ucloud_bucket_host            = @ucloud_public_read ? uploader[:ucloud_public_bucket_host] : uploader[:ucloud_private_bucket_host]
+        @ucloud_cdn_host               = @ucloud_public_read ? uploader[:ucloud_public_cdn_host] : uploader[:ucloud_private_cdn_host]
+        @ucloud_private_expire_seconds = uploader[:ucloud_private_expire_seconds] || 300
+
+        unless @ucloud_cdn_host.include?('//')
+          raise "config.ucloud_cdn_host requirement include // http:// or https://, but you give: #{@ucloud_cdn_host}"
+        end
+      end
+
+      # 上传文件
+      def put(path, file, headers = {})
+        path.sub!(PATH_PREFIX, '')
+
+        response = conn.put(path, file.read) do |req|
+          req.headers = headers
+          token = authorization(req.method, headers['Content-Type'], path)
+          req.headers['Authorization'] = token
+        end
+
+        if response.success?
+          true
+        else
+          raise 'Ucloud上传失败'
+        end
+      end
+
+      # 读取文件
+      def get(path)
+        path.sub!(PATH_PREFIX, '')
+        response = conn.get(url(path))
+
+        if response.success?
+          return response
+        else
+          raise 'Ucloud Get File Fail'
+        end
+      end
+
+      # 删除文件
+      def delete(path)
+        path.sub!(PATH_PREFIX, '')
+        response = conn.delete(url(path)) do |req|
+          req.headers['Authorization'] = authorization(req.method, nil, path)
+        end
+
+        if response.success?
+          true
+        else
+          raise 'Ucloud Get File Fail'
+        end
+      end
+
+      def url(path)
+        if @ucloud_public_read
+          public_get_url(path)
+        else
+          private_get_url(path)
+        end
+      end
+
+      # 公开的访问地址
+      def public_get_url(path)
+        path.sub!(PATH_PREFIX, '')
+        [@ucloud_cdn_host, path].join('/')
+      end
+
+      # 私有空间访问地址
+      def private_get_url(path)
+        public_get_url(path) + privite_get_url_auth(path)
+      end
+
+      private
+
+      def conn
+        @conn ||= begin
+          Faraday.new(url: @ucloud_bucket_host) do |req|
+            req.request :url_encoded
+            req.adapter Faraday.default_adapter
+          end
+        end
+      end
+
+      # 私密查看url的认证信息
+      def privite_get_url_auth(path)
+        expired_ts = private_expire_ts
+        signed_str = signature(string_to_sign('GET', nil, path, expired_ts))
+        "?UCloudPublicKey=#{@ucloud_public_key}&Expires=#{expired_ts}&Signature=#{signed_str}"
+      end
+
+      def private_expire_ts
+        @ucloud_private_expire_seconds + Time.now.to_i
+      end
+
+      def authorization(http_method, content_type, path)
+        signed_str = signature(string_to_sign(http_method, content_type, path))
+        "UCloud " + @ucloud_public_key + ":" + signed_str
+      end
+
+      def signature(str)
+        Base64.strict_encode64(OpenSSL::HMAC.digest('sha1', @ucloud_private_key, str))
+      end
+
+      def string_to_sign(http_method, ori_content_type, path, expired_ts = nil)
+        http_verb = "#{http_method.upcase}\n"
+        content_md5 = "\n"
+        content_type = "#{ori_content_type}\n"
+        timestamp = "#{expired_ts}\n"
+        full_path = "/#{@ucloud_bucket}/#{path}"
+        http_verb + content_md5 + content_type + timestamp + full_path
+      end
+    end
+end