Mastodon API: timelines and minor fixes

1 files changed, 112 insertions, 49 deletions

diff --git a/src/main/java/com/juick/www/api/Mastodon.java b/src/main/java/com/juick/www/api/Mastodon.java
index 9ea937d1..7591ec92 100644
--- a/src/main/java/com/juick/www/api/Mastodon.java
+++ b/src/main/java/com/juick/www/api/Mastodon.java
@@ -17,16 +17,20 @@
 
 package com.juick.www.api;
 
+import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.databind.PropertyNamingStrategies;
 import com.fasterxml.jackson.databind.annotation.JsonNaming;
 import com.juick.model.Chat;
 import com.juick.model.Message;
 import com.juick.model.User;
 import com.juick.service.ChatService;
+import com.juick.service.MessagesService;
 import com.juick.service.TagService;
 import com.juick.service.UserService;
+import com.juick.util.HttpBadRequestException;
+import com.juick.util.MessageUtils;
 import com.juick.www.WebApp;
-import com.juick.www.api.webfinger.model.Account;
+import jakarta.validation.Valid;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
@@ -36,16 +40,13 @@ import org.springframework.security.oauth2.core.ClientAuthenticationMethod;
 import org.springframework.security.oauth2.server.authorization.client.RegisteredClient;
 import org.springframework.security.oauth2.server.authorization.client.RegisteredClientRepository;
 import org.springframework.security.oauth2.server.authorization.settings.ClientSettings;
+import org.springframework.security.oauth2.server.authorization.settings.TokenSettings;
 import org.springframework.web.bind.annotation.*;
 
 import javax.inject.Inject;
-
+import java.time.Duration;
 import java.time.Instant;
