From c7c6956c72d36091c63482a71e6ee5055f431f6a Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Tue, 17 Dec 2019 23:47:23 +0800
Subject: [PATCH] first init

---
 Cargo.toml | 15 +++++++++
 README.md  | 11 ++++++-
 src/lib.rs | 94 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 119 insertions(+), 1 deletion(-)
 create mode 100644 Cargo.toml
 create mode 100644 src/lib.rs

diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..a1e6c88
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,15 @@
+[package]
+name = "dingtalk"
+version = "0.0.1"
+authors = ["Hatter Jiang <jht5945@gmail.com>"]
+edition = "2018"
+description = "DingTalk Util"
+readme = "README.md"
+license = "MIT"
+
+[dependencies]
+base64 = "0.11.0"
+rust-crypto = "0.2.36"
+reqwest = "0.9.22"
+urlencoding = "1.0.0"
+json = "0.11.14"
diff --git a/README.md b/README.md
index 6f75876..a79b2f5 100644
--- a/README.md
+++ b/README.md
@@ -1,3 +1,12 @@
 # dingtalk
 
-DingTalk util
\ No newline at end of file
+DingTalk util
+
+
+```rust
+pub fn main() {
+    let dt = DingTalk::new("<token>", "");
+    dt.send_text("Hello world!").ok();
+}
+```
+
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..ead9aa0
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,94 @@
+#[macro_use]
+extern crate json;
+
+use std::time::SystemTime;
+use std::io::{
+    Error,
+    ErrorKind,
+};
+use crypto::{
+    mac::{
+        Mac,
+        MacResult,
+    },
+    hmac::Hmac,
+    sha2::Sha256,
+};
+
+const CONTENT_TYPE: &str = "Content-Type";
+const APPLICATION_JSON_UTF8: &str = "application/json; charset=utf-8";
+
+const DEFAULT_DINGTALK_ROBOT_URL: &str = "https://oapi.dingtalk.com/robot/send?access_token=";
+
+pub struct DingTalk<'a> {
+    pub access_token: &'a str,
+    pub sec_token: &'a str,
+}
+
+impl <'a> DingTalk<'a> {
+
+    pub fn new(access_token: &'a str, sec_token: &'a str) -> Self {
+        DingTalk {
+            access_token: access_token,
+            sec_token: sec_token,
+        }
+    }
+
+    pub fn send_text(&self, text_message: &str) -> Result<(), Box<dyn std::error::Error>> {
+        self.send(&json::stringify(object!{
+            "msgtype" => "text",
+            "text" => object! {
+                "content" => text_message,
+            }
+        }))
+    }
+
+    pub fn send_markdown(&self, title: &str, text: &str) -> Result<(), Box<dyn std::error::Error>> {
+        self.send(&json::stringify(object!{
+            "msgtype" => "markdown",
+            "markdown" => object! {
+                "title" => title,
+                "text" => text,
+            }
+        }))
+    }
+
+    pub fn send(&self, json_message: &str) -> Result<(), Box<dyn std::error::Error>> {
+        let client = reqwest::Client::new();
+        let response = client.post(&self.generate_signed_url())
+              .header(CONTENT_TYPE, APPLICATION_JSON_UTF8)
+              .body(json_message.as_bytes().to_vec())
+              .send()?;
+
+        match response.status().as_u16() {
+            200_u16 => Ok(()),
+            _ => Err(Box::new(Error::new(ErrorKind::Other, format!("Unknown status: {}", response.status().as_u16())))),
+        }
+    }
+
+    pub fn generate_signed_url(&self) -> String {
+        let mut signed_url = String::with_capacity(1024);
+        signed_url.push_str(DEFAULT_DINGTALK_ROBOT_URL);
+        signed_url.push_str(&urlencoding::encode(self.access_token));
+
+        if self.sec_token != "" {
+            let timestamp = &format!("{}", SystemTime::now().duration_since(SystemTime::UNIX_EPOCH).unwrap().as_millis());
+            let timestamp_and_secret = &format!("{}\n{}", timestamp, self.sec_token);
+            let hmac_sha256 = base64::encode(calc_hmac_sha256(self.sec_token.as_bytes(), timestamp_and_secret.as_bytes()).code());
+
+            signed_url.push_str("&timestamp=");
+            signed_url.push_str(timestamp);
+            signed_url.push_str("&sign=");
+            signed_url.push_str(&urlencoding::encode(&hmac_sha256));
+        }
+        
+        signed_url
+    }
+}
+
+
+fn calc_hmac_sha256(key: &[u8], message: &[u8]) -> MacResult {
+    let mut hmac = Hmac::new(Sha256::new(), key);
+    hmac.input(message);
+    hmac.result()
+}