/******************************************************************************* * Copyright (c) 2013, 2014 Lectorius, Inc. * Authors: * Vijay Pandurangan ([email protected]) * Evan Jones ([email protected]) * Adam Hilss ([email protected]) * * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU 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 General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. * * You can contact the authors at [email protected]. *******************************************************************************/ package co.mitro.core.alerts; import java.sql.SQLException; import java.util.Collection; import java.util.Collections; import java.util.HashMap; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import co.mitro.analysis.AuditLogProcessor.ActionType; import co.mitro.core.server.Manager; import co.mitro.core.server.data.DBEmailQueue; import co.mitro.core.server.data.DBFutureAlert; import co.mitro.core.server.data.DBFutureAlert.AlertPredicate; import co.mitro.core.server.data.DBProcessedAudit; import com.google.common.collect.ImmutableList; import com.j256.ormlite.stmt.QueryBuilder; public class FirstSecretEmailer extends BaseEmailAlerter { private static final Logger logger = LoggerFactory.getLogger(FirstSecretEmailer.class); @Override protected boolean internalProcess(AlerterContext context) throws SQLException { QueryBuilder<DBProcessedAudit,Integer> builder = getBuilder(context, ImmutableList.of(ActionType.CREATE_SECRET), AlertUserType.ACTOR); builder.setCountOf(true); if (context.manager.processedAuditDao.countOf(builder.prepare()) > 0) { enqueueEmail(context, Collections.<String,String>emptyMap(), DBEmailQueue.Type.MANDRILL_SAVE_FIRST_SECRET); return true; } return false; } @Override protected AlertPredicate getMatchingAlertType() { return AlertPredicate.EMAIL_ON_FIRST_SECRET; } @Override public int createNewDBFutureAlertsForProcessedAudits( Manager mgr, Collection<DBProcessedAudit> audits, long minTimestampMs) throws SQLException { int count = 0; for (DBProcessedAudit action : audits) { if (action.getAction() == ActionType.SIGNUP) { DBFutureAlert.addNew(mgr, AlertPredicate.EMAIL_ON_FIRST_SECRET, action, action.getActor(), minTimestampMs); ++count; } } return count; } }