From e5702a238c17bee9781fc34117184db15257a0e1 Mon Sep 17 00:00:00 2001
From: Bigsk <xiaxinzhe@xiaxinzhe.cn>
Date: Sat, 20 Aug 2022 22:54:29 +0800
Subject: [PATCH] v3 cache function

---
 cache.py     | 28 ++++++++++++++++++++++++++++
 donors/.keep |  0
 fonts/.keep  |  0
 index.php    | 46 +++++++++++++++++++++++++++++++++++++++-------
 teams/.keep  |  0
 5 files changed, 67 insertions(+), 7 deletions(-)
 create mode 100644 cache.py
 create mode 100644 donors/.keep
 create mode 100644 fonts/.keep
 create mode 100644 teams/.keep

diff --git a/cache.py b/cache.py
new file mode 100644
index 0000000..85299b8
--- /dev/null
+++ b/cache.py
@@ -0,0 +1,28 @@
+import os, time
+import requests
+
+while True:
+    # Read donor lists
+    for file in os.listdir("donors"):
+        print(file)
+        try:
+            donor_object = requests.get("https://api.foldingathome.org/user/{}".format(file))
+            if donor_object.status_code == 200:
+                with open(os.path.join("donors", file), "w+") as fb:
+                    fb.write(donor_object.text)
+        except Exception as e:
+            print(e)
+
+    # Read team lists
+    for file in os.listdir("teams"):
+        print(file)
+        try:
+            donor_object = requests.get("https://api.foldingathome.org/team/{}".format(file))
+            if donor_object.status_code == 200:
+                with open(os.path.join("teams", file), "w+") as fb:
+                    fb.write(donor_object.text)
+        except Exception as e:
+            print(e)
+
+    time.sleep(5)
+    
\ No newline at end of file
diff --git a/donors/.keep b/donors/.keep
new file mode 100644
index 0000000..e69de29
diff --git a/fonts/.keep b/fonts/.keep
new file mode 100644
index 0000000..e69de29
diff --git a/index.php b/index.php
index 2b7a7c6..ff3f51a 100644
--- a/index.php
+++ b/index.php
@@ -15,6 +15,9 @@ if (!isset($_GET["lang"]) || !in_array($_GET["lang"], $langs)) {
 // Get donor
 $donor = $_GET["donor"];
 
+// Get team
+$team = $_GET["team"];
+
 // Setting foor ssl
 $arr_context_options = array(
     "ssl" => array(
@@ -24,14 +27,43 @@ $arr_context_options = array(
     ) ,
 );
 
-// Request for data
-$donor_data = file_get_contents("https://api.foldingathome.org/user/".$donor, false, stream_context_create($arr_context_options)) or die("api failed");
-$donor_object = json_decode($donor_data, true);
-if ($DEBUG) var_dump($donor_object);
+// Read record
+if (file_exists("donors/".$donor)) {
+    $local_donor = file_get_contents("donors/".$donor);
+    if (!$local_donor) {
+        // Request for data
+        $donor_data = @file_get_contents("https://api.foldingathome.org/user/".$donor, false, stream_context_create($arr_context_options)) or die("api failed");
+        $donor_object = json_decode($donor_data, true);
+        if (array_key_exists("status", $donor_object)) {
+            exit("api failed");
+        }
+        if ($DEBUG) var_dump($donor_object);
+    } else {
+        $donor_object = json_decode($local_donor, true);
+    }
+}
+if (file_exists("teams/".$team)) {
+    $local_team = file_get_contents("teams/".$team);
+    if (!$local_team) {
+        // Request for data
+        $team_data = @file_get_contents("https://api.foldingathome.org/team/".$team, false, stream_context_create($arr_context_options)) or die("api failed");
+        $team_object = json_decode($team_data, true);
+        if (array_key_exists("status", $team_object)) {
+            exit("api failed");
+        }
+        if ($DEBUG) var_dump($team_object);
+    } else {
+        $team_object = json_decode($local_donor, true);
+    }
+}
 
-$team_data = file_get_contents("https://api.foldingathome.org/team/".$_GET["team"], false, stream_context_create($arr_context_options)) or die("api failed");
-$team_object = json_decode($team_data, true);
-if ($DEBUG) var_dump($team_object);
+// Save record
+if (!file_exists("donors/".$donor)) {
+    file_put_contents("donors/".$donor, "");
+}
+if (!file_exists("teams/".$team)) {
+    file_put_contents("teams/".$team, "");
+}
 
 // Load canvas
 $canvas = imagecreatetruecolor(465, 92);
diff --git a/teams/.keep b/teams/.keep
new file mode 100644
index 0000000..e69de29