[Fleet] Request diagnostics

x-pack/plugins/fleet/common/constants/routes.ts

@@ -127,6 +127,8 @@ export const AGENT_API_ROUTES = {
   BULK_UNENROLL_PATTERN: `${API_ROOT}/agents/bulk_unenroll`,
   REASSIGN_PATTERN: `${API_ROOT}/agents/{agentId}/reassign`,
   BULK_REASSIGN_PATTERN: `${API_ROOT}/agents/bulk_reassign`,
+  REQUEST_DIAGNOSTICS_PATTERN: `${API_ROOT}/agents/{agentId}/request_diagnostics`,
+  BULK_REQUEST_DIAGNOSTICS_PATTERN: `${API_ROOT}/agents/bulk_request_diagnostics`,
   AVAILABLE_VERSIONS_PATTERN: `${API_ROOT}/agents/available_versions`,
   STATUS_PATTERN: `${API_ROOT}/agent_status`,
   DATA_PATTERN: `${API_ROOT}/agent_status/data`,
@@ -137,6 +139,8 @@ export const AGENT_API_ROUTES = {
   CURRENT_UPGRADES_PATTERN: `${API_ROOT}/agents/current_upgrades`,
   ACTION_STATUS_PATTERN: `${API_ROOT}/agents/action_status`,
   LIST_TAGS_PATTERN: `${API_ROOT}/agents/tags`,
+  LIST_UPLOADS_PATTERN: `${API_ROOT}/agents/{agentId}/uploads`,
+  GET_UPLOAD_FILE_PATTERN: `${API_ROOT}/agents/files/{fileId}/{fileName}`,
 };
 
 export const ENROLLMENT_API_KEY_ROUTES = {

x-pack/plugins/fleet/common/experimental_features.ts

@@ -15,6 +15,7 @@ export const allowedExperimentalValues = Object.freeze({
   createPackagePolicyMultiPageLayout: true,
   packageVerification: true,
   showDevtoolsRequest: true,
+  showRequestDiagnostics: false,
 });
 
 type ExperimentalConfigKeys = Array<keyof ExperimentalFeatures>;

x-pack/plugins/fleet/common/services/routes.ts

@@ -206,6 +206,16 @@ export const agentRouteService = {
     AGENT_API_ROUTES.ACTIONS_PATTERN.replace('{agentId}', agentId),
   getListTagsPath: () => AGENT_API_ROUTES.LIST_TAGS_PATTERN,
   getAvailableVersionsPath: () => AGENT_API_ROUTES.AVAILABLE_VERSIONS_PATTERN,
+  getRequestDiagnosticsPath: (agentId: string) =>
+    AGENT_API_ROUTES.REQUEST_DIAGNOSTICS_PATTERN.replace('{agentId}', agentId),
+  getBulkRequestDiagnosticsPath: () => AGENT_API_ROUTES.BULK_REQUEST_DIAGNOSTICS_PATTERN,
+  getListAgentUploads: (agentId: string) =>
+    AGENT_API_ROUTES.LIST_UPLOADS_PATTERN.replace('{agentId}', agentId),
+  getAgentFileDownloadLink: (fileId: string, fileName: string) =>
+    AGENT_API_ROUTES.GET_UPLOAD_FILE_PATTERN.replace('{fileId}', fileId).replace(
+      '{fileName}',
+      fileName
+    ),
 };
 
 export const outputRoutesService = {

x-pack/plugins/fleet/common/types/models/agent.ts

@@ -37,7 +37,8 @@ export type AgentActionType =
   | 'POLICY_REASSIGN'
   | 'CANCEL'
   | 'FORCE_UNENROLL'
-  | 'UPDATE_TAGS';
+  | 'UPDATE_TAGS'
+  | 'REQUEST_DIAGNOSTICS';
 
 type FleetServerAgentComponentStatusTuple = typeof FleetServerAgentComponentStatuses;
 export type FleetServerAgentComponentStatus = FleetServerAgentComponentStatusTuple[number];
@@ -142,6 +143,15 @@ export interface ActionStatus {
   creationTime: string;
 }
 
+export interface AgentDiagnostics {
+  id: string;
+  name: string;
+  createTime: string;
+  filePath: string;
+  status: 'READY' | 'AWAITING_UPLOAD' | 'DELETED' | 'IN_PROGRESS';
+  actionId: string;
+}
+
 // Generated from FleetServer schema.json
 export interface FleetServerAgentComponentUnit {
   id: string;

x-pack/plugins/fleet/common/types/rest_spec/agent.ts

@@ -7,7 +7,14 @@
 
 import type { SearchHit } from '@kbn/es-types';
 
-import type { Agent, AgentAction, ActionStatus, CurrentUpgrade, NewAgentAction } from '../models';
+import type {
+  Agent,
+  AgentAction,
+  ActionStatus,
+  CurrentUpgrade,
+  NewAgentAction,
+  AgentDiagnostics,
+} from '../models';
 
 import type { ListResult, ListWithKuery } from './common';
 
@@ -38,6 +45,10 @@ export interface GetOneAgentResponse {
   item: Agent;
 }
 
+export interface GetAgentUploadsResponse {
+  items: AgentDiagnostics[];
+}
+
 export interface PostNewAgentActionRequest {
   body: {
     action: Omit<NewAgentAction, 'agents'>;
@@ -121,6 +132,16 @@ export interface PostBulkAgentReassignRequest {
   };
 }
 
+export type PostRequestDiagnosticsResponse = BulkAgentAction;
+export type PostBulkRequestDiagnosticsResponse = BulkAgentAction;
+
+export interface PostRequestBulkDiagnosticsRequest {
+  body: {
+    agents: string[] | string;
+    batchSize?: number;
+  };
+}
+
 export type PostBulkAgentReassignResponse = BulkAgentAction;
 
 export type PostBulkUpdateAgentTagsResponse = BulkAgentAction;

x-pack/plugins/fleet/public/applications/fleet/sections/agents/agent_details_page/components/actions_menu.tsx

@@ -19,6 +19,8 @@ import {
 } from '../../components';
 import { useAgentRefresh } from '../hooks';
 import { isAgentUpgradeable, policyHasFleetServer } from '../../../../services';
+import { AgentRequestDiagnosticsModal } from '../../components/agent_request_diagnostics_modal';
+import { ExperimentalFeaturesService } from '../../../../services';
 
 export const AgentDetailsActionMenu: React.FunctionComponent<{
   agent: Agent;
@@ -32,9 +34,11 @@ export const AgentDetailsActionMenu: React.FunctionComponent<{
   const [isReassignFlyoutOpen, setIsReassignFlyoutOpen] = useState(assignFlyoutOpenByDefault);
   const [isUnenrollModalOpen, setIsUnenrollModalOpen] = useState(false);
   const [isUpgradeModalOpen, setIsUpgradeModalOpen] = useState(false);
+  const [isRequestDiagnosticsModalOpen, setIsRequestDiagnosticsModalOpen] = useState(false);
   const isUnenrolling = agent.status === 'unenrolling';
 
   const hasFleetServer = agentPolicy && policyHasFleetServer(agentPolicy);
+  const { showRequestDiagnostics } = ExperimentalFeaturesService.get();
 
   const onClose = useMemo(() => {
     if (onCancelReassign) {
@@ -44,6 +48,70 @@ export const AgentDetailsActionMenu: React.FunctionComponent<{
     }
   }, [onCancelReassign, setIsReassignFlyoutOpen]);
 
+  const menuItems = [
+    <EuiContextMenuItem
+      icon="pencil"
+      onClick={() => {
+        setIsReassignFlyoutOpen(true);
+      }}
+      disabled={!agent.active}
+      key="reassignPolicy"
+    >
+      <FormattedMessage
+        id="xpack.fleet.agentList.reassignActionText"
+        defaultMessage="Assign to new policy"
+      />
+    </EuiContextMenuItem>,
+    <EuiContextMenuItem
+      icon="trash"
+      disabled={!hasFleetAllPrivileges || !agent.active}
+      onClick={() => {
+        setIsUnenrollModalOpen(true);
+      }}
+    >
+      {isUnenrolling ? (
+        <FormattedMessage
+          id="xpack.fleet.agentList.forceUnenrollOneButton"
+          defaultMessage="Force unenroll"
+        />
+      ) : (
+        <FormattedMessage
+          id="xpack.fleet.agentList.unenrollOneButton"
+          defaultMessage="Unenroll agent"
+        />
+      )}
+    </EuiContextMenuItem>,
+    <EuiContextMenuItem
+      icon="refresh"
+      disabled={!isAgentUpgradeable(agent, kibanaVersion)}
+      onClick={() => {
+        setIsUpgradeModalOpen(true);
+      }}
+    >
+      <FormattedMessage
+        id="xpack.fleet.agentList.upgradeOneButton"
+        defaultMessage="Upgrade agent"
+      />
+    </EuiContextMenuItem>,
+  ];
+
+  if (showRequestDiagnostics) {
+    menuItems.push(
+      <EuiContextMenuItem
+        icon="download"
+        disabled={!hasFleetAllPrivileges}
+        onClick={() => {
+          setIsRequestDiagnosticsModalOpen(true);
+        }}
+      >
+        <FormattedMessage
+          id="xpack.fleet.agentList.diagnosticsOneButton"
+          defaultMessage="Request diagnostics .zip"
+        />
+      </EuiContextMenuItem>
+    );
+  }
+
   return (
     <>
       {isReassignFlyoutOpen && (
@@ -77,6 +145,17 @@ export const AgentDetailsActionMenu: React.FunctionComponent<{
           />
         </EuiPortal>
       )}
+      {isRequestDiagnosticsModalOpen && (
+        <EuiPortal>
+          <AgentRequestDiagnosticsModal
+            agents={[agent]}
+            agentCount={1}
+            onClose={() => {
+              setIsRequestDiagnosticsModalOpen(false);
+            }}
+          />
+        </EuiPortal>
+      )}
       <ContextMenuActions
         button={{
           props: { iconType: 'arrowDown', iconSide: 'right', color: 'primary' },
@@ -87,52 +166,7 @@ export const AgentDetailsActionMenu: React.FunctionComponent<{
             />
           ),
         }}
-        items={[
-          <EuiContextMenuItem
-            icon="pencil"
-            onClick={() => {
-              setIsReassignFlyoutOpen(true);
-            }}
-            disabled={!agent.active}
-            key="reassignPolicy"
-          >
-            <FormattedMessage
-              id="xpack.fleet.agentList.reassignActionText"
-              defaultMessage="Assign to new policy"
-            />
-          </EuiContextMenuItem>,
-          <EuiContextMenuItem
-            icon="trash"
-            disabled={!hasFleetAllPrivileges || !agent.active}
-            onClick={() => {
-              setIsUnenrollModalOpen(true);
-            }}
-          >
-            {isUnenrolling ? (
-              <FormattedMessage
-                id="xpack.fleet.agentList.forceUnenrollOneButton"
-                defaultMessage="Force unenroll"
-              />
-            ) : (
-              <FormattedMessage
-                id="xpack.fleet.agentList.unenrollOneButton"
-                defaultMessage="Unenroll agent"
-              />
-            )}
-          </EuiContextMenuItem>,
-          <EuiContextMenuItem
-            icon="refresh"
-            disabled={!isAgentUpgradeable(agent, kibanaVersion)}
-            onClick={() => {
-              setIsUpgradeModalOpen(true);
-            }}
-          >
-            <FormattedMessage
-              id="xpack.fleet.agentList.upgradeOneButton"
-              defaultMessage="Upgrade agent"
-            />
-          </EuiContextMenuItem>,
-        ]}
+        items={menuItems}
       />
     </>
   );