Add aes_key to config & client shortcut

Jeff Lai
1 parent 9c6b3253
Showing 2 changed files with 29 additions and 1 deletions Show diff stats
lib/dingtalk/client.rb
lib/dingtalk/config.rb
@@ -0,0 +1,24 @@
+module Dingtalk
+  class Client
+    attr_accessor :corp_id
+
+    def initialize(corp_id)
+      @corp_id = corp_id
+    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 suite
+      Api::Suite.new
+    end
+
+    def department
+      Api::Department.new(@corp_id)
+    end
+  end
+end
@@ -18,9 +18,13 @@ module Dingtalk
     def suite_secret
       @suite_secret ||= config.suite_secret
     end
+
+    def suite_aes_key
+      @suite_aes_key ||= config.suite_aes_key
+    end
   end
   class Config
-    attr_accessor :redis, :redis_options, :suite_key, :suite_secret
+    attr_accessor :redis, :redis_options, :suite_key, :suite_secret, :suite_aes_key
   end
 end
@@ -0,0 +1,24 @@		@@ -0,0 +1,24 @@
	1	+module Dingtalk
	2	+ class Client
	3	+ attr_accessor :corp_id
	4	+
	5	+ def initialize(corp_id)
	6	+ @corp_id = corp_id
	7	+ end
	8	+
	9	+ def decrypt(echo_str)
	10	+ aes_key = Base64.decode64(Dingtalk.suite_aes_key + '=')
	11	+ content, status = Dingtalk::Prpcrypt.decrypt(aes_key, echo_str, Dingtalk.suite_key)
	12	+ # TODO check status
	13	+ JSON.parse(content)
	14	+ end
	15	+
	16	+ def suite
	17	+ Api::Suite.new
	18	+ end
	19	+
	20	+ def department
	21	+ Api::Department.new(@corp_id)
	22	+ end
	23	+ end
	24	+end