org.whispersystems.textsecuregcm.sms.TwilioSmsSender.java Source code

Introduction

Here is the source code for org.whispersystems.textsecuregcm.sms.TwilioSmsSender.java

Source

/**
 * Copyright (C) 2013 Open WhisperSystems
 *
 * 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 org.whispersystems.textsecuregcm.sms;

import com.codahale.metrics.Meter;
import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.SharedMetricRegistries;
import com.google.common.base.Optional;
import com.twilio.sdk.TwilioRestClient;
import com.twilio.sdk.TwilioRestException;
import com.twilio.sdk.resource.factory.CallFactory;
import com.twilio.sdk.resource.factory.MessageFactory;
import org.apache.http.NameValuePair;
import org.apache.http.message.BasicNameValuePair;
import org.whispersystems.textsecuregcm.configuration.TwilioConfiguration;
import org.whispersystems.textsecuregcm.util.Constants;
import org.whispersystems.textsecuregcm.util.Util;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Random;

import static com.codahale.metrics.MetricRegistry.name;

public class TwilioSmsSender {

    public static final String SAY_TWIML = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + "<Response>\n"
            + "    <Say voice=\"woman\" language=\"en\" loop=\"3\">" + SmsSender.VOX_VERIFICATION_TEXT
            + "%s.</Say>\n" + "</Response>";

    private final MetricRegistry metricRegistry = SharedMetricRegistries.getOrCreate(Constants.METRICS_NAME);
    private final Meter smsMeter = metricRegistry.meter(name(getClass(), "sms", "delivered"));
    private final Meter voxMeter = metricRegistry.meter(name(getClass(), "vox", "delivered"));

    private final String accountId;
    private final String accountToken;
    private final ArrayList<String> numbers;
    private final String messagingServicesId;
    private final String localDomain;
    private final Random random;

    public TwilioSmsSender(TwilioConfiguration config) {
        this.accountId = config.getAccountId();
        this.accountToken = config.getAccountToken();
        this.numbers = new ArrayList<>(config.getNumbers());
        this.localDomain = config.getLocalDomain();
        this.messagingServicesId = config.getMessagingServicesId();
        this.random = new Random(System.currentTimeMillis());
    }

    public void deliverSmsVerification(String destination, Optional<String> clientType, String verificationCode)
            throws IOException, TwilioRestException {
        TwilioRestClient client = new TwilioRestClient(accountId, accountToken);
        MessageFactory messageFactory = client.getAccount().getMessageFactory();
        List<NameValuePair> messageParams = new LinkedList<>();
        messageParams.add(new BasicNameValuePair("To", destination));

        if (Util.isEmpty(messagingServicesId)) {
            messageParams.add(new BasicNameValuePair("From", getRandom(random, numbers)));
        } else {
            messageParams.add(new BasicNameValuePair("MessagingServiceSid", messagingServicesId));
        }

        if ("ios".equals(clientType.orNull())) {
            messageParams.add(new BasicNameValuePair("Body",
                    String.format(SmsSender.SMS_IOS_VERIFICATION_TEXT, verificationCode, verificationCode)));
        } else {
            messageParams.add(new BasicNameValuePair("Body",
                    String.format(SmsSender.SMS_VERIFICATION_TEXT, verificationCode)));
        }

        try {
            messageFactory.create(messageParams);
        } catch (RuntimeException damnYouTwilio) {
            throw new IOException(damnYouTwilio);
        }

        smsMeter.mark();
    }

    public void deliverVoxVerification(String destination, String verificationCode)
            throws IOException, TwilioRestException {
        TwilioRestClient client = new TwilioRestClient(accountId, accountToken);
        CallFactory callFactory = client.getAccount().getCallFactory();
        Map<String, String> callParams = new HashMap<>();
        callParams.put("To", destination);
        callParams.put("From", getRandom(random, numbers));
        callParams.put("Url", "https://" + localDomain + "/v1/accounts/voice/twiml/" + verificationCode);

        try {
            callFactory.create(callParams);
        } catch (RuntimeException damnYouTwilio) {
            throw new IOException(damnYouTwilio);
        }

        voxMeter.mark();
    }

    private String getRandom(Random random, ArrayList<String> elements) {
        return elements.get(random.nextInt(elements.size()));
    }

}