Merge pull request #26 from storyblok/task/int-738

task(int-738): List spaces from all regions in spaces command

Author: Thiago Saife

src/cli.js

@@ -435,8 +435,9 @@ program
       if (!api.isAuthorized()) {
         await api.processLogin()
       }
+      const { region } = creds.get()
 
-      await tasks.listSpaces(api)
+      await tasks.listSpaces(api, region)
     } catch (e) {
       console.log(chalk.red('X') + ' An error ocurred to listing spaces: ' + e.message)
       process.exit(1)

src/tasks/list-spaces.js

@@ -1,35 +1,64 @@
 const chalk = require('chalk')
-
 /**
  * @method listSpaces
  * @param api - Pass the api instance as a parameter
  * @return {Promise}
  */
 
-const listSpaces = async (api) => {
+const listSpaces = async (api, currentRegion) => {
+  const isChinaEnv = currentRegion === 'cn'
+  const regionOptions = {
+    eu: 'Europe',
+    us: 'United States'
+  }
   console.log()
   console.log(chalk.green('✓') + ' Loading spaces...')
-  console.log()
 
   if (!api) {
     console.log(chalk.red('X') + 'Api instance is required to make the request')
     return []
   }
 
-  const spaces = await api.getAllSpaces()
-    .then(res => res)
-    .catch(err => Promise.reject(err))
+  if (isChinaEnv) {
+    const spaces = await api.getAllSpacesByRegion(currentRegion)
+      .then(res => res)
+      .catch(err => Promise.reject(err))
 
-  if (!spaces) {
-    console.log(chalk.red('X') + ' No spaces were found for this user ')
-    return []
-  }
-
-  spaces.map(space => {
-    console.log(`${space.name} (id: ${space.id})`)
-  })
+    if (!spaces) {
+      console.log(chalk.red('X') + ' No spaces were found for this user ')
+      return []
+    }
+    console.log(chalk.blue(' -') + ' Spaces From China region:')
 
-  return spaces
+    spaces.map(space => {
+      console.log(`    ${space.name} (id: ${space.id})`)
+    })
+    return spaces
+  } else {
+    const spacesList = []
+    for (const key in regionOptions) {
+      spacesList.push(await api.getAllSpacesByRegion(key)
+        .then((res) => {
+          return {
+            key,
+            res
+          }
+        })
+        .catch(err => Promise.reject(err)))
+    }
+    if (!spacesList) {
+      console.log(chalk.red('X') + ' No spaces were found for this user ')
+      return []
+    }
+    spacesList.forEach(region => {
+      console.log()
+      console.log(`${chalk.blue(' -')} Spaces From ${regionOptions[region.key]} region:`)
+      region.res.forEach((space) => {
+        console.log(`    ${space.name} (id: ${space.id})`)
+      })
+    })
+    return spacesList
+  }
 }
 
 module.exports = listSpaces

src/utils/api.js

@@ -240,8 +240,13 @@ module.exports = {
     return client[method](_path, props)
   },
 
-  async getAllSpaces () {
-    return await this.getClient()
+  async getAllSpacesByRegion (region) {
+    const customClient = new Storyblok({
+      accessToken: this.accessToken,
+      oauthToken: this.oauthToken,
+      region
+    }, this.apiSwitcher(region))
+    return await customClient
       .get('spaces/', {})
       .then(res => res.data.spaces || [])
       .catch(err => Promise.reject(err))

tests/units/list-spaces.spec.js

@@ -1,6 +1,12 @@
 const { listSpaces } = require('../../src/tasks/')
 const { FAKE_SPACES } = require('../constants')
 
+const REGION_FLAGS = {
+  UNITED_STATES: 'us',
+  EUROPE: 'eu',
+  CHINA: 'cn'
+}
+
 describe('Test spaces method', () => {
   it('Testing list-spaces funtion without api instance', async () => {
     try {
@@ -10,13 +16,34 @@ describe('Test spaces method', () => {
       console.error(e)
     }
   })
-  it('Testing list-spaces funtion with api instance', async () => {
+
+  it('Testing list-spaces function for China region', async () => {
     const FAKE_API = {
-      getAllSpaces: jest.fn(() => Promise.resolve(FAKE_SPACES()))
+      getAllSpacesByRegion: jest.fn(() => Promise.resolve(FAKE_SPACES()))
     }
     expect(
-      await listSpaces(FAKE_API)
+      await listSpaces(FAKE_API, REGION_FLAGS.CHINA)
     ).toEqual(FAKE_SPACES())
-    expect(FAKE_API.getAllSpaces).toHaveBeenCalled()
+    expect(FAKE_API.getAllSpacesByRegion).toHaveBeenCalled()
+  })
+
+  it('Testing list-spaces funtion for Europe and United States regions', async () => {
+    const FAKE_API = {
+      getAllSpacesByRegion: jest.fn(() => Promise.resolve(FAKE_SPACES()))
+    }
+    const response = [
+      {
+        key: REGION_FLAGS.EUROPE,
+        res: [...FAKE_SPACES()]
+      },
+      {
+        key: REGION_FLAGS.UNITED_STATES,
+        res: [...FAKE_SPACES()]
+      }
+    ]
+
+    expect(
+      await listSpaces(FAKE_API, REGION_FLAGS.EUROPE)
+    ).toEqual(response)
   })
 })