From 56870db1adfb83c415787bdcb9471ba04d3eb86f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=AB=8F=E8=A8=AA=E5=AD=90?= <suwako@076.moe>
Date: Fri, 21 Jun 2024 22:14:36 +0900
Subject: =?UTF-8?q?=E3=82=B9=E3=83=88=E3=83=AC=E3=83=BC=E3=82=B8=E3=81=AE?=
 =?UTF-8?q?=E8=A1=A8=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.c        | 10 +++++++++-
 src/storage.c | 15 +++++++++++++++
 src/storage.h |  6 ++++++
 3 files changed, 30 insertions(+), 1 deletion(-)
 create mode 100644 src/storage.c
 create mode 100644 src/storage.h

diff --git a/main.c b/main.c
index 9fefc47..3dd62fc 100644
--- a/main.c
+++ b/main.c
@@ -20,6 +20,7 @@
 #include "src/cpu.h"
 #include "src/gpu.h"
 #include "src/memory.h"
+#include "src/storage.h"
 
 const char *sofname = "farfetch";
 const char *version = "0.0.1";
@@ -239,13 +240,20 @@ int main(int argc, char *argv[]) {
   printf("\n");
   lc++;
 
+  const char *storage = display_storage();
+  if (storage) {
+    printf("%s ", LOGO[lc]);
+    printf("%sStorage%s: %s\n", color, reset, storage);
+    lc++;
+    free((void *)storage);
+  }
+
   for (size_t i = lc; i < ls; i++) {
     printf("%s\n", LOGO[i]);
   }
 
   // TODO:
   // * 端末
-  // * ストレージ
 
   return 0;
 }
diff --git a/src/storage.c b/src/storage.c
new file mode 100644
index 0000000..a6e21fa
--- /dev/null
+++ b/src/storage.c
@@ -0,0 +1,15 @@
+#include "storage.h"
+#include "common.h"
+
+#include <string.h>
+
+const char *display_storage() {
+  const char *iszfs = run_command_s("LC_ALL=C zpool list 2>&1");
+  if (strstr(iszfs, "command not found: zpool")) {
+    return run_command_s("zpool list | awk 'NR>1 {print $1 \": \" $3 \" / \" $2}' | "
+        "sed ':a;N;$!ba;s//, /g'");
+  }
+
+  return run_command_s("df -h | awk 'NR>1 {print $1 \": \" $3 \" / \" $2}' | "
+      "sed ':a;N;$!ba;s/\\n/, /g'");
+}
diff --git a/src/storage.h b/src/storage.h
new file mode 100644
index 0000000..9ec2217
--- /dev/null
+++ b/src/storage.h
@@ -0,0 +1,6 @@
+#ifndef STORAGE_H
+#define STORAGE_H
+
+const char *display_storage();
+
+#endif
-- 
cgit v1.2.3