com.authy.api.Users Maven / Gradle / Ivy

Go to download
package com.authy.api;

import com.authy.AuthyException;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author Julian Camargo
 */
public class Users extends Resource {
    public static final String NEW_USER_PATH = "/protected/json/users/new";
    public static final String DELETE_USER_PATH = "/protected/json/users/delete/";
    public static final String SMS_PATH = "/protected/json/sms/";
    public static final String ONE_CODE_CALL_PATH = "/protected/json/call/";
    public static final String USER_STATUS_PATH = "/protected/json/users/%d/status";
    public static final String DEFAULT_COUNTRY_CODE = "1";

    public Users(String uri, String key) {
        super(uri, key, Resource.JSON_CONTENT_TYPE);
    }

    public Users(String uri, String key, boolean testFlag) {
        super(uri, key, testFlag, Resource.JSON_CONTENT_TYPE);
    }

    /**
     * Create a new user using his e-mail, phone and country code.
     *
     * @param email
     * @param phone
     * @param countryCode
     * @return a User instance
     */
    public com.authy.api.User createUser(String email, String phone, String countryCode) throws AuthyException {
        User user = new User(email, phone, countryCode);
        final Response response = this.post(NEW_USER_PATH, user);
        return userFromJson(response.getStatus(), response.getBody());
    }

    /**
     * Create a new user using his e-mail and phone. It uses USA country code by default.
     *
     * @param email
     * @param phone
     * @return a User instance
     */
    public com.authy.api.User createUser(String email, String phone) throws AuthyException {
        return createUser(email, phone, DEFAULT_COUNTRY_CODE);
    }

    /**
     * Send token via sms to a user.
     *
     * @param userId
     * @return Hash instance with API's response.
     */
    public Hash requestSms(int userId) throws AuthyException {
        return requestSms(userId, new HashMap<>(0));
    }

    /**
     * Send token via sms to a user with some options defined.
     *
     * @param userId
     * @param options
     * @return Hash instance with API's response.
     */
    public Hash requestSms(int userId, Map options) throws AuthyException {
        MapToResponse opt = new MapToResponse(options);
        final Response response = this.get(SMS_PATH + Integer.toString(userId), opt);
        return instanceFromJson(response.getStatus(), response.getBody());
    }

    /**
     * Send token via call to a user.
     *
     * @param userId
     * @return Hash instance with API's response.
     */
    public Hash requestCall(int userId) throws AuthyException {
        return requestCall(userId, new HashMap<>(0));
    }

    /**
     * Send token via call to a user with some options defined.
     *
     * @param userId
     * @param options
     * @return Hash instance with API's response.
     */
    public Hash requestCall(int userId, Map options) throws AuthyException {
        MapToResponse opt = new MapToResponse(options);
        final Response response = this.get(ONE_CODE_CALL_PATH + Integer.toString(userId), opt);
        return instanceFromJson(response.getStatus(), response.getBody());
    }

    /**
     * Delete a user.
     *
     * @param userId
     * @return Hash instance with API's response.
     */
    public Hash deleteUser(int userId) throws AuthyException {
        final Response response = this.post(DELETE_USER_PATH + Integer.toString(userId), null);
        return instanceFromJson(response.getStatus(), response.getBody());
    }

    /**
     * Get user status.
     *
     * @return object containing user status
     */
    public UserStatus requestStatus(int userId) throws AuthyException {
        final Response response = this.get(String.format(USER_STATUS_PATH, userId), null);
        UserStatus userStatus = userStatusFromJson(response);
        return userStatus;
    }

    private com.authy.api.User userFromJson(int status, String content) throws AuthyException {
        com.authy.api.User user = new com.authy.api.User(status, content);
        if (user.isOk()) {
            JSONObject userJson = new JSONObject(content);
            user.setId(userJson.getJSONObject("user").getInt("id"));
        } else {
            Error error = errorFromJson(content);
            user.setError(error);
        }
        return user;
    }

    private Hash instanceFromJson(int status, String content) throws AuthyException {
        Hash hash = new Hash(status, content);
        if (hash.isOk()) {
            try {
                JSONObject jsonResponse = new JSONObject(content);
                String message = jsonResponse.optString("message");
                hash.setMessage(message);

                boolean success = jsonResponse.optBoolean("success");
                hash.setSuccess(success);

                String token = jsonResponse.optString("token");
                hash.setToken(token);
            } catch (JSONException e) {
                throw new AuthyException("Invalid response from server", e);
            }
        } else {
            Error error = errorFromJson(content);
            hash.setError(error);
        }

        return hash;
    }

    private UserStatus userStatusFromJson(Response response) throws AuthyException {
        UserStatus userStatus = new UserStatus(response.getStatus(), response.getBody());
        if (userStatus.isOk()) {
            try {
                JSONObject jsonResponse = new JSONObject(response.getBody());
                String message = jsonResponse.optString("message");
                userStatus.setMessage(message);

                boolean success = jsonResponse.optBoolean("success");
                userStatus.setSuccess(success);

                JSONObject status = jsonResponse.getJSONObject("status");
                int userId = status.getInt("authy_id");
                userStatus.setUserId(userId);

                boolean confirmed = status.getBoolean("confirmed");
                userStatus.setConfirmed(confirmed);

                boolean registered = status.getBoolean("registered");
                userStatus.setRegistered(registered);

                int countryCode = status.getInt("country_code");
                userStatus.setCountryCode(countryCode);

                String phoneNumber = status.getString("phone_number");
                userStatus.setPhoneNumber(phoneNumber);

                JSONArray devicesArray = status.getJSONArray("devices");
                List devices = userStatus.getDevices();
                for (int i = 0; i < devicesArray.length(); i++) {
                    devices.add(devicesArray.getString(i));
                }

            } catch (JSONException e) {
                throw new AuthyException("Invalid response from server", e);
            }
        } else {
            Error error = errorFromJson(response.getBody());
            userStatus.setError(error);
        }

        return userStatus;
    }

    static class MapToResponse implements Formattable {
        private Map options;

        MapToResponse(Map options) {
            this.options = options;
        }

        public String toXML() {
            return "";
        }

        public Map toMap() {
            return options;
        }
    }

    @XmlRootElement(name = "user")
    static class User implements Formattable {
        String email, cellphone, countryCode;

        public User() {
        }

        public User(String email, String cellphone, String countryCode) {
            this.email = email;
            this.cellphone = cellphone;
            this.countryCode = countryCode;
        }

        @XmlElement(name = "email")
        public String getEmail() {
            return email;
        }

        public void setEmail(String email) {
            this.email = email;
        }

        @XmlElement(name = "cellphone")
        public String getCellphone() {
            return cellphone;
        }

        public void setCellphone(String cellphone) {
            this.cellphone = cellphone;
        }

        @XmlElement(name = "country_code")
        public String getCountryCode() {
            return countryCode;
        }

        public void setCountryCode(String countryCode) {
            this.countryCode = countryCode;
        }

        public String toXML() {
            StringWriter sw = new StringWriter();
            String xml = "";

            try {
                JAXBContext context = JAXBContext.newInstance(this.getClass());
                Marshaller marshaller = context.createMarshaller();

                marshaller.marshal(this, sw);

                xml = sw.toString();
            } catch (JAXBException e) {
                e.printStackTrace();
            }
            return xml;
        }

        public Map toMap() {

            Map map = new HashMap<>();
            map.put("email", email);
            map.put("cellphone", cellphone);
            map.put("country_code", countryCode);

            return map;
        }

        @Override
        public String toJSON() {
            JSONObject json = new JSONObject();
            json.put("user", toMap());
            return json.toString();
        }
    }
}