DailyUsageSender.java

package com.sixsq.slipstream.action;

import com.sixsq.slipstream.action.usage.MailUsage;
import com.sixsq.slipstream.action.usage.MailUtils;
import com.sixsq.slipstream.action.usage.MailsBuilder;
import com.sixsq.slipstream.action.usage.UsageSummaries;
import com.sixsq.slipstream.es.CljElasticsearchHelper;
import com.sixsq.slipstream.persistence.User;
import com.sixsq.slipstream.persistence.UserParameter;
import com.sixsq.slipstream.util.Notifier;
import com.sixsq.slipstream.util.SscljProxy;
import org.restlet.Context;
import org.restlet.data.Parameter;
import org.restlet.engine.header.Header;
import org.restlet.resource.ClientResource;
import org.restlet.util.Series;

import java.util.*;
import java.util.logging.Logger;

/*
 * +=================================================================+
 * SlipStream Server (WAR)
 * =====
 * Copyright (C) 2013 SixSq Sarl (sixsq.com)
 * =====
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * -=================================================================-
 */
public class DailyUsageSender {

    private static final String USAGE_RESOURCE = "api/usage";

    private static final Logger logger = Logger.getLogger(DailyUsageSender.class.getName());

    public static void main(String[] args) {
        CljElasticsearchHelper.init();
        sendDailyUsageEmails();
        System.exit(0);
    }

    public static void sendDailyUsageEmails() {

        logger.info("Will send emails");
        List<User> usersToEmail = usersFilteredByMailUsage(UserParameter.MAIL_USAGE_DAILY);
        if(usersToEmail == null || usersToEmail.isEmpty()){
            logger.info("No users to send daily email to. Returning.");
            return;
        }

        Map<String, String> nameEmails  = nameEmails(usersToEmail);
        Set<String> userNames           = nameEmails.keySet();
        logger.info("List of user names to send email to : " + userNames);

        String response = getJsonYesterdayUsage(userNames);
        if(response == null) {
            logger.warning("No response from Usage service. Returning.");
            return;
        }

        UsageSummaries usageSummaries = UsageSummaries.fromJson(response);

        MailsBuilder mailsBuilder = new MailsBuilder(yesterday(), today());
        List<MailUsage> mailUsages = mailsBuilder.buildMails(usageSummaries, nameEmails);

        for(MailUsage mailUsage : mailUsages){
            trySendEmail(mailUsage);
        }
    }

    private static void trySendEmail(MailUsage mailUsage) {
        String recipient = mailUsage.to();
        boolean mailSentOK = Notifier.sendHTMLNotification(recipient, mailUsage.body());
        if(mailSentOK) {
            logger.info("Daily usage mail successfully sent to " + recipient);
        } else {
            logger.warning("Unable to send daily usage mail to " + recipient);
        }
    }


    private static String getJsonYesterdayUsage(Set<String> userNames) {
        try {
            String resourceUri = USAGE_RESOURCE + "?" + cimiQueryStringUsageYesterday(userNames);

            logger.info("Will query Usage resource with uri = '" + resourceUri + "'");

            return SscljProxy.get(resourceUri, "super ADMIN", true).getEntity().getText();

        } catch (Exception e) {
            logger.warning("Unable to getJsonYesterdayUsage :" + e.getMessage());
            return null;
        }
    }

    protected static String cimiQueryStringUsageYesterday(Set<String> userNames) {
        String yesterday = MailUtils.formatDate(yesterday());
        String today = MailUtils.formatDate(today());
        return MailUtils.cimiQueryStringUsage(userNames, yesterday, today);
    }

    protected static Date yesterday() {
        Calendar yesterday = Calendar.getInstance();
        yesterday.add(Calendar.DATE, -1);
        return yesterday.getTime();
    }

    protected static Date today() {
        return new Date();
    }

    private static List<User> usersFilteredByMailUsage(String mailUsage) {
        List<User> result = new ArrayList<User>();
        for (User u: User.list()){
            if(mailUsage.equals(u.getMailUsage())) {
                result.add(u);
            }
        }
        return result;
    }


    private static Map<String, String> nameEmails(List<User> users) {
        Map<String, String> result = new HashMap<String, String>();
        for(User user : users){
            result.put(user.getName(), user.getEmail());
        }
        return result;
    }

}