From d8a5542abcfc2b43934e040781c83d48efd2c622 Mon Sep 17 00:00:00 2001
From: Edward Shen <code@eddie.sh>
Date: Mon, 4 May 2020 19:47:27 -0400
Subject: [PATCH] move things into aux file

---
 .gitignore                    |  3 +-
 aux/filter_raw_character_data | 76 +++++++++++++++++++++++++++++++++++
 aux/get_recruit_data          |  8 ++++
 aux/init_db                   | 32 +++++++++++++++
 4 files changed, 118 insertions(+), 1 deletion(-)
 create mode 100755 aux/filter_raw_character_data
 create mode 100755 aux/get_recruit_data
 create mode 100755 aux/init_db

diff --git a/.gitignore b/.gitignore
index 404d39d..67151fc 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,4 +2,5 @@
 *.sqlite
 *.sqlite-shm
 *.sqlite-wal
-.env
\ No newline at end of file
+.env
+aux/**.json
\ No newline at end of file
diff --git a/aux/filter_raw_character_data b/aux/filter_raw_character_data
new file mode 100755
index 0000000..3eeeae3
--- /dev/null
+++ b/aux/filter_raw_character_data
@@ -0,0 +1,76 @@
+#!/usr/bin/env python
+
+import json
+import sys
+
+raw_json = json.load(sys.stdin)
+
+tags = {
+  1: "Guard",
+  2: "Sniper",
+  3: "Defender",
+  4: "Medic",
+  5: "Supporter",
+  6: "Caster",
+  7: "Specialist",
+  8: "Vanguard",
+  9: "Melee",
+  10: "Ranged",
+  11: "Top Operator",
+  12: "Crowd-Control",
+  13: "Nuker",
+  14: "Senior Operator",
+  15: "Healing",
+  16: "Support",
+  17: "Starter",
+  18: "DP-Recovery",
+  19: "DPS",
+  20: "Survival",
+  21: "AoE",
+  22: "Defense",
+  23: "Slow",
+  24: "Debuff",
+  25: "Fast-Redeploy",
+  26: "Shift",
+  27: "Summon",
+  28: "Robot"
+}
+
+char_data = {}
+en_recruitable = {}
+cn_recruitable = {}
+for key, entry in raw_json.items():
+  parsed_data = {
+    "stars": entry["star"],
+    "tags": entry["tags"],
+    # 0 = headhunt only, 1 = china only, 2 = global
+    "recruitment": min(len(entry["recruitment"]), 2)
+  }
+  op_name = entry["appellation"]
+  char_data[op_name] = parsed_data
+
+  if parsed_data["recruitment"] == 2:
+    for tag in entry["tags"]:
+      if tags[tag] not in en_recruitable:
+        en_recruitable[tags[tag]] = []
+      if tags[tag] not in cn_recruitable:
+        cn_recruitable[tags[tag]] = []
+
+      en_recruitable[tags[tag]].append(op_name)
+      cn_recruitable[tags[tag]].append(op_name)
+  elif parsed_data["recruitment"] == 1:
+    for tag in entry["tags"]:
+      if tags[tag] not in cn_recruitable:
+        cn_recruitable[tags[tag]] = []
+      cn_recruitable[tags[tag]].append(op_name)
+
+project_root = sys.argv[1]
+
+with open(project_root + "/aux/char_data.json", "w") as f:
+  f.write(json.dumps(char_data))
+
+with open(project_root + "/aux/en_recruitable.json", "w") as f:
+  f.write(json.dumps(en_recruitable))
+
+with open(project_root + "/aux/cn_recruitable.json", "w") as f:
+  f.write(json.dumps(cn_recruitable))
\ No newline at end of file
diff --git a/aux/get_recruit_data b/aux/get_recruit_data
new file mode 100755
index 0000000..73d17c0
--- /dev/null
+++ b/aux/get_recruit_data
@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+
+set -euo pipefail
+
+PROJECT_ROOT=$(cargo locate-project | sed -E 's#.*"(/.*)/Cargo.toml"}#\1#')
+DATASET="https://raw.githubusercontent.com/Tsuk1ko/arknights-toolbox/master/src/data/character.json"
+
+curl "$DATASET" | cat | "$PROJECT_ROOT"/aux/filter_raw_character_data "$PROJECT_ROOT"
diff --git a/aux/init_db b/aux/init_db
new file mode 100755
index 0000000..ca038e7
--- /dev/null
+++ b/aux/init_db
@@ -0,0 +1,32 @@
+#!/usr/bin/env bash
+
+set -euo pipefail
+
+DB_NAME="data.sqlite"
+PROJECT_ROOT=$(cargo locate-project | sed -E 's#.*"(/.*)/Cargo.toml"}#\1#')
+
+if [ ! -f "$PROJECT_ROOT/$DB_NAME" ]; then
+  sqlite3 "$PROJECT_ROOT/$DB_NAME" <<EOF
+  CREATE TABLE IF NOT EXISTS Heck (
+    id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL UNIQUE,
+    count INTEGER NOT NULL
+  );
+
+  CREATE TABLE IF NOT EXISTS RollCount (
+      user_id TEXT PRIMARY KEY UNIQUE NOT NULL,
+      count INTEGER NOT NULL
+  );
+
+  CREATE TABLE IF NOT EXISTS OperatorCount (
+    user_id TEXT NOT NULL,
+    operator TEXT NOT NULL,
+    count INTEGER NOT NULL,
+    UNIQUE(user_id, operator)
+  );
+
+  INSERT INTO Heck (id, count) VALUES (1, 0)
+    ON CONFLICT(id) DO NOTHING;
+EOF
+else
+  echo "$DB_NAME exists, exiting."
+fi