refactor: add function setAccessTokenConfig

James Tsai · James Tsai · commit f516a9202d17 · 2022-03-08T16:00:47.000+08:00
diff --git a/src/command.ts b/src/command.ts
@@ -1,9 +1,8 @@
 import API from '@hackmd/api'
 import {CliUx, Command} from '@oclif/core'
-import fs from 'fs'
 
 import config from './config'
-import {getConfigFilePath} from './utils'
+import {setAccessTokenConfig} from './utils'
 
 export default abstract class HackMDCommand extends Command {
   async getAPIClient() {
@@ -13,14 +12,7 @@ export default abstract class HackMDCommand extends Command {
     try {
       await APIClient.getMe()
       if (!config.accessToken) {
-        const configFilePath = getConfigFilePath()
-        const newConfigFile = require(configFilePath)
-        newConfigFile.accessToken = token
-        fs.writeFile(configFilePath, JSON.stringify(newConfigFile, null, 2), function (err) {
-          if (err) {
-            throw err
-          }
-        })
+        setAccessTokenConfig(token)
       }
 
       return APIClient
diff --git a/src/commands/logout.ts b/src/commands/logout.ts
@@ -1,8 +1,7 @@
 import {Flags} from '@oclif/core'
-import fs from 'fs'
 
 import HackMDCommand from '../command'
-import {getConfigFilePath} from '../utils'
+import {setAccessTokenConfig} from '../utils'
 
 export default class Logout extends HackMDCommand {
   static description = 'Login to HackMD server from CLI'
@@ -20,14 +19,7 @@ You've logged out successfully
 
   async run() {
     try {
-      const configFilePath = getConfigFilePath()
-      const newConfigFile = require(configFilePath)
-      newConfigFile.accessToken = ''
-      fs.writeFile(configFilePath, JSON.stringify(newConfigFile, null, 2), function (err) {
-        if (err) {
-          throw err
-        }
-      })
+      setAccessTokenConfig('')
       this.log("You've logged out successfully")
     } catch (err) {
       this.error(err)
diff --git a/src/utils.ts b/src/utils.ts
@@ -1,3 +1,4 @@
+import fs from 'fs'
 import {homedir} from 'os'
 import * as path from 'path'
 
@@ -11,3 +12,14 @@ export function getConfigFilePath() {
 
   return path.join(configDir, 'config.json')
 }
+
+export function setAccessTokenConfig(token: string) {
+  const configFilePath = getConfigFilePath()
+  const newConfigFile = require(configFilePath)
+  newConfigFile.accessToken = token
+  fs.writeFile(configFilePath, JSON.stringify(newConfigFile, null, 2), function (err) {
+    if (err) {
+      throw err
+    }
+  })
+}
