Search in sources :

Example 1 with AuthPollResponseError

use of com.cloudcraftgaming.discal.api.object.json.google.AuthPollResponseError in project DisCal-Discord-Bot by NovaFox161.

the class Authorization method pollForAuth.

void pollForAuth(Poll poll) {
    GuildSettings settings = DatabaseManager.getManager().getSettings(poll.getGuild().getLongID());
    try {
        String body = "client_id=" + clientData.getClientId() + "&client_secret=" + clientData.getClientSecret() + "&code=" + poll.getDevice_code() + "&grant_type=http://oauth.net/grant_type/device/1.0";
        // Execute
        com.mashape.unirest.http.HttpResponse<JsonNode> response = Unirest.post("https://www.googleapis.com/oauth2/v4/token").header("Content-Type", "application/x-www-form-urlencoded").body(body).asJson();
        // Handle response.
        if (response.getStatus() == 403) {
            // Handle access denied
            Message.sendDirectMessage(MessageManager.getMessage("AddCalendar.Auth.Poll.Failure.Deny", settings), poll.getUser());
        } else if (response.getStatus() == 400) {
            try {
                // See if auth is pending, if so, just reschedule.
                Type type = new TypeToken<AuthPollResponseError>() {
                }.getType();
                AuthPollResponseError apre = new Gson().fromJson(response.getBody().toString(), type);
                if (apre.error.equalsIgnoreCase("authorization_pending")) {
                    // Response pending
                    PollManager.getManager().scheduleNextPoll(poll);
                } else if (apre.error.equalsIgnoreCase("expired_token")) {
                    Message.sendDirectMessage(MessageManager.getMessage("AddCalendar.Auth.Poll.Failure.Expired", settings), poll.getUser());
                } else {
                    Message.sendDirectMessage(MessageManager.getMessage("Notification.Error.Network", settings), poll.getUser());
                    Logger.getLogger().debug(poll.getUser(), "Poll Failure!", "Status code: " + response.getStatus() + " | " + response.getStatusText() + " | " + response.getBody().toString(), this.getClass(), true);
                }
            } catch (Exception e) {
                // Auth is not pending, error occurred.
                Logger.getLogger().exception(poll.getUser(), "Failed to poll for authorization to google account.", e, this.getClass(), true);
                Logger.getLogger().debug(poll.getUser(), "More info on failure", "Status code: " + response.getStatus() + " | " + response.getStatusText() + " | " + response.getBody().toString(), this.getClass(), true);
                Message.sendDirectMessage(MessageManager.getMessage("Notification.Error.Network", settings), poll.getUser());
            }
        } else if (response.getStatus() == 429) {
            // We got rate limited... oops. Let's just poll half as often.
            poll.setInterval(poll.getInterval() * 2);
            PollManager.getManager().scheduleNextPoll(poll);
        } else if (response.getStatus() == HttpStatusCodes.STATUS_CODE_OK) {
            // Access granted
            Type type = new TypeToken<AuthPollResponseGrant>() {
            }.getType();
            AuthPollResponseGrant aprg = new Gson().fromJson(response.getBody().toString(), type);
            // Save credentials securely.
            GuildSettings gs = DatabaseManager.getManager().getSettings(poll.getGuild().getLongID());
            AESEncryption encryption = new AESEncryption(gs);
            gs.setEncryptedAccessToken(encryption.encrypt(aprg.access_token));
            gs.setEncryptedRefreshToken(encryption.encrypt(aprg.refresh_token));
            DatabaseManager.getManager().updateSettings(gs);
            try {
                Calendar service = CalendarAuth.getCalendarService(gs);
                List<CalendarListEntry> items = service.calendarList().list().setMinAccessRole("writer").execute().getItems();
                Message.sendDirectMessage(MessageManager.getMessage("AddCalendar.Auth.Poll.Success", settings), poll.getUser());
                for (CalendarListEntry i : items) {
                    if (!i.isDeleted()) {
                        EmbedBuilder em = new EmbedBuilder();
                        em.withAuthorIcon(Main.client.getGuildByID(266063520112574464L).getIconURL());
                        em.withAuthorName("DisCal");
                        em.withTitle(MessageManager.getMessage("Embed.AddCalendar.List.Title", settings));
                        em.appendField(MessageManager.getMessage("Embed.AddCalendar.List.Name", settings), i.getSummary(), false);
                        em.appendField(MessageManager.getMessage("Embed.AddCalendar.List.TimeZone", settings), i.getTimeZone(), false);
                        em.appendField(MessageManager.getMessage("Embed.AddCalendar.List.ID", settings), i.getId(), false);
                        em.withUrl(CalendarMessageFormatter.getCalendarLink(i.getId()));
                        em.withColor(56, 138, 237);
                        Message.sendDirectMessage(em.build(), poll.getUser());
                    }
                }
            // Response will be handled in guild, and will check. We already saved the tokens anyway.
            } catch (IOException e1) {
                // Failed to get calendars list and check for calendars.
                Logger.getLogger().exception(poll.getUser(), "Failed to list calendars from external account!", e1, this.getClass(), true);
                Message.sendDirectMessage(MessageManager.getMessage("AddCalendar.Auth.Poll.Failure.ListCalendars", settings), poll.getUser());
            }
        } else {
            // Unknown network error...
            Message.sendDirectMessage(MessageManager.getMessage("Notification.Error.Network", settings), poll.getUser());
            Logger.getLogger().debug(poll.getUser(), "Network error; poll failure", "Status code: " + response.getStatus() + " | " + response.getStatusText() + " | " + response.getBody().toString(), this.getClass(), true);
        }
    } catch (Exception e) {
        // Handle exception.
        Logger.getLogger().exception(poll.getUser(), "Failed to poll for authorization to google account", e, this.getClass(), true);
        Message.sendDirectMessage(MessageManager.getMessage("Notification.Error.Unknown", settings), poll.getUser());
    }
}
Also used : AESEncryption(com.cloudcraftgaming.discal.api.crypto.AESEncryption) Calendar(com.google.api.services.calendar.Calendar) Gson(com.google.gson.Gson) JsonNode(com.mashape.unirest.http.JsonNode) IOException(java.io.IOException) AuthPollResponseGrant(com.cloudcraftgaming.discal.api.object.json.google.AuthPollResponseGrant) IOException(java.io.IOException) Type(java.lang.reflect.Type) CalendarListEntry(com.google.api.services.calendar.model.CalendarListEntry) EmbedBuilder(sx.blah.discord.util.EmbedBuilder) TypeToken(com.google.gson.reflect.TypeToken) List(java.util.List) AuthPollResponseError(com.cloudcraftgaming.discal.api.object.json.google.AuthPollResponseError) GuildSettings(com.cloudcraftgaming.discal.api.object.GuildSettings)

Aggregations

AESEncryption (com.cloudcraftgaming.discal.api.crypto.AESEncryption)1 GuildSettings (com.cloudcraftgaming.discal.api.object.GuildSettings)1 AuthPollResponseError (com.cloudcraftgaming.discal.api.object.json.google.AuthPollResponseError)1 AuthPollResponseGrant (com.cloudcraftgaming.discal.api.object.json.google.AuthPollResponseGrant)1 Calendar (com.google.api.services.calendar.Calendar)1 CalendarListEntry (com.google.api.services.calendar.model.CalendarListEntry)1 Gson (com.google.gson.Gson)1 TypeToken (com.google.gson.reflect.TypeToken)1 JsonNode (com.mashape.unirest.http.JsonNode)1 IOException (java.io.IOException)1 Type (java.lang.reflect.Type)1 List (java.util.List)1 EmbedBuilder (sx.blah.discord.util.EmbedBuilder)1