From 57a3e7757c8b98116b5f8ed9c5658663c1094701 Mon Sep 17 00:00:00 2001
From: el7cosmos <el@elabee.me>
Date: Thu, 17 Sep 2020 16:23:26 +0700
Subject: [PATCH] Attendance command

---
 .github/workflows/publish.yaml |  19 ++++++
 .github/workflows/release.yml  |   2 +-
 Cargo.lock                     |   2 +-
 Cargo.toml                     |   6 +-
 src/attendance/checkin.rs      |   5 ++
 src/attendance/checkout.rs     |   5 ++
 src/attendance/mod.rs          | 117 +++++++++++++++++++++++++++++++++
 src/client/mod.rs              |  62 ++++++++++++++++-
 src/config.rs                  |   6 +-
 src/main.rs                    |   6 ++
 10 files changed, 222 insertions(+), 8 deletions(-)
 create mode 100644 .github/workflows/publish.yaml
 create mode 100644 src/attendance/checkin.rs
 create mode 100644 src/attendance/checkout.rs
 create mode 100644 src/attendance/mod.rs

diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
new file mode 100644
index 0000000..e4d6fd4
--- /dev/null
+++ b/.github/workflows/publish.yaml
@@ -0,0 +1,19 @@
+name: Publish
+
+on:
+  push:
+    tags:
+      - '*'
+
+env:
+  CARGO_TERM_COLOR: always
+  CARGO_REGISTRY_TOKEN: ${{ secrets.CARGO_REGISTRY_TOKEN }}
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v2
+      - run: cargo publish
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 488262c..926412a 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -69,7 +69,7 @@ jobs:
       - name: Archive
         run: |
           cp target/${{ matrix.target }}/release/${{ matrix.bin }} .
-          tar czf talenta.tar.gz ${{ matrix.bin }} README.md
+          tar czf talenta.tar.gz ${{ matrix.bin }} LICENSE-APACHE LICENSE-MIT README.md
       - name: Upload Release Asset
         id: upload-release-asset
         uses: actions/upload-release-asset@v1
