From c22484db7cb7e5148ec0703181414494c6ab0e9a Mon Sep 17 00:00:00 2001
From: kPherox <admin@mail.kr-kp.com>
Date: Sat, 27 Apr 2019 18:59:20 +0900
Subject: [PATCH] Replace promise to await & trycatch

---
 src/main/Auth.ts | 157 ++++++++++++++++++++++++-----------------------
 1 file changed, 79 insertions(+), 78 deletions(-)

diff --git a/src/main/Auth.ts b/src/main/Auth.ts
index 7bcd0fb1..3265a55d 100644
--- a/src/main/Auth.ts
+++ b/src/main/Auth.ts
@@ -1,5 +1,5 @@
 import { ipcMain, Event, shell, app } from "electron"
-import Mastodon, { Response, Account } from "megalodon"
+import Mastodon, { Response, Account, OAuth } from "megalodon"
 import { join } from "path"
 import Datastore from "nedb"
 
@@ -20,94 +20,95 @@ export default class Auth {
     ipcMain.on("new-account-setup", async (event: Event, instance: string) => {
       const SCOPES: string = "read write follow"
       let url: string | null
-      Mastodon.registerApp(
-        "TheDesk",
-        {
-          scopes: SCOPES
-        },
-        "https://" + instance
-      )
-        .then(appData => {
-          clientId = appData.clientId
-          clientSecret = appData.clientSecret
-          url = appData.url
-          if (url) {
-            shell.openExternal(
-              url,
-              {
-                activate: false
-              },
-              err => {
-                if (err) console.log(err)
-              }
-            )
-          } else {
-            event.sender.send(`error`, {
-              id: "ERROR_GET_AUTHURL",
-              message: "Failed to get auth URL to login."
-            })
-          }
-        })
-        .catch((err: Error) =>
-          event.sender.send(`error`, {
-            id: "ERROR_CONNECTION",
-            message: "Connection error",
-            meta: err
-          })
+      try {
+        let appData: OAuth.AppData = await Mastodon.registerApp(
+          "TheDesk",
+          {
+            scopes: SCOPES
+          },
+          "https://" + instance
         )
+        clientId = appData.clientId
+        clientSecret = appData.clientSecret
+        url = appData.url
+        if (url === undefined || url === null) {
+          event.sender.send(`error`, {
+            id: "ERROR_GET_AUTHURL",
+            message: "Failed to get auth URL to login."
+          })
+          return
+        }
+        shell.openExternal(
+          url,
+          {
+            activate: false
+          },
+          err => {
+            if (err) console.log(err)
+          }
+        )
+      } catch (err) {
+        let error: Error = err
+        event.sender.send(`error`, {
+          id: "ERROR_CONNECTION",
+          message: "Connection error",
+          meta: error
+        })
+      }
     })
     ipcMain.on(
       "new-account-auth",
       async (event: Event, code: string, instance: string) => {
         let url: string = "https://" + instance
-        Mastodon.fetchAccessToken(clientId, clientSecret, code, url)
-          .then((tokenData: Partial<{ accessToken: string }>) => {
-            if (tokenData.accessToken) {
-              const client = new Mastodon(
-                tokenData.accessToken,
-                url + "/api/v1"
-              )
+        try {
+          let tokenData: Partial<{ accessToken: string }> = await Mastodon.fetchAccessToken(clientId, clientSecret, code, url)
+          if (tokenData.accessToken === undefined) {
+            event.sender.send(`error`, {
+              id: "ERROR_GET_TOKEN",
+              message: "Failed to get access token."
+            })
+            return
+          }
 
-              client
-                .get<Account>("/accounts/verify_credentials")
-                .then((resp: Response<Account>) => {
-                  let you = resp.data
-                  let db = new Datastore({
-                    filename: join(app.getPath("userData"), "account.db"),
-                    autoload: true
-                  })
-                  let docs: AccountDoc = {
-                    domain: instance,
-                    acct: you.acct,
-                    avatar: you.avatar,
-                    avatarStatic: you.avatar_static,
-                    accessToken: tokenData.accessToken,
-                  }
-                  db.insert(docs, function (err, newDocs) {
-                    if (err) {
-                      event.sender.send(`error`, {
-                        id: "ERROR_YOU_TRY_ANOTHER_ACCOUNT",
-                        message: "You cannot login already logined account."
-                      })
-                    } else {
-                      event.sender.send(`login-complete`, newDocs)
-                    }
-                  })
-                })
-            } else {
+          const client = new Mastodon(
+            tokenData.accessToken,
+            url + "/api/v1"
+          )
+
+          let resp: Response<Account> = await client.get<Account>("/accounts/verify_credentials")
+          let you = resp.data
+
+          let db = new Datastore({
+            filename: join(app.getPath("userData"), "account.db"),
+            autoload: true
+          })
+
+          let docs: AccountDoc = {
+            domain: instance,
+            acct: you.acct,
+            avatar: you.avatar,
+            avatarStatic: you.avatar_static,
+            accessToken: tokenData.accessToken,
+          }
+
+          db.insert(docs, function (err, newDocs) {
+            if (err) {
               event.sender.send(`error`, {
-                id: "ERROR_GET_TOKEN",
-                message: "Failed to get access token."
+                id: "ERROR_YOU_TRY_ANOTHER_ACCOUNT",
+                message: "You cannot login already logined account."
               })
+            } else {
+              event.sender.send(`login-complete`, newDocs)
             }
           })
-          .catch((err: Error) =>
-            event.sender.send(`error`, {
-              id: "ERROR_CONNECTION",
-              message: "Connection error",
-              meta: err
-            })
-          )
+        } catch (err) {
+          let error: Error = err
+          event.sender.send(`error`, {
+            id: "ERROR_CONNECTION",
+            message: "Connection error",
+            meta: error
+          })
+        }
       }
     )
   }