Merge

Author: jcobis

AUTHORS

@@ -116,3 +116,4 @@ Leo Generali <leo.d.generali@gmail.com>
 Ivan Medina <medinadiazivan@gmail.com>
 Nick Larew <nick.larew@10gen.com>
 Shun Miyazawa <34241526+miya@users.noreply.github.com>
+Kyle Lai <122811196+kylelai1@users.noreply.github.com>

package-lock.json

@@ -68,7 +68,6 @@
     "mongodb-collection-model": "^5.35.3",
     "mongodb-ns": "^3.0.1",
     "mongodb-schema": "^12.6.3",
-    "numeral": "^2.0.6",
     "react": "^17.0.2",
     "react-redux": "^8.1.3",
     "redux": "^4.2.1",

packages/compass-collection/package.json

@@ -1,3 +1,4 @@
+import type { MongoDBFieldType } from '@mongodb-js/compass-generative-ai';
 import { MockDataGeneratorStep } from './types';
 
 export const StepButtonLabelMap = {
@@ -10,3 +11,91 @@ export const StepButtonLabelMap = {
 
 export const DEFAULT_DOCUMENT_COUNT = 1000;
 export const MAX_DOCUMENT_COUNT = 100000;
+
+/**
+ * Map of MongoDB types to available Faker v9 methods.
+ * Not all Faker methods are included here.
+ * More can be found in the Faker.js API: https://v9.fakerjs.dev/api/
+ */
+export const MONGO_TYPE_TO_FAKER_METHODS: Record<
+  MongoDBFieldType,
+  Array<string>
+> = {
+  String: [
+    'lorem.word',
+    'lorem.words',
+    'lorem.sentence',
+    'lorem.paragraph',
+    'person.firstName',
+    'person.lastName',
+    'person.fullName',
+    'person.jobTitle',
+    'internet.displayName',
+    'internet.email',
+    'internet.emoji',
+    'internet.password',
+    'internet.url',
+    'internet.domainName',
+    'internet.userName',
+    'phone.number',
+    'location.city',
+    'location.country',
+    'location.streetAddress',
+    'location.zipCode',
+    'location.state',
+    'company.name',
+    'company.catchPhrase',
+    'color.human',
+    'commerce.productName',
+    'commerce.department',
+    'finance.accountName',
+    'finance.currencyCode',
+    'git.commitSha',
+    'string.uuid',
+    'string.alpha',
+    'string.alphanumeric',
+    'system.fileName',
+    'system.filePath',
+    'system.mimeType',
+    'book.title',
+    'music.songName',
+    'food.dish',
+    'animal.type',
+    'vehicle.model',
+    'hacker.phrase',
+    'science.chemicalElement',
+  ],
+  Number: [
+    'number.binary',
+    'number.octal',
+    'number.hex',
+    'commerce.price',
+    'date.weekday',
+    'internet.port',
+    'number.int',
+    'number.float',
+    'finance.amount',
+    'location.latitude',
+    'location.longitude',
+  ],
+  Int32: ['number.int', 'finance.amount'],
+  Long: ['number.int', 'number.bigInt'],
+  Decimal128: ['number.float', 'finance.amount'],
+  Boolean: ['datatype.boolean'],
+  Date: [
+    'date.recent',
+    'date.past',
+    'date.future',
+    'date.anytime',
+    'date.birthdate',
+  ],
+  Timestamp: ['date.recent', 'date.past', 'date.future', 'date.anytime'],
+  ObjectId: ['database.mongodbObjectId'],
+  Binary: ['string.hexadecimal', 'string.binary'],
+  RegExp: ['lorem.word', 'string.alpha'],
+  Code: ['lorem.sentence', 'lorem.paragraph', 'git.commitMessage'],
+  MinKey: ['number.int'],
+  MaxKey: ['number.int'],
+  Symbol: ['lorem.word', 'string.symbol'],
+  DBRef: ['database.mongodbObjectId'],
+};

packages/compass-collection/src/components/mock-data-generator-modal/constants.ts

@@ -1,5 +1,6 @@
 import {
   Body,
+  compactBytes,
   css,
   palette,
   spacing,
@@ -9,7 +10,6 @@ import React, { useMemo } from 'react';
 import { connect } from 'react-redux';
 import type { CollectionState } from '../../modules/collection-tab';
 import type { SchemaAnalysisState } from '../../schema-analysis-types';
-import numeral from 'numeral';
 import { DEFAULT_DOCUMENT_COUNT, MAX_DOCUMENT_COUNT } from './constants';
 
 const BYTE_PRECISION_THRESHOLD = 1000;
@@ -40,8 +40,8 @@ const boldStyles = css({
 });
 
 const formatBytes = (bytes: number) => {
-  const precision = bytes <= BYTE_PRECISION_THRESHOLD ? '0' : '0.0';
-  return numeral(bytes).format(precision + 'b');
+  const decimals = bytes <= BYTE_PRECISION_THRESHOLD ? 0 : 1;
+  return compactBytes(bytes, true, decimals);
 };
 
 type ErrorState =
@@ -101,7 +101,7 @@ const DocumentCountScreen = ({
     }
   };
 
-  return schemaAnalysisState.status === 'complete' ? (
+  return (
     <div>
       <Body className={titleStyles}>
         Specify Number of Documents to Generate
@@ -130,9 +130,6 @@ const DocumentCountScreen = ({
         </div>
       </div>
     </div>
-  ) : (
-    // Not reachable since schema analysis must be finished before the modal can be opened
-    <div>We are analyzing your collection.</div>
   );
 };