diff --git a/Cargo.lock b/Cargo.lock
index afc67b9..150d713 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -1124,7 +1124,7 @@ dependencies = [
 
 [[package]]
 name = "talenta"
-version = "0.1.0"
+version = "0.2.0"
 dependencies = [
  "ansi_term 0.12.1",
  "assert_cmd",
diff --git a/Cargo.toml b/Cargo.toml
index b9a17cd..d5017ff 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "talenta"
-version = "0.1.0"
+version = "0.2.0"
 authors = ["el7cosmos <el@elabee.me>"]
 edition = "2018"
 description = "Talenta CLI"
@@ -10,6 +10,10 @@ license = "MIT OR Apache-2.0"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
+exclude = [
+    ".github/*"
+]
+
 [dependencies]
 ansi_term = "0.12"
 chrono = "0.4"
diff --git a/src/attendance/checkin.rs b/src/attendance/checkin.rs
new file mode 100644
index 0000000..d1686ff
--- /dev/null
+++ b/src/attendance/checkin.rs
@@ -0,0 +1,5 @@
+use structopt::StructOpt;
+
+#[derive(Default, Debug, StructOpt)]
+#[structopt(about = "Request checkin only attendance")]
+pub(super) struct Checkin {}
diff --git a/src/attendance/checkout.rs b/src/attendance/checkout.rs
new file mode 100644
index 0000000..e2ff107
--- /dev/null
+++ b/src/attendance/checkout.rs
@@ -0,0 +1,5 @@
+use structopt::StructOpt;
+
+#[derive(Default, Debug, StructOpt)]
+#[structopt(about = "Request checkout only attendance")]
+pub(super) struct Checkout {}
diff --git a/src/attendance/mod.rs b/src/attendance/mod.rs
new file mode 100644
index 0000000..b8b384e
--- /dev/null
+++ b/src/attendance/mod.rs
@@ -0,0 +1,117 @@
+mod checkin;
+mod checkout;
+
+use crate::attendance::checkin::Checkin;
+use crate::attendance::checkout::Checkout;
+use crate::client::Client;
+use crate::config::Config;
+use crate::date::Date;
+use crate::time::Time;
+use crate::Command;
+use ansi_term::Colour;
+use dialoguer::theme::ColorfulTheme;
+use dialoguer::Input;
+use reqwest::StatusCode;
+use structopt::clap::{Error, ErrorKind};
+use structopt::StructOpt;
+
+#[derive(Debug, StructOpt)]
+enum AttendanceCommand {
+    Checkin(Checkin),
+    Checkout(Checkout),
+}
+
+#[derive(Default, StructOpt)]
+#[structopt(about = "Request attendance")]
+pub(super) struct Attendance {
+    #[structopt(skip)]
+    client: Client,
+    #[structopt(skip)]
+    theme: ColorfulTheme,
+
+    #[structopt(long)]
+    reason: Option<String>,
+    #[structopt(
+        default_value,
+        long,
+        help = "Effective date (yyyy-mm-dd)",
+        value_name = "DATE"
+    )]
+    date: Date,
+    #[structopt(long, value_name = "TIME")]
+    checkin_time: Option<Time>,
+    #[structopt(long, value_name = "TIME")]
+    checkout_time: Option<Time>,
+
+    #[structopt(subcommand)]
+    cmd: Option<AttendanceCommand>,
+}
+
+impl Command for Attendance {
+    fn run(&self) {
+        let config = Config::load().unwrap();
+        let token = match config.token() {
+            None => Error::with_description(
+                &format!(
+                    "Not logged in yet. Try: {}",
+                    Colour::Blue.bold().paint("talenta login")
+                ),
+                ErrorKind::ValueValidation,
+            )
+            .exit(),
+            Some(token) => token,
+        };
+
+        match self.cmd.as_ref() {
+            None => {
+                let checkin_time = self.checkin_time.unwrap_or_else(|| {
+                    Input::with_theme(&self.theme)
+                        .with_prompt("Checkin time (HH:mm)")
+                        .interact()
+                        .unwrap()
+                });
+
+                let checkout_time = self.checkout_time.unwrap_or_else(|| {
+                    Input::with_theme(&self.theme)
+                        .with_prompt("Checkout time (HH:mm)")
+                        .interact()
+                        .unwrap()
+                });
+
+                let reason = self.reason.clone().unwrap_or_else(|| {
+                    Input::with_theme(&self.theme)
+                        .with_prompt("Reason")
+                        .interact()
+                        .unwrap()
+                });
+
+                match self.client.attendance_request(
+                    token,
+                    &reason,
+                    self.date.into(),
+                    checkin_time.into(),
+                    checkout_time.into(),
+                ) {
+                    Ok(response) => {
+                        let status = StatusCode::from_u16(response.status()).unwrap();
+                        match status.is_success() {
+                            true => Attendance::success(response.message()),
+                            false => Error::with_description(
+                                &response.message(),
+                                ErrorKind::ValueValidation,
+                            )
+                            .exit(),
+                        }
+                    }
+                    Err(err) => {
+                        Error::with_description(&err.to_string(), ErrorKind::ValueValidation).exit()
+                    }
+                };
+            }
+            Some(cmd) => match cmd {
+                AttendanceCommand::Checkin(_checkin) => unimplemented!(),
+                AttendanceCommand::Checkout(_checkout) => unimplemented!(),
+            },
+        }
+    }
+}
diff --git a/src/client/mod.rs b/src/client/mod.rs
index d8893b4..ad0f8c9 100644
--- a/src/client/mod.rs
+++ b/src/client/mod.rs
@@ -1,6 +1,8 @@
+use chrono::{NaiveDate, NaiveTime};
 use reqwest::blocking;
-use serde::Deserialize;
+use serde::{Deserialize, Serialize};
 use std::collections::HashMap;
+use std::error::Error;
 use url::{ParseError, Url};
 
 #[derive(Deserialize, Debug)]
@@ -22,6 +24,9 @@ impl<T> ApiResponse<T> {
     }
 }
 
+#[derive(Deserialize, Debug)]
+pub(super) struct ResponseData {}
+
 #[derive(Deserialize, Debug)]
 pub(super) struct LoginData {
     token: String,
@@ -33,6 +38,19 @@ impl LoginData {
     }
 }
 
