import {
  SSMClient,
  SendCommandCommand,
  GetCommandInvocationCommand
} from "@aws-sdk/client-ssm";

import {
  BedrockRuntimeClient,
  InvokeModelCommand
} from "@aws-sdk/client-bedrock-runtime";

/* =====================
   環境設定（ダミー）
   ===================== */
// 実環境では環境変数や
// Zabbixホスト名 → EC2タグ解決などを想定
const REGION = "<AWS_REGION>";
const INSTANCE_ID = "<TARGET_EC2_INSTANCE_ID>";

const ssm = new SSMClient({ region: REGION });
const bedrock = new BedrockRuntimeClient({ region: REGION });

export const handler = async (event) => {
  console.log("RAW EVENT:", JSON.stringify(event, null, 2));

  /* =====================
     0. Zabbix Webhook 受信
     ===================== */
  // Zabbixのトリガーアクションから
  // Webhookで送信された内容をパース
  const zabbixEvent = JSON.parse(event.body);

  /*
    zabbixEvent 例：
    {
      host: "<TARGET_HOST>",
      trigger: "Linux: FS [/]: Space is low",
      severity: "Warning",
      status: "PROBLEM",
      item_value: "89.5 %",
      time: "2025-xx-xx xx:xx:xx"
    }
  */
  console.log("ZABBIX EVENT:", zabbixEvent);

  /* =====================
     1. SSM RunCommand
     ===================== */

const commands = [
  "echo '### FILESYSTEM USAGE ###'",
  "df -h /",

  "echo '### TOP DIRECTORIES UNDER / ###'",
  "du -xh / --max-depth=2 | sort -h | tail -15",

  "echo '### TOP DIRECTORIES UNDER /var ###'",
  "du -xh /var --max-depth=2 | sort -h | tail -15",

  "echo '### LARGE FILES (>100MB) UNDER /var ###'",
  "find /var -type f -size +100M -exec ls -lh {} \\; 2>/dev/null | sort -k5 -h | tail -10",

  "echo '### RECENTLY UPDATED LARGE FILES ###'",
  "find /var -type f -size +100M -mmin -1440 -exec ls -lh {} \\; 2>/dev/null | sort -k6,7"
];


  const sendResult = await ssm.send(
    new SendCommandCommand({
      InstanceIds: [INSTANCE_ID],
      DocumentName: "AWS-RunShellScript",
      Parameters: { commands }
    })
  );

  const commandId = sendResult.Command.CommandId;
  console.log("SSM CommandId:", commandId);

  // 実行完了を最低限待機
  await new Promise(r => setTimeout(r, 5000));

  const invocation = await ssm.send(
    new GetCommandInvocationCommand({
      CommandId: commandId,
      InstanceId: INSTANCE_ID
    })
  );

  const ssmOutput = invocation.StandardOutputContent;
  console.log("=== SSM OUTPUT ===\n", ssmOutput);

  /* =====================
     2. Bedrock（Claude 3 Haiku）
     ===================== */
  // 取得した調査結果を
  // そのまま「報告文」に変換
  const prompt = `
あなたは熟練したインフラエンジニアです。
以下はディスク逼迫アラートとサーバ調査結果です。

【Zabbix Alert】
- Host: ${zabbixEvent.host}
- Trigger: ${zabbixEvent.trigger}
- Severity: ${zabbixEvent.severity}
- Status: ${zabbixEvent.status}
- Item value: ${zabbixEvent.item_value}
- Time: ${zabbixEvent.time}

【SSM 調査結果】
${ssmOutput}

以下を日本語で簡潔にまとめてください。
1. 障害の概要
2. 想定される原因
3. 影響範囲
4. 即時対応案
5. 恒久対応案
`;

  const bedrockResponse = await bedrock.send(
    new InvokeModelCommand({
      modelId: "anthropic.claude-3-haiku-20240307-v1:0",
      contentType: "application/json",
      accept: "application/json",
      body: JSON.stringify({
        anthropic_version: "bedrock-2023-05-31",
        max_tokens: 600,
        messages: [
          {
            role: "user",
            content: prompt
          }
        ]
      })
    })
  );

  const decoded = JSON.parse(
    Buffer.from(bedrockResponse.body).toString("utf-8")
  );

  const report = decoded.content?.[0]?.text ?? "No output";

  console.log("=== AI REPORT ===\n", report);

  /* =====================
     3. レスポンス返却
     ===================== */
  // 実運用では：
  // ・Slack通知
  // ・チケット起票
  // ・メール本文生成
  // などに拡張可能
  return {
    statusCode: 200,
    body: JSON.stringify({
      status: "ok",
      report
    })
  };
};

 1. 障害の概要:
アプリケーションサーバ "app-server-01" のファイルシステム "/" の使用率が 83.37% と高く、
ディスク容量が不足しているという警告が発生しています。

2. 想定される原因:
ログファイルの肥大化が主な原因と考えられます。
"/var/log/httpd/access_log" ファイルが 5.0 GB と非常に大きくなっています。
最近更新された大きなファイルも見られることから、
ログファイルが急激に増大している可能性があります。

3. 影響範囲:
ファイルシステムの使用率が高いため、サーバ上の他のアプリケーションやサービスの動作に支障をきたす可能性があります。

4. 即時対応案:
　1. ログファイルの整理を行う
  　 - "/var/log/httpd/access_log" ファイルをアーカイブするか削除する
  　 - その他の大きなログファイルも確認し、必要に応じて対処する
　2. ファイルシステムの使用率を監視し、再びしきい値を超えないよう注意する

5. 恒久対応案:
　1. ログファイルの自動削除/アーカイブ設定を行う
   　- logrotate などを使ってログファイルの自動管理を設定する
　2. ファイルシステムのサイズ拡張を検討する
  　 - ディスク容量の増設や別ディスクの追加など、ファイルシステムの容量を拡張する