From 0f4fd1d2f7bc906494cc33a742638b9f15c56910 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=E8=AB=8F=E8=A8=AA=E5=AD=90?= <suwako@076.moe>
Date: Fri, 26 Apr 2024 18:14:15 +0900
Subject: [PATCH] =?utf8?q?=E5=90=8D=E5=89=8D=E3=81=A8=E3=81=97=E3=81=A6?=
 =?utf8?q?=E4=BF=9D=E5=AD=98=E3=82=AA=E3=83=97=E3=82=B7=E3=83=A7=E3=83=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 CHANGELOG.md |   1 +
 main.c       | 126 +++++++++++++++++++++++++++++++++++++++++----------
 2 files changed, 103 insertions(+), 24 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 2185549..9c80fdb 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,6 @@
 # 0.2.0
 * ファイル名はパラメートルを付いたら、ファイル名は拡張子に
+* 名前として保存オプション
 
 # 0.1.0
 * 最初リリース
diff --git a/main.c b/main.c
index 2415780..4280305 100644
--- a/main.c
+++ b/main.c
@@ -2,6 +2,7 @@
 #include <stdlib.h>
 #include <libgen.h>
 #include <string.h>
+#include <unistd.h>
 
 #include <curl/curl.h>
 
@@ -9,6 +10,11 @@ const char* sofname = "odl";
 const char* version = "0.2.0";
 char* filename;
 
+int opt;
+int output_flag = 0;
+int version_flag = 0;
+int err_flag = 0;
+
 char* get_filename(const char* url) {
   char* fn_start = strrchr(url, '/');
   if (fn_start == NULL) {
@@ -71,52 +77,124 @@ int progress_callback(void *cp, double dt, double dn, double ut, double un) {
   return 0;
 }
 
+void handle_o(int argc, char* argv[]) {
+  output_flag = 1;
+  if (optind < argc) {
+    if (filename != NULL) {
+      fprintf(
+        stderr,
+        "-oをご利用する場合、1つのファイルだけをダウンロード出来ます。\n"
+      );
+    }
+    filename = argv[optind];
+  } else {
+    fprintf(stderr, "-oの後でファイル名がありません。");
+    err_flag = 1;
+  }
+}
+
+void flags(int opt, int argc, char* argv[]) {
+  switch (opt) {
+    case 'o':
+      handle_o(argc, argv);
+      break;
+    case 'v':
+      version_flag = 1;
+      break;
+    default:
+      err_flag = 1;
+      printf("usage: %s-%s [-ov] [url ...]\n", sofname, version);
+      break;
+  }
+}
+
+int downloader(CURL* curl, char* filename, const char* url) {
+  curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);
+  curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0L);
+
+  FILE* file = fopen(filename, "wb");
+  if (!file) {
+    perror("ファイルを開けません。");
+    curl_easy_cleanup(curl);
+    return -1;
+  }
+
+  curl_easy_setopt(curl, CURLOPT_URL, url);
+  curl_easy_setopt(curl, CURLOPT_WRITEDATA, file);
+  CURLcode res = curl_easy_perform(curl);
+  fclose(file);
+
+  if (res != CURLE_OK) {
+    fprintf(stderr, "\nダウンロードに失敗： %s\n", curl_easy_strerror(res));
+    curl_easy_cleanup(curl);
+    return -1;
+  }
+
+  printf("\n");
+
+  return 0;
+}
+
 int main(int argc, char* argv[]) {
   if (argc < 2) {
-    printf("usage: %s-%s [url ...]\n", sofname, version);
+    printf("usage: %s-%s [-ov] [url ...]\n", sofname, version);
     return 1;
   }
 
+  while ((opt = getopt(argc, argv, "ov")) != -1) {
+    flags(opt, argc, argv);
+  }
+
+  if (err_flag == 1) {
+    return 1;
+  }
+
+  if (version_flag == 1) {
+    printf("%s-%s\n", sofname, version);
+    return 0;
+  }
+
   CURL* curl = curl_easy_init();
   if (!curl) {
     perror("curlを初期設置に失敗。");
     return -1;
   }
 
-  for (int i = 1; i < argc; i++) {
-    const char* url = argv[i];
-    filename = get_filename(url);
+  // 一つのファイルだけが可能
+  if (output_flag == 1) {
+    if (optind >= argc) {
+      fprintf(stderr, "-oの後でURLがありません。");
+      return 1;
+    }
+
     if (filename == NULL) {
-      fprintf(stderr, "URLからファイル名を抽出出来ませんでした。\n");
-      continue;
+      fprintf(stderr, "-oの後でファイル名がありません。");
+      return 1;
     }
 
-    curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);
-    curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0L);
+    filename = argv[optind];
+    const char* url = argv[optind+1];
+    downloader(curl, filename, url);
 
-    FILE* file = fopen(filename, "wb");
-    if (!file) {
-      curl_easy_cleanup(curl);
-      perror("ファイルを開けません。");
-      return -1;
-    }
+    curl_easy_cleanup(curl);
+    printf("\nダウンロードに完了しました。\n");
 
-    curl_easy_setopt(curl, CURLOPT_URL, url);
-    curl_easy_setopt(curl, CURLOPT_WRITEDATA, file);
-    CURLcode res = curl_easy_perform(curl);
-    fclose(file);
+    return 0;
+  }
 
-    if (res != CURLE_OK) {
-      curl_easy_cleanup(curl);
-      fprintf(stderr, "ダウンロードに失敗： %s\n", curl_easy_strerror(res));
-      return -1;
+  // 複数ファイル名可能
+  for (int i = optind; i < argc; i++) {
+    const char* url = argv[i];
+    filename = get_filename(url);
+    if (filename == NULL) {
+      fprintf(stderr, "URLからファイル名を抽出出来ませんでした。\n");
+      continue;
     }
 
-    printf("\n");
+    downloader(curl, filename, url);
   }
 
   curl_easy_cleanup(curl);
-
   printf("\nダウンロードに完了しました。\n");
 
   return 0;
-- 
2.43.0