package com.juick.components;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.juick.ExternalToken;
import com.juick.User;
import com.juick.components.mpns.MPNSError;
import com.juick.components.mpns.MPNSToken;
import com.juick.service.component.*;
import com.juick.util.MessageUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.text.StringEscapeUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.*;
import org.springframework.http.converter.StringHttpMessageConverter;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.HttpClientErrorException;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.util.UriComponentsBuilder;

import javax.annotation.PostConstruct;
import javax.inject.Inject;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Created by vital on 29.03.2017.
 */
public class MPNSManager implements NotificationListener {

    private static Logger logger = LoggerFactory.getLogger(MPNSManager.class);

    private String accessToken;

    @Inject
    private ObjectMapper jsonMapper;
    @Value("${wns_application_sip:}")
    private String applicationSip;
    @Value("${wns_client_secret:}")
    private String applicationSecret;
    private RestTemplate wnsService;

    @Inject
    private NotificationsManager notificationsManager;

    @PostConstruct
    public void authenticate() throws IOException {
        String url = "https://login.live.com/accesstoken.srf";
        MultiValueMap<String, String> form = new LinkedMultiValueMap<>();
        form.add("grant_type", "client_credentials");
        form.add("client_id", applicationSip);
        form.add("client_secret", applicationSecret);
        form.add("scope", "notify.windows.com");
        HttpHeaders httpHeaders = new HttpHeaders();
        httpHeaders.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
        wnsService = new RestTemplate();
        wnsService.getMessageConverters().add(new StringHttpMessageConverter(StandardCharsets.UTF_8));
        HttpEntity<MultiValueMap<String, String>> entity = new HttpEntity<>(form, httpHeaders);
        ResponseEntity<String> response = wnsService.exchange(url, HttpMethod.POST, entity, String.class);
        String responseBody = response.getBody();
        HttpStatus statusCode = response.getStatusCode();
        if (statusCode != HttpStatus.OK) {
            MPNSError error = jsonMapper.readValue(responseBody, MPNSError.class);
            throw new IOException(error.getError() + ": " + error.getErrorDescription());
        }
        MPNSToken token = jsonMapper.readValue(responseBody, MPNSToken.class);
        if (token.getTokenType().length() >= 1) {
            token.setTokenType(Character.toUpperCase(token.getTokenType().charAt(0)) + token.getTokenType().substring(1));
        }
        accessToken = token.getTokenType() + " " + token.getAccessToken();
        logger.info("MPNS authenticated");
    }

    void sendNotification(final String url, final String xml) throws IOException {
        HttpHeaders httpHeaders = new HttpHeaders();
        httpHeaders.setContentType(new MediaType("text", "xml", StandardCharsets.UTF_8));
        httpHeaders.set("Authorization", accessToken);
        httpHeaders.set("X-WNS-Type", "wns/toast");
        HttpEntity<String> requestEntity = new HttpEntity<>(xml, httpHeaders);
        try {
            UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(url);
            wnsService.exchange(builder.build(true).toUri(), HttpMethod.POST, requestEntity, Void.class);
        } catch (HttpClientErrorException ex) {
            HttpStatus statusCode = ex.getStatusCode();
            if (statusCode == HttpStatus.GONE) {
                // expired
                logger.info("{} is scheduled to remove", url);
                notificationsManager.addInvalidMPNSToken(url);
            } else {
                String headersContent = ex.getResponseHeaders().entrySet().stream()
                        .filter(x -> x.getKey().startsWith("X-WNS-") || x.getKey().startsWith("WWW-"))
                                        .map(x -> x.getKey() + ": " + String.join(",", x.getValue()))
                                        .collect(Collectors.joining("\n"));
                throw new IOException(headersContent);
            }
        }
    }

    @Override
    public void processMessageEvent(MessageEvent messageEvent) {
        com.juick.Message jmsg = messageEvent.getMessage();
        List<User> users = messageEvent.getUsers();

        List<String> urls = users.stream().flatMap(u -> u.getTokens().stream()).filter(d -> d.getType().equals("mpns"))
                .map(ExternalToken::getToken).collect(Collectors.toList());

        if (urls.isEmpty()) {
            logger.info("WNS: no recipients");
        } else {
            try {
                String text1 = "@" + jmsg.getUser().getName();
                if (!jmsg.getTags().isEmpty()) {
                    text1 += ":" + StringEscapeUtils.escapeXml11(MessageUtils.getTagsString(jmsg));
                }
                String text2 = StringEscapeUtils.escapeXml11(StringUtils.defaultString(jmsg.getText()));
                String xml = "<?xml version=\"1.0\" encoding=\"utf-8\"?>"
                        + "<toast>"
                        + "<visual>"
                        + "<binding template=\"ToastImageAndText02\">"
                        + "<image id=\"1\" src=\"http://i.juick.com/as/" + jmsg.getUser().getUid() + ".png\" />"
                        + "<text id=\"1\">" + text1 + "</text>"
                        + "<text id=\"2\">" + text2 + "</text>"
                        + "</binding>"
                        + "</visual>"
                        + "<commands>"
                        + "<command arguments=\"/ThreadView.xaml?mid=" + jmsg.getMid() + "\" />"
                        + "</commands>"
                        + "</toast>";
                logger.trace(xml);
                for (String url : urls) {
                    logger.info("WNS: {}", url);
                    sendNotification(url, xml);
                }
            } catch (IOException | IllegalStateException ex) {
                logger.error("WNS: ", ex);
            }
        }
    }

    @Override
    public void processSubscribeEvent(SubscribeEvent subscribeEvent) {

    }

    @Override
    public void processLikeEvent(LikeEvent likeEvent) {

    }

    @Override
    public void processPingEvent(PingEvent pingEvent) {

    }

    @Override
    public void processMessageReadEvent(MessageReadEvent messageReadEvent) {

    }
}