Added asuna.ga as an image source

Serious-senpai · Jan 15, 2023 · cbea3ad · cbea3ad
1 parent 8614ac3
commit cbea3ad
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 10 deletions.
diff --git a/lib/errors.dart b/lib/errors.dart
@@ -9,5 +9,5 @@ abstract class ApplicationException implements Exception {
 
 class RequestCancelledException implements ApplicationException {
   @override
-  final String message = "Request to fetch image has been cancelled";
+  final message = "Request to fetch image has been cancelled";
 }
diff --git a/lib/sources.dart b/lib/sources.dart
@@ -44,10 +44,10 @@ abstract class ImageSource {
 
 class _BaseImageSource extends ImageSource {
   @override
-  final Set<String> sfw = <String>{};
+  final sfw = <String>{};
 
   @override
-  final Set<String> nsfw = <String>{};
+  final nsfw = <String>{};
 
   @override
   String get baseUrl => throw UnimplementedError;
@@ -75,11 +75,11 @@ class _BaseImageSource extends ImageSource {
   }
 }
 
-class WaifuPics extends _BaseImageSource {
+class WaifuPICS extends _BaseImageSource {
   @override
-  final String baseUrl = "api.waifu.pics";
+  final baseUrl = "api.waifu.pics";
 
-  WaifuPics(ImageClient client) : super(client);
+  WaifuPICS(ImageClient client) : super(client);
 
   @override
   Future<void> populateCategories() async {
@@ -100,11 +100,11 @@ class WaifuPics extends _BaseImageSource {
   }
 }
 
-class WaifuIm extends _BaseImageSource {
+class WaifuIM extends _BaseImageSource {
   @override
-  final String baseUrl = "api.waifu.im";
+  final baseUrl = "api.waifu.im";
 
-  WaifuIm(ImageClient client) : super(client);
+  WaifuIM(ImageClient client) : super(client);
 
   @override
   Future<void> populateCategories() async {
@@ -145,6 +145,41 @@ class WaifuIm extends _BaseImageSource {
   }
 }
 
+class AsunaGA extends _BaseImageSource {
+  @override
+  final baseUrl = "asuna.ga";
+
+  final _urlMap = <String, String>{};
+
+  AsunaGA(ImageClient client) : super(client);
+
+  @override
+  Future<void> populateCategories() async {
+    final converter = <String, String>{"wholesome_foxes": "foxes"};
+
+    var response = await http.get(Uri.https("asuna.ga", "/api"));
+    var data = json.decode(response.body);
+    for (var tag in data["allEndpoints"]) {
+      var url = data["endpointInfo"][tag]["url"];
+      tag = converter[tag] ?? tag;
+      _urlMap[tag] = url;
+
+      sfw.add(tag);
+    }
+  }
+
+  @override
+  Future<String> getImageUrl(String category, {required bool isSfw}) async {
+    var response = await http.get(Uri.parse(_urlMap[category]!));
+    var data = json.decode(response.body);
+    return data["url"];
+  }
+}
+
 List<ImageSource> constructSources(ImageClient client) {
-  return <ImageSource>[WaifuPics(client), WaifuIm(client)];
+  return <ImageSource>[
+    WaifuPICS(client),
+    WaifuIM(client),
+    AsunaGA(client),
+  ];
 }