|
| 1 | +const { TableServiceClient, AzureNamedKeyCredential } = require('@azure/data-tables'); |
1 | 2 | var async = require('async'); |
2 | 3 |
|
3 | 4 | module.exports = function(collection, reliesOn, callback) { |
4 | 5 | if (!reliesOn['storageAccounts.listKeys']) return callback(); |
5 | 6 |
|
6 | | - var azureStorage = require('azure-storage'); |
| 7 | + if (!collection['tableService']['listTablesSegmented']) collection['tableService']['listTablesSegmented'] = {}; |
| 8 | + if (!collection['tableService']['getTableAcl']) collection['tableService']['getTableAcl'] = {}; |
7 | 9 |
|
8 | | - if (!collection['queueService']['listQueuesSegmented']) collection['queueService']['listQueuesSegmented'] = {}; |
9 | | - if (!collection['queueService']['getQueueAcl']) collection['queueService']['getQueueAcl'] = {}; |
10 | | - |
11 | | - // Loop through regions and properties in reliesOn |
12 | | - async.eachOfLimit(reliesOn['storageAccounts.listKeys'], 10,function(regionObj, region, cb) { |
13 | | - collection['queueService']['listQueuesSegmented'][region] = {}; |
14 | | - collection['queueService']['getQueueAcl'][region] = {}; |
| 10 | + async.eachOfLimit(reliesOn['storageAccounts.listKeys'], 10, function(regionObj, region, cb) { |
| 11 | + collection['tableService']['listTablesSegmented'][region] = {}; |
| 12 | + collection['tableService']['getTableAcl'][region] = {}; |
15 | 13 |
|
16 | 14 | async.eachOfLimit(regionObj, 10, function(subObj, resourceId, sCb) { |
17 | | - collection['queueService']['listQueuesSegmented'][region][resourceId] = {}; |
18 | | - |
19 | | - if (subObj && subObj.data && subObj.data.keys && subObj.data.keys[0] && subObj.data.keys[0].value) { |
20 | | - // Extract storage account name from resourceId |
21 | | - var storageAccountName = resourceId.substring(resourceId.lastIndexOf('/') + 1); |
22 | | - var storageService = new azureStorage['QueueService'](storageAccountName, subObj.data.keys[0].value); |
23 | | - |
24 | | - storageService.listQueuesSegmented(null, function(serviceErr, serviceResults) { |
25 | | - if (serviceErr || !serviceResults) { |
26 | | - collection['queueService']['listQueuesSegmented'][region][resourceId].err = (serviceErr || 'No data returned'); |
27 | | - sCb(); |
28 | | - } else { |
29 | | - collection['queueService']['listQueuesSegmented'][region][resourceId].data = serviceResults.entries; |
30 | | - |
31 | | - // Add ACLs |
32 | | - async.eachLimit(serviceResults.entries, 10, function(entryObj, entryCb) { |
33 | | - var entryId = `${resourceId}/queueService/${entryObj.name}`; |
34 | | - collection['queueService']['getQueueAcl'][region][entryId] = {}; |
35 | | - |
36 | | - storageService.getQueueAcl(entryObj.name, function(getErr, getData) { |
37 | | - if (getErr || !getData) { |
38 | | - collection['queueService']['getQueueAcl'][region][entryId].err = (getErr || 'No data returned'); |
39 | | - } else { |
40 | | - collection['queueService']['getQueueAcl'][region][entryId].data = getData; |
41 | | - } |
42 | | - entryCb(); |
43 | | - }); |
44 | | - }, function() { |
45 | | - sCb(); |
46 | | - }); |
| 15 | + collection['tableService']['listTablesSegmented'][region][resourceId] = {}; |
| 16 | + |
| 17 | + const key = subObj && subObj.data && subObj.data.keys && subObj.data.keys[0] && subObj.data.keys[0].value? subObj.data.keys[0].value:null; |
| 18 | + if (!key) return sCb(); |
| 19 | + |
| 20 | + const storageAccountName = resourceId.substring(resourceId.lastIndexOf('/') + 1); |
| 21 | + const credential = new AzureNamedKeyCredential(storageAccountName, key); |
| 22 | + const serviceClient = new TableServiceClient( |
| 23 | + `https://${storageAccountName}.table.core.windows.net`, |
| 24 | + credential |
| 25 | + ); |
| 26 | + |
| 27 | + const tables = []; |
| 28 | + |
| 29 | + (async() => { |
| 30 | + try { |
| 31 | + for await (const table of serviceClient.listTables()) { |
| 32 | + tables.push(table.name); |
47 | 33 | } |
48 | | - }); |
49 | | - } else { |
50 | | - sCb(); |
51 | | - } |
52 | | - }, function() { |
53 | | - cb(); |
54 | | - }); |
55 | | - }, function() { |
56 | | - callback(); |
57 | | - }); |
| 34 | + |
| 35 | + collection['tableService']['listTablesSegmented'][region][resourceId].data = tables; |
| 36 | + |
| 37 | + async.eachLimit(tables, 10, async(tableName, tableCb) => { |
| 38 | + const tableId = `${resourceId}/tableService/${tableName}`; |
| 39 | + collection['tableService']['getTableAcl'][region][tableId] = {}; |
| 40 | + |
| 41 | + try { |
| 42 | + const aclResponse = await serviceClient.getAccessPolicy(tableName); |
| 43 | + collection['tableService']['getTableAcl'][region][tableId].data = aclResponse; |
| 44 | + } catch (getErr) { |
| 45 | + collection['tableService']['getTableAcl'][region][tableId].err = getErr.message || getErr; |
| 46 | + } |
| 47 | + |
| 48 | + tableCb(); |
| 49 | + }, sCb); |
| 50 | + } catch (tableErr) { |
| 51 | + collection['tableService']['listTablesSegmented'][region][resourceId].err = tableErr.message || tableErr; |
| 52 | + sCb(); |
| 53 | + } |
| 54 | + })(); |
| 55 | + }, cb); |
| 56 | + }, callback); |
58 | 57 | }; |
0 commit comments