Merge pull request #2 from obot-ai/AddException

Fetcherが想定外のレスポンスを取得した場合に例外を発生させるように改修

example/example.dart

@@ -20,19 +20,28 @@ void main(List<String> args) async {
     }
   }
 
-  print("Fetching [$locale] data from $host with API key $apiKey");
+  Generator generator = Generator();
 
+  print("Fetching [$locale] data from $host with API key $apiKey");
   Fetcher fetcher = Fetcher(
       apiKey: apiKey,
       getEndpoint: (String locale) {
         return "$host/input_completion/$locale/";
       });
 
-  List<LocaleDataItem> jaData = await fetcher.fetch(locale);
-  print("Fetched ${jaData.length} items: $jaData");
-
-  Generator generator = Generator();
-  generator.loadData(locale, jaData);
+  try {
+    List<LocaleDataItem> localeData = await fetcher.fetch(locale);
+    print("Fetched ${localeData.length} items: $localeData");
+
+    generator.loadData(locale, localeData);
+  } on FetchFailedException catch (e) {
+    print("Failed to fetch data. Exception: $e");
+    print("ResponseBody: ${e.responseBody}");
+    return;
+  } on UnexpectedResponseBodyException catch (e) {
+    print("Unexpected response body. Exception: $e");
+    return;
+  }
 
   while (true) {
     print("Enter a keyword to get completions:");

lib/obot_completion_generator.dart

@@ -1,6 +1,11 @@
 library;
 
 export 'src/generator.dart' show Generator;
+export 'src/exceptions.dart'
+    show
+        FetcherException,
+        FetchFailedException,
+        UnexpectedResponseBodyException;
 export 'src/fetcher.dart' show Fetcher;
 export 'src/types.dart'
     show

lib/src/exceptions.dart

@@ -0,0 +1,33 @@
+class FetcherException implements Exception {
+  final String message;
+
+  FetcherException(this.message);
+
+  @override
+  String toString() {
+    return 'FetcherException(message: $message)';
+  }
+}
+
+class FetchFailedException extends FetcherException {
+  final int statusCode;
+  final String responseBody;
+
+  FetchFailedException(super.message, this.statusCode, this.responseBody);
+
+  @override
+  String toString() {
+    return 'FetchFailedException(message: $message, statusCode: $statusCode)';
+  }
+}
+
+class UnexpectedResponseBodyException extends FetcherException {
+  final int statusCode;
+
+  UnexpectedResponseBodyException(super.message, this.statusCode);
+
+  @override
+  String toString() {
+    return 'UnexpectedResponseBodyException(message: $message, statusCode: $statusCode)';
+  }
+}

lib/src/fetcher.dart

@@ -1,6 +1,7 @@
 import 'dart:convert';
 import 'dart:io';
 
+import 'exceptions.dart';
 import 'types.dart';
 
 class Fetcher {
@@ -56,10 +57,21 @@ class Fetcher {
     HttpClientResponse response = await request.close();
 
     try {
+      // レスポンスを読み取る
       responseBody = await response.transform(utf8.decoder).join();
+    } catch (e) {
+      // UTF-8でデコードできない場合は例外を投げる
+      throw UnexpectedResponseBodyException(
+          "Failed to read response from $endpoint", response.statusCode);
     } finally {
       client.close();
     }
+
+    if (response.statusCode != 200) {
+      // ステータスコードが200以外の場合は例外を投げる
+      throw FetchFailedException("Failed to fetch data from $endpoint",
+          response.statusCode, responseBody);
+    }
     return responseBody;
   }
 }