Add signature helper to client

Jeff Lai
1 parent 32dd5f4b
Showing 1 changed file with 19 additions and 1 deletions Show diff stats
lib/dingtalk/client.rb
@@ -7,12 +7,17 @@ module Dingtalk
     end
  
     def decrypt(echo_str)
-      aes_key = Base64.decode64(Dingtalk.suite_aes_key + '=')
       content, status = Dingtalk::Prpcrypt.decrypt(aes_key, echo_str, Dingtalk.suite_key)
       # TODO check status
       JSON.parse(content)
     end
  
+    def signature(return_str)
+      encrypt = Dingtalk::Prpcrypt.encrypt(aes_key, return_str, Dingtalk.suite_key)
+      sort_params = [suite.suite_access_token, timestamp, nonce, encrypt].sort.join
+      Digest::SHA1.hexdigest(sort_params)
+    end
+
     def suite
       Api::Suite.new
     end
@@ -20,5 +25,18 @@ module Dingtalk
     def department
       Api::Department.new(@corp_id)
     end
+
+    private
+      def aes_key
+        Base64.decode64(Dingtalk.suite_aes_key + '=')
+      end
+
+      def timestamp
+        Time.now.to_i.to_s
+      end
+
+      def nonce
+        SecureRandom.hex
+      end
   end
 end
...	...	@@ -7,12 +7,17 @@ module Dingtalk
7	7	end
8	8
9	9	def decrypt(echo_str)
10		- aes_key = Base64.decode64(Dingtalk.suite_aes_key + '=')
11	10	content, status = Dingtalk::Prpcrypt.decrypt(aes_key, echo_str, Dingtalk.suite_key)
12	11	# TODO check status
13	12	JSON.parse(content)
14	13	end
15	14
	15	+ def signature(return_str)
	16	+ encrypt = Dingtalk::Prpcrypt.encrypt(aes_key, return_str, Dingtalk.suite_key)
	17	+ sort_params = [suite.suite_access_token, timestamp, nonce, encrypt].sort.join
	18	+ Digest::SHA1.hexdigest(sort_params)
	19	+ end
	20	+
16	21	def suite
17	22	Api::Suite.new
18	23	end
...	...	@@ -20,5 +25,18 @@ module Dingtalk
20	25	def department
21	26	Api::Department.new(@corp_id)
22	27	end
	28	+
	29	+ private
	30	+ def aes_key
	31	+ Base64.decode64(Dingtalk.suite_aes_key + '=')
	32	+ end
	33	+
	34	+ def timestamp
	35	+ Time.now.to_i.to_s
	36	+ end
	37	+
	38	+ def nonce
	39	+ SecureRandom.hex
	40	+ end
23	41	end
24	42	end
...	...