+#[allow(non_snake_case)]
+#[derive(Serialize, Debug)]
+struct AttendanceRequestBody {
+    datepicker_request_submit: String,
+    hour_checkin: Option<String>,
+    minute_checkin: Option<String>,
+    hour_checkout: Option<String>,
+    minute_checkout: Option<String>,
+    reason: String,
+    useCheckIn: bool,
+    useCheckOut: bool,
+}
+
 #[derive(Default, Debug)]
 pub(super) struct Client {
     client: blocking::Client,
@@ -43,7 +61,7 @@ impl Client {
         &self,
         email: &str,
         password: &str,
-    ) -> Result<ApiResponse<LoginData>, Box<dyn std::error::Error>> {
+    ) -> Result<ApiResponse<LoginData>, Box<dyn Error>> {
         let mut map = HashMap::new();
         map.insert("email", email);
         map.insert("password", password);
@@ -55,6 +73,46 @@ impl Client {
             .json()?)
     }
 
+    pub(super) fn attendance_request(
+        &self,
+        token: &str,
+        reason: &str,
+        date: NaiveDate,
+        checkin: Option<NaiveTime>,
+        checkout: Option<NaiveTime>,
+    ) -> Result<ApiResponse<ResponseData>, Box<dyn Error>> {
+        let json = AttendanceRequestBody {
+            datepicker_request_submit: date.to_string(),
+            hour_checkin: match checkin {
+                Some(time) => Some(time.format("%H").to_string()),
+                None => None,
+            },
+            minute_checkin: match checkin {
+                Some(time) => Some(time.format("%M").to_string()),
+                None => None,
+            },
+            hour_checkout: match checkout {
+                Some(time) => Some(time.format("%H").to_string()),
+                None => None,
+            },
+            minute_checkout: match checkout {
+                Some(time) => Some(time.format("%M").to_string()),
+                None => None,
+            },
+            reason: reason.into(),
+            useCheckIn: checkin.is_some(),
+            useCheckOut: checkout.is_some(),
+        };
+
+        Ok(self
+            .client
+            .post(Client::build_url("attendance-request")?)
+            .bearer_auth(token)
+            .json(&json)
+            .send()?
+            .json()?)
+    }
+
     fn build_url(path: &str) -> Result<Url, ParseError> {
         const BASE_URL: &str = "https://api-mobile.talenta.co/api/v1/";
         let base = Url::parse(BASE_URL).expect("hardcoded URL is known to be valid");
diff --git a/src/config.rs b/src/config.rs
index 1226d14..c624cce 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -5,7 +5,7 @@ const PKG_NAME: &str = env!("CARGO_PKG_NAME");
 
 #[derive(Default, Serialize, Deserialize, Debug)]
 pub(super) struct Config {
-    token: String,
+    token: Option<String>,
 }
 
 impl Config {
@@ -14,14 +14,14 @@ impl Config {
     }
 
     pub(super) fn with_token(token: String) -> Config {
-        Config { token }
+        Config { token: Some(token) }
     }
 
     pub(super) fn store(&self) -> Result<(), ConfyError> {
         confy::store(PKG_NAME, self)
     }
 
-    pub(super) fn token(&self) -> &str {
+    pub fn token(&self) -> &Option<String> {
         &self.token
     }
 }
diff --git a/src/main.rs b/src/main.rs
index b0dd3fe..0f93278 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,4 @@
+use crate::attendance::Attendance;
 use crate::login::Login;
 use ansi_term::Colour;
 use std::io::Write;
@@ -5,9 +6,12 @@ use std::{io, process};
 use structopt::clap::AppSettings;
 use structopt::StructOpt;
 
+mod attendance;
 mod client;
 mod config;
+mod date;
 mod login;
+mod time;
 
 trait Command {
     fn run(&self);
@@ -30,6 +34,7 @@ struct App {
 #[derive(StructOpt)]
 enum RootCommand {
     Login(Login),
+    Attendance(Attendance),
 }
 
 fn main() {
@@ -37,5 +42,6 @@ fn main() {
 
     match app.cmd {
         RootCommand::Login(login) => login.run(),
+        RootCommand::Attendance(attendance) => attendance.run(),
     }
 }