struggle 2.2.2 → 2.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0de663c31ad25e4b730c910a5bb3a1d75fe00ac5
-  data.tar.gz: 1e85ed90acdac7bb95e1ed61eef1607999a24a3d
+  metadata.gz: 02424ee5858b52a0fb130dd9cf972d3482212d08
+  data.tar.gz: 1eddbe2e05520e2d9fe91408f04aa3021f5d5ab8
 SHA512:
-  metadata.gz: 4611e93533123a3025394be48eaf1d07e29478f8b97cb92f70e2efc3b4355ae73a38560c1722a3bc9ae0f117c7eabf7ab7d66ead4b790a02da140c3702dba00a
-  data.tar.gz: d5869ca27cd5feefad64df206681d0078a8d37b7e034e24ec8fccce7631c5577749aa3aecf525d8955a31481345ae369fb83a98123484d92ba39d16e419afefd
+  metadata.gz: 9a174385cfe42e9e2d34ef15f7ff2680ff30d6d6298013af8161e2cbfe90d91483f7a7318062aaf23bddaf99672216b2999cd62bfc422efc86d396292a3772b8
+  data.tar.gz: 512e145db165b6d2c25a0dc35566180132020a4b5e5ba8d275764a3c57c7445c5d734ee88ab3930079c6f4c57e6ce897a8d41999ed4fadab8dd4f75817b44191

data/lib/struggle/aes.rb

@@ -1,20 +1,76 @@
-module Struggle
-  class Aes
-    require 'openssl'
-    def self.encrypt(data, key ,iv)
-      cipher = OpenSSL::Cipher::AES.new(256, :CBC)
-      cipher.encrypt
-      cipher.key = key
-      cipher.iv = iv
-      Base64.encode64(cipher.update(data) + cipher.final)
+class Aes
+  # aes =
+  def initialize(key)
+    @@key = key
+  end
+
+  # 加密，key 秘钥， encrypted_string 需要加密的内容
+  def encrypt(encrypted_string)
+    aes = OpenSSL::Cipher::Cipher.new("AES-256-ECB").clone
+    aes.encrypt
+    aes.key = @@key
+    txt = aes.update(encrypted_string) << aes.final
+    txt.unpack('H*')[0].upcase
+  end
+
+  # 解密，key 秘钥， decrypted_string 需要解密的内容
+  def decrypt(decrypted_string)
+    aes = OpenSSL::Cipher::Cipher.new("AES-256-ECB").clone
+    aes.decrypt
+    aes.key = @@key
+    aes.update([decrypted_string].pack('H*')) << aes.final
+  end
+
+  # 文件内容加密并保存
+  def file_encrypt(filename)
+    if File.exist? filename
+      puts filename
+      content = File.read(filename)
+      unless content.blank?
+        File.write(filename, encrypt(Base64.encode64(content.force_encoding("UTF-8"))))
+      end
     end
+  end
 
-    def self.decrypt(data, key ,iv)
-      decipher = OpenSSL::Cipher::AES.new(256, :CBC)
-      decipher.decrypt
-      decipher.key = key
-      decipher.iv = iv
-      decipher.update(Base64.decode64(data)) + decipher.final
+  # 文件内容解密并保存
+  def file_decrypt(filename)
+    if File.exist? filename
+      puts filename
+      content = File.read(filename)
+      unless content.blank?
+        File.write(filename, Base64.decode64(decrypt(content)).force_encoding("UTF-8"))
+      end
     end
   end
-end
+
+  # 目录递归所有文件加密
+  def dir_encrypt(path)
+    get_file_list(path).each do |fn|
+      file_encrypt(fn)
+    end
+  end
+
+  # 目录递归所有文件解密
+  def dir_decrypt(path)
+    get_file_list(path).each do |fn|
+      file_decrypt(fn)
+    end
+  end
+
+  private
+  # 递归目录下所有文件
+  def get_file_list(path)
+    files = []
+    Dir.entries(path).each do |sub|
+      if sub != '.' && sub != '..'
+        if File.directory?("#{path}/#{sub}")
+          get_file_list("#{path}/#{sub}")
+        else
+          files << "#{path}/#{sub}"
+        end
+      end
+    end
+    files
+  end
+
+end

data/lib/struggle/getui.rb

