/*
 * Copyright (C) 2008-2020, Juick
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.juick.www.api.xnodeinfo2;

import com.cliqset.xrd.Link;
import com.cliqset.xrd.XRD;
import com.fasterxml.jackson.annotation.JsonView;
import com.juick.www.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;
import java.util.HashMap;
import java.util.Map;

@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"));
        Map<String, String> metadata = new HashMap<>();
        metadata.put("email", "support@juick.com");
        nodeInfo.setMetadata(metadata);
        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 WHERE reply_uri IS NULL",
                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");
    }
}