feat: display all record counts

Author: iowillhoit

messages/recordcounts.json

@@ -1,6 +1,7 @@
 {
-  "commandDescription": "display record counts for the specified standard and custom objects\nUse this command to get an approximate count of the records in standard or custom objects in your org. These record counts are the same as the counts listed in the Storage Usage page in Setup. The record counts are approximate because they're calculated asynchronously and your org’s storage usage isn’t updated immediately.",
+  "commandDescription": "display record counts for the specified standard and custom objects\nUse this command to get an approximate count of the records in standard or custom objects in your org. These record counts are the same as the counts listed in the Storage Usage page in Setup. The record counts are approximate because they're calculated asynchronously and your org's storage usage isn't updated immediately. Executing without the '--sobjecttype' flag will display all available record counts.",
   "examples": [
+    "sfdx force:limits:recordcounts:display",
     "sfdx force:limits:recordcounts:display -s Account,Contact,Lead,Opportunity",
     "sfdx force:limits:recordcounts:display -s Account,Contact -u [email protected]"
   ],

src/commands/force/limits/api/display.ts

@@ -6,7 +6,7 @@
  */
 import * as os from 'os';
 import { SfdxCommand } from '@salesforce/command';
-import { Messages, SfdxError } from '@salesforce/core';
+import { Messages, SfError } from '@salesforce/core';
 
 Messages.importMessagesDirectory(__dirname);
 const messages = Messages.loadMessages('@salesforce/plugin-limits', 'display');
@@ -48,7 +48,7 @@ export class LimitsApiDisplayCommand extends SfdxCommand {
 
       return limits;
     } catch (err) {
-      throw SfdxError.wrap(err);
+      throw SfError.wrap(err);
     }
   }
 }

src/commands/force/limits/recordcounts/display.ts

@@ -6,7 +6,7 @@
  */
 import * as os from 'os';
 import { flags, FlagsConfig, SfdxCommand, SfdxResult } from '@salesforce/command';
-import { Messages, SfdxError } from '@salesforce/core';
+import { Messages, SfError } from '@salesforce/core';
 
 Messages.importMessagesDirectory(__dirname);
 const messages = Messages.loadMessages('@salesforce/plugin-limits', 'recordcounts');
@@ -39,28 +39,31 @@ export class LimitsRecordCountsDisplayCommand extends SfdxCommand {
   protected static readonly flagsConfig: FlagsConfig = {
     sobjecttype: flags.array({
       char: 's',
-      required: true,
       description: messages.getMessage('sobjecttypeFlagDescription'),
     }),
   };
 
   public async run(): Promise<RecordCount[]> {
     try {
-      const sobjecttypeString = (this.flags.sobjecttype as string[]).join();
+      const sobjectsPassed = this.flags.sobjecttype as string[];
+      const sobjectsQuery = sobjectsPassed ? `=${sobjectsPassed.join()}` : '';
+
       const conn = this.org.getConnection();
-      const geturl = `${conn.baseUrl()}/limits/recordCount?sObjects=${sobjecttypeString}`;
+      const geturl = `${conn.baseUrl()}/limits/recordCount?sObjects${sobjectsQuery}`;
       const result = await conn.request<Result>(geturl);
 
       // if an object is requested, but there's 0 of them on the server, append that object to the result
-      sobjecttypeString.split(',').forEach((name) => {
-        if (!result.sObjects.find((record) => record.name === name)) {
-          result.sObjects.push({ name, count: 0 });
-        }
-      });
+      if (sobjectsPassed) {
+        sobjectsPassed.forEach((name) => {
+          if (!result.sObjects.find((record) => record.name === name)) {
+            result.sObjects.push({ name, count: 0 });
+          }
+        });
+      }
 
       return result.sObjects;
     } catch (err) {
-      throw SfdxError.wrap(err);
+      throw SfError.wrap(err);
     }
   }
 }

test/commands/recordcounts.nut.ts

@@ -38,6 +38,15 @@ describe('recordcounts:display', () => {
     expect(output.status).to.equal(0);
   });
 
+  it('Displays all records (json)', () => {
+    const output = execCmd<RecordCount[]>(`force:limits:recordcounts:display -u ${username} --json`, {
+      ensureExitCode: 0,
+    }).jsonOutput;
+
+    expect(output.result).length.greaterThan(10);
+    expect(output.status).to.equal(0);
+  });
+
   it('Displays the records (human readable)', () => {
     const command = `force:limits:recordcounts:display -s Account -u ${username}`;
     const result = execCmd(command, { ensureExitCode: 0 });