feat(cli-repl): add configuration to enable log compression

Author: gagik

packages/cli-repl/src/cli-repl.spec.ts

@@ -324,6 +324,7 @@ describe('CliRepl', function () {
           'disableLogging',
           'logLocation',
           'logRetentionDays',
+          'logCompression',
         ] satisfies (keyof CliUserConfig)[]);
       });
 
@@ -1457,6 +1458,14 @@ describe('CliRepl', function () {
               testRetentionDays
             );
           });
+
+          it('can set log compression', async function () {
+            cliRepl.config.logCompression = true;
+            await cliRepl.start(await testServer.connectionString(), {});
+
+            expect(await cliRepl.getConfig('logCompression')).equals(true);
+            expect(cliRepl.logManager?._options.gzip).equals(true);
+          });
         });
 
         it('times out fast', async function () {

packages/cli-repl/src/cli-repl.ts

@@ -260,6 +260,7 @@ export class CliRepl implements MongoshIOProvider {
         (await this.getConfig('logLocation')) ||
         this.shellHomeDirectory.localPath('.'),
       retentionDays: await this.getConfig('logRetentionDays'),
+      gzip: await this.getConfig('logCompression'),
       maxLogFileCount: +(
         process.env.MONGOSH_TEST_ONLY_MAX_LOG_FILE_COUNT || 100
       ),

packages/e2e-tests/test/e2e.spec.ts

@@ -1691,6 +1691,49 @@ describe('e2e', function () {
           ).join('');
         };
 
+        describe('with custom log compression', function () {
+          const customLogDir = useTmpdir();
+
+          it('should created compressed files when enabled', async function () {
+            const globalConfig = path.join(homedir, 'globalconfig.conf');
+            await fs.writeFile(
+              globalConfig,
+              `mongosh:\n  logLocation: ${JSON.stringify(
+                customLogDir.path
+              )}\n  logCompression: true`
+            );
+
+            shell = this.startTestShell({
+              args: ['--nodb'],
+              env: {
+                ...env,
+                MONGOSH_GLOBAL_CONFIG_FILE_FOR_TESTING: globalConfig,
+              },
+              forceTerminal: true,
+            });
+
+            await shell.waitForPrompt();
+
+            const logFile = path.join(
+              customLogDir.path,
+              `${shell.logId as string}_log`
+            );
+            const logFileGzip = path.join(
+              customLogDir.path,
+              `${shell.logId as string}_log.gz`
+            );
+
+            // Only the gzipped file should exist
+            expect(await getFilesState([logFile, logFileGzip])).equals('01');
+
+            const logContent = await fs.readFile(logFileGzip);
+
+            // gzipped files start with 0x1f 0x8b
+            expect(logContent[0]).equals(0x1f);
+            expect(logContent[1]).equals(0x8b);
+          });
+        });
+
         describe('with custom log retention days', function () {
           const customLogDir = useTmpdir();
 

packages/types/src/index.spec.ts

@@ -31,6 +31,13 @@ describe('config validation', function () {
     expect(await validate('logRetentionDays', -1)).to.equal(
       'logRetentionDays must be a positive integer'
     );
+    expect(await validate('logCompression', 'foo')).to.equal(
+      'logCompression must be a boolean'
+    );
+    expect(await validate('logCompression', -1)).to.equal(
+      'logCompression must be a boolean'
+    );
+    expect(await validate('logCompression', false)).to.equal(null);
     expect(await validate('showStackTraces', 'foo')).to.equal(
       'showStackTraces must be a boolean'
     );

packages/types/src/index.ts

@@ -510,6 +510,7 @@ export class CliUserConfig extends SnippetShellUserConfig {
   disableLogging = false;
   logLocation: string | undefined = undefined;
   logRetentionDays = 30;
+  logCompression = false;
 }
 
 export class CliUserConfigValidator extends SnippetShellUserConfigValidator {
@@ -540,6 +541,7 @@ export class CliUserConfigValidator extends SnippetShellUserConfigValidator {
       case 'disableLogging':
       case 'forceDisableTelemetry':
       case 'showStackTraces':
+      case 'logCompression':
         if (typeof value !== 'boolean') {
           return `${key} must be a boolean`;
         }