feat: update S3 key handling for fleet agent downloads

- Modified the S3 key construction in getFleetAgent functions to use specific package filenames for macOS and Windows, improving clarity and ensuring the correct files are retrieved.
- Introduced constants for package filenames to enhance maintainability and readability of the code.

Author: Marfuen

apps/app/src/app/s3.ts

@@ -137,14 +137,15 @@ export function extractS3KeyFromUrl(url: string): string {
 
 export async function getFleetAgent({ os }: { os: 'macos' | 'windows' | 'linux' }) {
   const fleetBucketName = process.env.FLEET_AGENT_BUCKET_NAME;
+  const fleetAgentFileName = 'Comp AI Agent-1.0.0-arm64.dmg';
 
   if (!fleetBucketName) {
     throw new Error('FLEET_AGENT_BUCKET_NAME is not defined.');
   }
 
   const getFleetAgentCommand = new GetObjectCommand({
     Bucket: fleetBucketName,
-    Key: `${os}/fleet-osquery.pkg`,
+    Key: `${os}/${fleetAgentFileName}`,
   });
 
   const response = await s3Client.send(getFleetAgentCommand);

apps/portal/src/app/api/download-agent/route.ts

@@ -92,7 +92,9 @@ export async function GET(req: NextRequest) {
 
     // Get package from S3 and stream it
     const packageFilename = getPackageFilename(os);
-    const packageKey = `${os}/fleet-osquery.${os === 'macos' ? 'pkg' : 'msi'}`;
+    const macosPackageFilename = 'Comp AI Agent-1.0.0-arm64.dmg';
+    const windowsPackageFilename = 'fleet-osquery.msi';
+    const packageKey = `${os}/${os === 'macos' ? macosPackageFilename : windowsPackageFilename}`;
 
     const getObjectCommand = new GetObjectCommand({
       Bucket: fleetBucketName,

apps/portal/src/utils/s3.ts

@@ -132,9 +132,12 @@ export async function getFleetAgent({ os }: { os: 'macos' | 'windows' }) {
       throw new Error(`Unsupported OS: ${os}`);
   }
 
+  const macosPackageFilename = 'Comp AI Agent-1.0.0-arm64.dmg';
+  const windowsPackageFilename = 'fleet-osquery.msi';
+
   const getFleetAgentCommand = new GetObjectCommand({
     Bucket: fleetBucketName,
-    Key: `${os}/fleet-osquery.${extension}`,
+    Key: `${os}/${os === 'macos' ? macosPackageFilename : windowsPackageFilename}`,
   });
 
   const response = await s3Client.send(getFleetAgentCommand);