@@ -0,0 +1,123 @@
+require 'digest'
+class Getui
+  class<<self
+    AppID = "XxESwIkfcwAKhgIcQ0iyl8"
+    AppKey = "9WY9yq0cxj5O7kPJT72om9"
+    AppSecret = "RkZ9pNxnXl7hHYG4NX1s19"
+    MasterSecret = "ie5ftISOpv97cgTBsiigaA"
+
+    # 签权，获取权限。返回bool值，成功签权返回true，否则返回false。 @@auth_token 全局token
+    def sign
+      timestamp = "#{Time.now.to_i}000"
+      sign = Digest::SHA256.hexdigest(AppKey+timestamp+MasterSecret)
+      # sign = Digest::SHA256.base64digest sign
+      command = <<EOF
+curl -H "Content-Type: application/json" \\
+    https://restapi.getui.com/v1/#{AppID}/auth_sign \\
+    -XPOST -d '{ "sign":"#{sign}",
+    "timestamp":"#{timestamp}",
+    "appkey":"#{AppKey}"
+    }'
+EOF
+      r = JSON.parse `#{command}`
+      r["result"] == "ok" ? @@auth_token = r["auth_token"] : nil
+    end
+
+    # 关闭签权，return bool，success true， error false
+    def close
+      command = <<EOF
+curl -H "authtoken: #{@@auth_token}" \
+    https://restapi.getui.com/v1/#{AppID}/auth_close \
+    -XPOST
+EOF
+      r = JSON.parse `#{command}`
+      r["result"] == "ok"
+    end
+
+    # 单推
+    # 参数 cid：client_id， title: 标题， text: 消息内容， is_offline是否离线默认否。
+    # 返回值 bool，success true， error false
+    def onePush(cid, title, text, is_offline=false)
+      if sign
+        requestid = Random.new.rand(1000000000000000000000..99999999999999999999999)
+        command = <<EOF
+curl -H "Content-Type: application/json" \
+    -H "authtoken:#{@@auth_token}" \
+     https://restapi.getui.com/v1/#{AppID}/push_single \
+     -XPOST -d '{
+                   "message": {
+                   "appkey": "#{AppKey}",
+                   "is_offline": #{is_offline},
+                   "offline_expire_time":100000000,
+                   "msgtype": "notification"
+                },
+                "notification": {
+                    "style": {
+                        "type": 0,
+                        "text": "#{text}",
+                        "title": "#{title}"
+                    },
+                    "transmission_type": true,
+                    "transmission_content": "透传内容"
+                },
+                "cid": "#{cid}",
+                "requestid": "#{requestid}"
+            }'
+EOF
+        r = JSON.parse `#{command}`
+        close
+        return r["result"] == "ok"
+      else
+        return false
+      end
+    end
+
+    # 多推
+    # 参数 clients: 客户信息数组格式[{cid: "xxx", title: "xxx", text: "xxx", is_offline: false}],
+    #              具体组成 cid：client_id， title: 标题， text: 消息内容， is_offline是否离线默认否。
+    # 返回值 bool，success true， error false
+    def manyPush(clients)
+      if sign
+        msg_list = []
+        clients.each do |c|
+          requestid = Random.new.rand(1000000000000000000000..99999999999999999999999)
+          msg_list << {
+              "message": {
+                  "appkey": "#{AppKey}",
+                  "is_offline": c[:is_offline] || false,
+                  "offline_expire_time": 100000000,
+                  "msgtype": "notification"
+              },
+              "notification": {
+                  "style": {
+                      "type": 0,
+                      "text": "#{c[:text]}",
+                      "title": "#{c[:title]}"
+                  },
+                  "transmission_type": true,
+                  "transmission_content": "透传内容"
+              },
+              "cid": "#{c[:cid]}",
+              "requestid": "#{requestid}"
+          }
+        end
+
+        command = <<EOF
+curl -H "Content-Type: application/json"  \
+     -H "authtoken:#{@@auth_token}" \
+     https://restapi.getui.com/v1/#{AppID}/push_single_batch  \
+     -XPOST -d '{
+                 "msg_list": #{msg_list.to_json},
+                 "need_detail":true
+            }'
+EOF
+        r = JSON.parse `#{command}`
+        close
+        return r["result"] == "ok"
+      else
+        return false
+      end
+    end
+
+  end
+end

data/lib/struggle/http.rb

@@ -3,31 +3,33 @@ module Struggle
     require 'net/http'
     require 'net/https'
     require 'uri'
-    def self.get(url, params=nil, header=nil)
-      uri = URI.parse(url)
-      http = Net::HTTP.new(uri.host, uri.port)
-      http.use_ssl = true if uri.scheme == 'https'
-      request = Net::HTTP::Get.new(uri.request_uri)
+
+    attr_reader :uri
+    attr_reader :request
+
+    def initialize(url)
+      @uri = URI.parse(url)
+      @http = Net::HTTP.new(@uri.host, @uri.port)
+      @http.use_ssl = true if @uri.scheme == 'https'
+      @http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+    end
+
+    def get(params=nil, header=nil)
+      @request = Net::HTTP::Get.new(@uri.request_uri, header)
       if !params.blank?