-import java.util.Arrays;
-import java.util.Collection;
-import java.util.Collections;
-import java.util.List;
-import java.util.UUID;
+import java.util.*;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -63,27 +64,31 @@ public class Mastodon {
     TagService tagService;
     @Inject
     ChatService chatService;
+    @Inject
+    MessagesService messagesService;
+    @Inject
+    User serviceUser;
 
-    @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
-    public record ApplicationRequest(String clientName,
-            String redirectUris,
-            String scopes) {
+    public record ApplicationRequest(
+            @JsonProperty("client_name") String clientName,
+            @JsonProperty("redirect_uris") String redirectUris,
+            @JsonProperty("scopes") String scopes) {
     }
 
     @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
     public record ApplicationResponse(String id,
-            String name,
-            String redirectUri,
-            String clientId,
-            String clientSecret) {
+                                      String name,
+                                      String redirectUri,
+                                      String clientId,
+                                      String clientSecret) {
     }
 
     @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
     public record CredentialAccount(String id, String username, String acct,
-            String displayName,
-            Integer followersCount,
-            Integer followingCount,
-            String avatar) {
+                                    String displayName,
+                                    Integer followersCount,
+                                    Integer followingCount,
+                                    String avatar) {
 
     }
 
@@ -91,16 +96,7 @@ public class Mastodon {
         return s != null ? Arrays.asList(s.split(" ")) : Collections.emptyList();
     }
 
-    @PostMapping(value = "/api/v1/apps", consumes = { MediaType.APPLICATION_JSON_VALUE })
-    public ApplicationResponse apps(@RequestBody ApplicationRequest application) {
-        return apps(application.clientName(), application.redirectUris(), application.scopes());
-    }
-
-    @PostMapping(value = "/api/v1/apps", consumes = { MediaType.APPLICATION_FORM_URLENCODED_VALUE,
-            MediaType.MULTIPART_FORM_DATA_VALUE })
-    public ApplicationResponse apps(@RequestParam("client_name") String clientName,
-            @RequestParam("redirect_uris") String redirectUris,
-            @RequestParam("scopes") String scopes) {
+    private ApplicationResponse registerApp(String clientName, String redirectUri, String scopes) {
         var secret = UUID.randomUUID().toString();
         RegisteredClient registeredClient = RegisteredClient.withId(UUID.randomUUID().toString())
                 .clientId(UUID.randomUUID().toString())
@@ -110,8 +106,9 @@ public class Mastodon {
                         ClientAuthenticationMethod.CLIENT_SECRET_POST,
                         ClientAuthenticationMethod.CLIENT_SECRET_BASIC)))
                 .authorizationGrantType(AuthorizationGrantType.AUTHORIZATION_CODE)
-                .redirectUri(redirectUris)
+                .redirectUri(redirectUri)
                 .scopes((coll) -> coll.addAll(parseScopes(scopes)))
+                .tokenSettings(TokenSettings.builder().accessTokenTimeToLive(Duration.ofDays(365)).build())
                 .clientSettings(ClientSettings.builder().requireAuthorizationConsent(true).build()).build();
         registeredClientRepository.save(registeredClient);
         return new ApplicationResponse(
@@ -122,12 +119,38 @@ public class Mastodon {
                 secret);
     }
 
-    public record Instance(String domain) {
+    @PostMapping(value = "/api/v1/apps", consumes = {MediaType.APPLICATION_JSON_VALUE})
+    public ApplicationResponse apps(@RequestBody @Valid ApplicationRequest application) {
+        return registerApp(application.clientName(), application.redirectUris(), application.scopes());
+    }
+
+    @PostMapping(value = "/api/v1/apps", consumes = {MediaType.APPLICATION_FORM_URLENCODED_VALUE,
+            MediaType.MULTIPART_FORM_DATA_VALUE})
+    public ApplicationResponse apps(@RequestParam("client_name") String clientName,
+                                    @RequestParam("redirect_uris") String redirectUris,
+                                    @RequestParam("scopes") String scopes) {
+        return registerApp(clientName, redirectUris, scopes);
+    }
+
+    public record Contact(
+            String email,
+            CredentialAccount account
+    ) {
+
+    }
+
+    public record Instance(
+            String uri,
+            String title,
+            String description,
+            String version,
+            String email
+    ) {
     }
 
-    @GetMapping({ "/api/v1/instance", "/api/v2/instance" })
+    @GetMapping({"/api/v1/instance", "/api/v2/instance"})
     public Instance getInstance() {
-        return new Instance(domain);
+        return new Instance(domain, "Microblogging service", "Juick", "2.x","support@juick.com");
     }
 
     @GetMapping("/api/v1/accounts/verify_credentials")
@@ -135,13 +158,15 @@ public class Mastodon {
         return toAccount(visitor);
     }
 
-    @GetMapping({ "/api/v1/announcements", "/api/v1/lists", "/api/v1/custom_emojis" })
+    @GetMapping({"/api/v1/announcements", "/api/v1/lists", "/api/v1/custom_emojis", "/api/v1/notifications"})
     public ResponseEntity<List<Void>> arrayStubs() {
         return new ResponseEntity<>(List.of(), HttpStatus.OK);
     }
 
     public record Preferences() {
-    };
+    }
+
+    ;
 
     @GetMapping("/api/v1/preferences")
     public Preferences preferences() {
@@ -150,8 +175,8 @@ public class Mastodon {
 
     @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
     public record Filter(String id, String phrase, List<String> context,
-            Instant expiresAt, Boolean irreversible,
-            Boolean wholeWord) {
+                         Instant expiresAt, Boolean irreversible,
+                         Boolean wholeWord) {
 
     }
 
@@ -162,17 +187,17 @@ public class Mastodon {
 
     @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
     public record Relationship(String id, Boolean following, Boolean showingReblogs,
-            Boolean notyfing, Boolean followingBy, Boolean blocked, Boolean blockedBy,
-            Boolean muting, Boolean mutingNotifications, Boolean requested, Boolean domainBlocking,
-            Boolean endorsed, String note) {
+                               Boolean notyfing, Boolean followingBy, Boolean blocked, Boolean blockedBy,
+                               Boolean muting, Boolean mutingNotifications, Boolean requested, Boolean domainBlocking,
+                               Boolean endorsed, String note) {
 
     }
 
     private Relationship findRelationships(String visitorId, String userId) {
-        User user = userService.getUserByUID(Integer.valueOf(userId)).orElseThrow();
-        var readers = userService.getUserReaders(Integer.valueOf(visitorId));
-        var friends = userService.getUserFriends(Integer.valueOf(visitorId));
-        var bl = userService.getUserBLUsers(Integer.valueOf(visitorId));
+        User user = userService.getUserByUID(Integer.parseInt(userId)).orElseThrow(HttpBadRequestException::new);
+        var readers = userService.getUserReaders(Integer.parseInt(visitorId));
+        var friends = userService.getUserFriends(Integer.parseInt(visitorId));
+        var bl = userService.getUserBLUsers(Integer.parseInt(visitorId));
         var isFriend = friends.contains(user);
         var isReader = readers.contains(user);
         var isMuting = bl.contains(user);
@@ -183,21 +208,26 @@ public class Mastodon {
     }
 
     @GetMapping("/api/v1/accounts/relationships")
-    public List<Relationship> relationships(@ModelAttribute User visitor, @RequestParam(value="id[]") String[] ids) {
+    public List<Relationship> relationships(@ModelAttribute User visitor, @RequestParam(value = "id[]") String[] ids) {
         return Stream.of(ids).map(
-            id -> findRelationships(String.valueOf(visitor.getUid()), id)
+                id -> findRelationships(String.valueOf(visitor.getUid()), id)
         ).collect(Collectors.toList());
     }
+
     @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
     public record Status(String id, Instant createdAt, String inReplyToId, String inReplyToAccountId,
-                         Boolean sensitive, String spoilerText, String visibility, String content) {}
+                         Boolean sensitive, String spoilerText, String visibility, String content) {
+    }
+
     @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
     public record Conversation(
             String id,
             Boolean unread,
             List<CredentialAccount> accounts,
             Status lastStatus
-    ) {}
+    ) {
+    }
+
     public CredentialAccount toAccount(User user) {
         return new CredentialAccount(
                 String.valueOf(user.getUid()),
@@ -212,7 +242,7 @@ public class Mastodon {
     public Status mapLastMessage(Chat chat) {
         return new Status(
                 String.valueOf(chat.getLastMessageTimestamp().toEpochMilli()),
-                        chat.getLastMessageTimestamp(),
+                chat.getLastMessageTimestamp(),
                 null,
                 null,
                 false,
@@ -236,4 +266,37 @@ public class Mastodon {
                 this::toConversation
         ).collect(Collectors.toList());
     }
+
+    public Status toStatus(Message message) {
+        return new Status(
+                String.valueOf(message.getMid()),
+                message.getCreated(),
+                null,
+                null,
+                MessageUtils.isSensitive(message),
+                "",
+                "public",
+                MessageUtils.formatMessage(message.getText())
+        );
+    }
+
+    @GetMapping("/api/v1/timelines/{timeline}")
+    public List<Status> publicTimeline(@ModelAttribute User visitor,
+                                       @PathVariable String timeline,
+                                       @RequestParam(value = "max_id", required = false, defaultValue = "0") int before,
+                                       @RequestParam(value = "only_media", required = false, defaultValue = "false") Boolean media) {
+        List<Integer> mids = List.of();
+        if (timeline.equals("public")) {
+            if (media) {
+                mids = messagesService.getPhotos(visitor.getUid(), before);
+            } else {
+                mids = messagesService.getAll(visitor.getUid(), before);
+            }
+        } else if (timeline.equals("home")) {
+            mids = messagesService.getMyFeed(visitor.getUid(), before, true);
+        }
+        return messagesService.getMessages(visitor, mids).stream()
+                .map(this::toStatus)
+                .collect(Collectors.toList());
+    }
 }