1 files changed, 33 insertions, 2 deletions
diff --git a/src/main/java/com/juick/server/api/xnodeinfo2/Info.java b/src/main/java/com/juick/server/api/xnodeinfo2/Info.java
index 36e36bdd..7ecd02cb 100644
--- a/src/main/java/com/juick/server/api/xnodeinfo2/Info.java
+++ b/src/main/java/com/juick/server/api/xnodeinfo2/Info.java
@@ -1,15 +1,22 @@
 package com.juick.server.api.xnodeinfo2;
 
+import com.cliqset.xrd.Link;
+import com.cliqset.xrd.XRD;
+import com.fasterxml.jackson.annotation.JsonView;
 import com.juick.server.api.xnodeinfo2.model.*;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.MediaType;
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.util.UriComponentsBuilder;
 
 import javax.inject.Inject;
+import java.net.URI;
 import java.time.ZonedDateTime;
 import java.time.temporal.ChronoUnit;
 import java.util.Arrays;
+import java.util.Collections;
 
 @RestController
 public class Info {
@@ -18,9 +25,9 @@ public class Info {
     @Inject
     private JdbcTemplate jdbcTemplate;
 
-    @GetMapping("/.well-known/x-nodeinfo2")
-    public NodeInfo showNodeInfo() {
+    private NodeInfo getCurrentNodeInfo(String version) {
         NodeInfo nodeInfo = new NodeInfo();
+        nodeInfo.setVersion(version);
         Server server = new Server();
         server.setBaseUrl(baseUri);
         server.setName("Juick");
@@ -47,4 +54,28 @@ public class Info {
         nodeInfo.setUsage(usage);
         return nodeInfo;
     }
+
+    @GetMapping(value = "/.well-known/x-nodeinfo2", consumes = MediaType.APPLICATION_JSON_VALUE)
+    @JsonView(NodeInfo.XNodeInfoView.class)
+    public NodeInfo showXNodeInfo() {
+        return getCurrentNodeInfo("1.0");
+    }
+
+    @GetMapping(value = "/.well-known/nodeinfo", consumes = MediaType.APPLICATION_JSON_VALUE)
+    public XRD getNodeInfoLinks() {
+        Link nodeinfo = new Link();
+        nodeinfo.setRel(URI.create("http://nodeinfo.diaspora.software/ns/schema/2.0"));
+        UriComponentsBuilder uriComponentsBuilder = UriComponentsBuilder.fromUriString(baseUri);
+        uriComponentsBuilder.replacePath("/api/nodeinfo/2.0");
+        nodeinfo.setHref(uriComponentsBuilder.build().toUri());
+        XRD xrd = new XRD();
+        xrd.setLinks(Collections.singletonList(nodeinfo));
+        return xrd;
+    }
+
+    @GetMapping(value = "/api/nodeinfo/2.0", consumes = MediaType.APPLICATION_JSON_VALUE)
+    @JsonView(NodeInfo.NodeInfoView.class)
+    public NodeInfo showNodeInfo() {
+        return getCurrentNodeInfo("2.0");
+    }
 }