-        request.form_data = params
-      end
-      if !header.nil?
-        request.initialize_http_header(header)
+        @request.form_data = params
       end
-      http.request(request)
+      @http.request(@request)
     end
-    def self.post(url, params=nil, header=nil)
-      uri = URI.parse(url)
-      http = Net::HTTP.new(uri.host, uri.port)
-      http.use_ssl = true if uri.scheme == 'https'
-      request = Net::HTTP::Post.new(uri.request_uri)
+
+    def post(params=nil, header=nil)
+      @request = Net::HTTP::Post.new(@uri.request_uri, header)
       if !params.blank?
-        request.set_form_data(params)
-      end
-      if !header.nil?
-        request.initialize_http_header(header)
+        # @request.set_form_data(params)
+        @request.body = params
       end
-      http.request(request)
+      @http.request(@request)
     end
+
   end
 end

data/lib/struggle/rsa.rb

@@ -0,0 +1,43 @@
+# = RSA工具，基于openssl，功能包括，秘钥生成，加密解密
+class Rsa
+  # 实例化Rsa类，keyfile是秘钥文件地址（公钥或私钥）文件格式pem
+  def initialize(keyfile)
+    keystr = File.read(keyfile)
+    if keystr
+      @@rsa = OpenSSL::PKey::RSA.new(keystr)
+    end
+  end
+
+  # Rsa.generate("/Users/apple/object")       =>  /Users/apple/object/private_key.pem ,public_key.pem
+  # 创建2048位秘钥文件
+  def self.generate(dir)
+    pk = OpenSSL::PKey::RSA.generate(2048)
+    File.write(dir+"/private_key.pem", pk.to_pem)
+    File.write(dir+"/public_key.pem", pk.public_key.to_pem)
+  end
+
+  # 私钥加密
+  def private_encrypt(value)
+    Base64.encode64(@@rsa.private_encrypt(value.force_encoding("UTF-8"))) unless value.blank?
+  end
+
+  # 私钥解密
+  def private_decrypt(value)
+    @@rsa.private_decrypt(Base64.decode64(value.force_encoding("UTF-8"))) unless value.blank?
+  end
+
+  # 公钥加密
+  def public_encrypt(value)
+    Base64.encode64(@@rsa.public_encrypt(value.force_encoding("UTF-8"))) unless value.blank?
+  end
+
+  # 公钥解密
+  def public_decrypt(value)
+    @@rsa.public_decrypt(Base64.decode64(value.force_encoding("UTF-8"))) unless value.blank?
+  end
+
+  # 签名
+  def sign(value)
+    @@rsa.sign("sha1", value)
+  end
+end

data/lib/struggle/sms.rb

@@ -9,4 +9,17 @@ module Struggle
       return response.body=="100"
     end
   end
+  # 云讯科技
+  class SmsYx
+    URL = "http://sandbox.ytx.net"
+    require 'net/https'
+    require 'uri'
+
+    def Sms.send(accountSID, authToken, version, appid, tel, templateId, content)
+      url = "#{URL}/#{version}/sid/#{accountSID}/sms/TemplateSMS.wx"
+      data = {"action":"templateSms","mobile":tel,"appid":appid, "templateId":"#{templateId}","datas": content}
+      response = Http.post(URL, data)
+      return response.body
+    end
+  end
 end

data/lib/struggle.rb

@@ -14,6 +14,8 @@ require 'struggle/sms'
 require 'struggle/tmagick'
 require 'struggle/code'
 require 'struggle/aes'
+require 'struggle/rsa'
+require 'struggle/getui'
 require 'struggle/concerns/string_extend'
 require 'struggle/concerns/int_extend'
 require 'struggle/concerns/decimal_extend'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: struggle
 version: !ruby/object:Gem::Version
-  version: 2.2.2
+  version: 2.3.0
 platform: ruby
 authors:
 - lean
@@ -52,7 +52,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: code,tfile,pager,http,logistic,pay,tmagick,class_extend,aes,ftp_tool,zip_tool;
+description: code,tfile,pager,http,logistic,pay,tmagick,class_extend,aes,rsa,ftp_tool,zip_tool;
 email: 54850915@qq.com
 executables: []
 extensions: []
@@ -69,9 +69,11 @@ files:
 - lib/struggle/concerns/string_extend.rb
 - lib/struggle/font/font.ttf
 - lib/struggle/ftp_tool.rb
+- lib/struggle/getui.rb
 - lib/struggle/http.rb
 - lib/struggle/logistic.rb
 - lib/struggle/pager.rb
+- lib/struggle/rsa.rb
 - lib/struggle/sms.rb
 - lib/struggle/tfile.rb
 - lib/struggle/tmagick.rb