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 {
    @Value("${ap_base_uri:http://localhost:8080/}")
    private String baseUri;
    @Inject
    private JdbcTemplate jdbcTemplate;

    private NodeInfo getCurrentNodeInfo(String version) {
        NodeInfo nodeInfo = new NodeInfo();
        nodeInfo.setVersion(version);
        Server server = new Server();
        server.setBaseUrl(baseUri);
        server.setName("Juick");
        server.setSoftware("Juick");
        server.setVersion("2.x");
        nodeInfo.setServer(server);
        nodeInfo.setProtocols(Arrays.asList("xmpp", "activitypub", "smtp"));
        ServiceInfo serviceInfo = new ServiceInfo();
        serviceInfo.setInbound(Arrays.asList("jabber", "mastodon", "email", "telegram"));
        serviceInfo.setOutbound(Arrays.asList("jabber", "mastodon", "telegram", "twitter", "email", "rss"));
        nodeInfo.setServices(serviceInfo);
        UserStats userStats = new UserStats();
        userStats.setTotal(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM users WHERE banned=0", Integer.class));
        userStats.setActiveMonth(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM users WHERE banned=0 AND last_seen > ?",
                Integer.class, ZonedDateTime.now().minus(1, ChronoUnit.MONTHS).toLocalDateTime()));
        userStats.setActiveHalfyear(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM users WHERE banned=0 AND last_seen > ?",
                Integer.class, ZonedDateTime.now().minus(6, ChronoUnit.MONTHS).toLocalDateTime()));
        Usage usage = new Usage();
        usage.setUsers(userStats);
        usage.setLocalPosts(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM messages",
                Integer.class));
        usage.setLocalComments(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM replies",
                Integer.class));
        nodeInfo.setUsage(usage);
        return nodeInfo;
    }

    @GetMapping(value = "/.well-known/x-nodeinfo2", produces = MediaType.APPLICATION_JSON_VALUE)
    @JsonView(NodeInfo.XNodeInfoView.class)
    public NodeInfo showXNodeInfo() {
        return getCurrentNodeInfo("1.0");
    }

    @GetMapping(value = "/.well-known/nodeinfo", produces = 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", produces = MediaType.APPLICATION_JSON_VALUE)
    @JsonView(NodeInfo.NodeInfoView.class)
    public NodeInfo showNodeInfo() {
        return getCurrentNodeInfo("2.0");
    }
}