LambdaでWikiを更新する実装と、EventBridgeで“置いていかれる”のを防ぐ設定【後編】

import json
import boto3
import csv
import io
import urllib.request
import urllib.parse
from datetime import datetime, timezone, timedelta
import os

# JST タイムゾーン
JST = timezone(timedelta(hours=9))


def get_backlog_api_key(ssm_parameter_name):
    """
    Systems Manager Parameter StoreからBacklog APIキーを取得する
    """
    ssm = boto3.client('ssm')
    try:
        print(f"Getting API key from Parameter Store: {ssm_parameter_name}")
        response = ssm.get_parameter(
            Name=ssm_parameter_name,
            WithDecryption=True  # SecureStringの場合は復号化
        )
        api_key = response['Parameter']['Value']
        # デバッグ用: APIキーの先頭と末尾の一部を表示（セキュリティのため全体は表示しない）
        print(f"API key retrieved: {api_key[:4]}...{api_key[-4:]} (length: {len(api_key)})")
        return api_key
    except Exception as e:
        print(f"Error getting API key from Parameter Store: {e}")
        raise


def get_csv_from_s3(bucket, key):
    """
    S3からCSVファイルを取得する
    """
    s3 = boto3.client('s3')
    try:
        response = s3.get_object(Bucket=bucket, Key=key)
        csv_content = response['Body'].read().decode('utf-8')
        return csv_content
    except Exception as e:
        print(f"Error getting CSV from S3: {e}")
        raise


def parse_instance_summary_csv(csv_content):
    """
    Patch Managerのインスタンスサマリー CSV を解析する

    CSVは1つのセクションで構成:
    - 1行目: ヘッダー
    - 2行目以降: 各インスタンスのサマリーデータ

    戻り値: インスタンスサマリーのリスト
    """
    instances = []

    reader = csv.DictReader(io.StringIO(csv_content))
    for row in reader:
        # 空行をスキップ
        if not row.get('Instance ID'):
            continue
        instances.append(row)

    print(f"Parsed {len(instances)} instances from CSV")
    return instances


def detect_os_from_platform(platform_name):
    """
    Platform name からOS種別を判定する

    判定ルール:
    - 'Windows' が含まれる → windows
    - それ以外（Amazon Linux, Red Hat, Ubuntu等） → linux
    """
    if not platform_name:
        return None

    if 'Windows' in platform_name:
        return 'windows'
    else:
        return 'linux'


def get_instance_patches(instance_id):
    """
    SSM APIでインスタンスのパッチ詳細を取得する

    describe_instance_patches APIを使用して、
    指定インスタンスのパッチ一覧を取得する
    """
    ssm = boto3.client('ssm')
    patches = []
    next_token = None

    try:
        while True:
            params = {
                'InstanceId': instance_id,
                'MaxResults': 50
            }
            if next_token:
                params['NextToken'] = next_token

            response = ssm.describe_instance_patches(**params)
            patches.extend(response.get('Patches', []))

            next_token = response.get('NextToken')
            if not next_token:
                break

        print(f"  Retrieved {len(patches)} patches for {instance_id}")
        return patches

    except Exception as e:
        print(f"  Error getting patches for {instance_id}: {e}")
        return []


def filter_missing_patches(patches):
    """
    Missingステータスのパッチのみをフィルタする
    """
    return [p for p in patches if p.get('State') == 'Missing']


def convert_summaries_to_backlog_table(summaries):
    """
    複数インスタンスのサマリーをBacklog Wiki形式の表に変換する
    """
    if not summaries:
        return "データがありません。"

    # 表示するカラムを定義（実際のCSVカラム名に合わせる）
    display_columns = [
        ('Instance ID', 'インスタンスID'),
        ('Instance name', 'インスタンス名'),
        ('Instance IP', 'IP'),
        ('Platform name', 'プラットフォーム'),
        ('Compliance status', 'コンプライアンス状態'),
        ('Compliance severity', '重要度'),
        ('Noncompliant Critical severity patch count', 'Critical'),
        ('Noncompliant High severity patch count', 'High'),
        ('Noncompliant Medium severity patch count', 'Medium'),
        ('Noncompliant Low severity patch count', 'Low'),
    ]

    # ヘッダー行を作成
    header_row = '| ' + ' | '.join([jp_name for _, jp_name in display_columns]) + ' |h'

    # 各インスタンスのデータ行を作成
    data_rows = []
    for summary in summaries:
        cells = []
        for en_name, _ in display_columns:
            value = summary.get(en_name, '-')
            # コンプライアンス状態に応じて色付け
            if en_name == 'Compliance status':
                if value == 'COMPLIANT':
                    value = '&color(green){COMPLIANT}'
                elif value == 'NON_COMPLIANT':
                    value = '&color(red){NON_COMPLIANT}'
                elif not value:
                    value = '&color(gray){未スキャン}'
            # プラットフォーム名にバージョンを結合
            if en_name == 'Platform name':
                platform_version = summary.get('Platform version', '')
                if platform_version:
                    if 'Windows' in str(value):
                        # Windowsの場合はビルド番号なので括弧付きで表示
                        value = f"{value} ({platform_version})"
                    else:
                        # Linuxの場合はそのまま結合
                        value = f"{value} {platform_version}"
            cells.append(str(value) if value else '-')
        data_rows.append('| ' + ' | '.join(cells) + ' |')

    return header_row + '\n' + '\n'.join(data_rows)


def convert_patches_to_backlog_table(patches):
    """
    パッチ詳細をBacklog Wiki形式の表に変換する
    SSM APIから取得したパッチ情報を変換
    """
    if not patches:
        return "更新対象のパッチはありません。"

    # 表示するカラムを定義（SSM APIのレスポンスに合わせる）
    display_columns = [
        ('Title', 'パッチ名'),
        ('KBId', 'KB/Patch ID'),
        ('State', '状態'),
        ('Severity', '重要度'),
        ('Classification', '分類'),
        ('CVEIds', 'CVE ID'),
    ]

    # ヘッダー行を作成
    header_row = '| ' + ' | '.join([jp_name for _, jp_name in display_columns]) + ' |h'

    # データ行を作成
    data_rows = []
    for patch in patches:
        cells = []
        for en_name, _ in display_columns:
            value = patch.get(en_name, '-')

            # CVEIdsはリストなのでカンマ区切りに変換
            if en_name == 'CVEIds' and isinstance(value, list):
                value = ', '.join(value) if value else '-'

            # 状態に応じて色付け
            if en_name == 'State':
                if value == 'Missing':
                    value = '&color(red){Missing}'
                elif value == 'Installed':
                    value = '&color(green){Installed}'
                elif value == 'InstalledOther':
                    value = '&color(blue){InstalledOther}'
            # 重要度に応じて色付け
            if en_name == 'Severity':
                if value == 'Critical':
                    value = '&color(red){Critical}'
                elif value == 'Important':
                    value = '&color(orange){Important}'
            cells.append(str(value) if value else '-')
        data_rows.append('| ' + ' | '.join(cells) + ' |')

    return header_row + '\n' + '\n'.join(data_rows)


def create_wiki_content(instances_data, os_type, csv_file):
    """
    Wikiページのコンテンツ全体を作成する

    instances_data: [{'summary': {...}, 'patches': [...]}, ...]

    構成:
    - 目次（#contents）
    - 全インスタンスのサマリー表
    - NON_COMPLIANTインスタンスのみパッチ詳細を表示
    """
    now = datetime.now(JST).strftime('%Y-%m-%d %H:%M:%S JST')

    # サマリー一覧を抽出
    summaries = [inst['summary'] for inst in instances_data]

    # 統計情報を計算
    total_count = len(instances_data)
    compliant_count = sum(1 for inst in instances_data if inst['summary'].get('Compliance status') == 'COMPLIANT')
    non_compliant_count = sum(1 for inst in instances_data if inst['summary'].get('Compliance status') == 'NON_COMPLIANT')
    unscanned_count = total_count - compliant_count - non_compliant_count

    # パッチ詳細セクションを生成（NON_COMPLIANTのインスタンスのみ）
    patch_sections = []
    for inst in instances_data:
        summary = inst['summary']
        patches = inst.get('patches', [])
        compliance_status = summary.get('Compliance status', '')

        # NON_COMPLIANTのインスタンスのみ詳細を表示
        if compliance_status != 'NON_COMPLIANT':
            continue

        instance_name = summary.get('Instance name', 'Unknown')
        instance_id = summary.get('Instance ID', 'Unknown')

        # Missingのパッチのみフィルタ
        missing_patches = filter_missing_patches(patches)

        section_content = f"""*** {instance_name}（{instance_id}）
未適用パッチ: {len(missing_patches)}件

{convert_patches_to_backlog_table(missing_patches)}"""

        patch_sections.append(section_content)

    # OS種別の表示名
    os_display_name = 'Linux' if os_type == 'linux' else 'Windows'

    # パッチ詳細セクションがない場合のメッセージ
    if patch_sections:
        patch_detail_content = '\n\n'.join(patch_sections)
    else:
        patch_detail_content = "すべてのインスタンスがコンプライアンス準拠状態です。未適用のパッチはありません。"

    content = f"""* Patch Manager コンプライアンスレポート（{os_display_name}）

#contents

** 概要
このページはAWS Systems Manager Patch Managerのパッチコンプライアンスレポートを自動更新しています。

- 最終更新日時: {now}
- 対象インスタンス数: {total_count}台
-- &color(green){{COMPLIANT}}: {compliant_count}台
-- &color(red){{NON_COMPLIANT}}: {non_compliant_count}台
-- 未スキャン: {unscanned_count}台
- ソースファイル: {csv_file}

** インスタンス一覧

{convert_summaries_to_backlog_table(summaries)}

** パッチ詳細（NON_COMPLIANTのみ）

{patch_detail_content}

** 凡例

*** コンプライアンス状態
- &color(green){{COMPLIANT}}: パッチが適用されており、コンプライアンス準拠
- &color(red){{NON_COMPLIANT}}: 未適用のパッチがあり、コンプライアンス非準拠
- &color(gray){{未スキャン}}: パッチベースラインが設定されていない、またはスキャン未実施
- Critical/High/Medium/Low: 各重要度の未適用（Missing）パッチ数

*** パッチ状態
- &color(red){{Missing}}: 未適用のパッチ（適用が必要）
- &color(green){{Installed}}: 正常にインストール済みのパッチ
- &color(blue){{InstalledOther}}: ベースライン外でインストールされたパッチ
- NotApplicable: このインスタンスには適用対象外のパッチ

*** パッチ重要度
- &color(red){{Critical}}: 緊急度が最も高いパッチ
- &color(orange){{Important}}: 重要なセキュリティパッチ
- Moderate: 中程度の重要度
- Low: 低い重要度

** 更新履歴
- {now}: 自動更新（Lambda関数による）
"""
    return content


def update_backlog_wiki(api_key, backlog_host, wiki_id, content):
    """
    Backlog WikiページをAPIで更新する
    """
    # APIキーはクエリパラメータに含める
    url = f"https://{backlog_host}/api/v2/wikis/{wiki_id}?apiKey={api_key}"
    print(f"Updating wiki ID: {wiki_id}")

    params = {
        'content': content
    }

    data = urllib.parse.urlencode(params).encode('utf-8')

    req = urllib.request.Request(url, data=data, method='PATCH')
    req.add_header('Content-Type', 'application/x-www-form-urlencoded')

    try:
        with urllib.request.urlopen(req) as response:
            result = json.loads(response.read().decode('utf-8'))
            print(f"Wiki updated successfully: {result.get('name')}")
            return result
    except urllib.error.HTTPError as e:
        error_body = e.read().decode('utf-8')
        print(f"Error updating wiki: {e.code} - {error_body}")
        print(f"Wiki ID: {wiki_id}")
        raise


def get_latest_csv_from_s3(bucket, prefix):
    """
    S3から最新のCSVファイルを1つだけ取得する

    LastModifiedが最新のCSVファイルを返す。
    コンプライアンスレポートが複数回エクスポートされて
    サフィックスが付いた場合でも、最新のものだけを処理する。
    """
    s3 = boto3.client('s3')

    try:
        response = s3.list_objects_v2(
            Bucket=bucket,
            Prefix=prefix
        )

        if 'Contents' not in response:
            raise Exception(f"No CSV files found in s3://{bucket}/{prefix}")

        # CSVファイルのみをフィルタし、LastModifiedでソート
        csv_objects = [
            obj for obj in response['Contents'] 
            if obj['Key'].endswith('.csv')
        ]

        if not csv_objects:
            raise Exception(f"No CSV files found in s3://{bucket}/{prefix}")

        # LastModifiedで降順ソートして最新のものを取得
        csv_objects.sort(key=lambda x: x['LastModified'], reverse=True)
        latest_csv = csv_objects[0]

        print(f"Found {len(csv_objects)} CSV files, using latest: {latest_csv['Key']} (LastModified: {latest_csv['LastModified']})")

        return latest_csv['Key']

    except Exception as e:
        print(f"Error listing S3 objects: {e}")
        raise


def get_wiki_id_for_os(os_type):
    """
    OS種別に対応するWiki IDを環境変数から取得する

    環境変数:
    - BACKLOG_WIKI_ID_LINUX: Linux用WikiページID
    - BACKLOG_WIKI_ID_WINDOWS: Windows用WikiページID
    """
    if os_type == 'linux':
        wiki_id = os.environ.get('BACKLOG_WIKI_ID_LINUX')
        if wiki_id:
            return wiki_id
        raise Exception("BACKLOG_WIKI_ID_LINUX environment variable is required for Linux instances")
    elif os_type == 'windows':
        wiki_id = os.environ.get('BACKLOG_WIKI_ID_WINDOWS')
        if wiki_id:
            return wiki_id
        raise Exception("BACKLOG_WIKI_ID_WINDOWS environment variable is required for Windows instances")

    return None


def process_instances_by_os(api_key, backlog_host, instances, csv_file):
    """
    インスタンスをOS種別ごとにグループ化して処理し、Wikiを更新する

    1. Platform nameからOS種別を判定
    2. NON_COMPLIANTのインスタンスのみパッチ詳細をAPIで取得
    3. OS別にWikiを更新
    """
    # OS種別ごとにインスタンスをグループ化
    os_groups = {
        'linux': [],
        'windows': []
    }

    for instance in instances:
        platform_name = instance.get('Platform name', '')
        os_type = detect_os_from_platform(platform_name)

        if os_type is None:
            print(f"Skipping instance {instance.get('Instance ID')}: OS type not detected from '{platform_name}'")
            continue

        os_groups[os_type].append(instance)

    results = []

    # OS種別ごとに処理
    for os_type, group_instances in os_groups.items():
        if not group_instances:
            print(f"No {os_type} instances found")
            continue

        wiki_id = get_wiki_id_for_os(os_type)
        print(f"Processing {len(group_instances)} {os_type} instances -> Wiki ID: {wiki_id}")

        # 各インスタンスのデータを準備
        instances_data = []
        for instance in group_instances:
            instance_data = {
                'summary': instance,
                'patches': []
            }

            # NON_COMPLIANTの場合のみパッチ詳細を取得
            compliance_status = instance.get('Compliance status', '')
            if compliance_status == 'NON_COMPLIANT':
                instance_id = instance.get('Instance ID')
                print(f"  Fetching patches for NON_COMPLIANT instance: {instance.get('Instance name')} ({instance_id})")
                patches = get_instance_patches(instance_id)
                instance_data['patches'] = patches

            instances_data.append(instance_data)

        # Wikiコンテンツを生成
        wiki_content = create_wiki_content(instances_data, os_type, csv_file)

        # Backlog Wikiを更新
        try:
            result = update_backlog_wiki(api_key, backlog_host, wiki_id, wiki_content)
            results.append({
                'os_type': os_type,
                'wiki_id': result.get('id'),
                'wiki_name': result.get('name'),
                'instances_count': len(group_instances),
                'non_compliant_count': sum(1 for inst in group_instances if inst.get('Compliance status') == 'NON_COMPLIANT')
            })
        except Exception as e:
            print(f"Error updating {os_type} wiki: {e}")
            results.append({
                'os_type': os_type,
                'error': str(e),
                'instances_count': len(group_instances)
            })

    return results


def lambda_handler(event, context):
    """
    Lambda関数のメインハンドラー

    S3バケット内の最新CSVファイル（インスタンスサマリー）を処理し、
    Platform name からOS種別を判定して対応するWikiページを更新する

    - 最新のCSVファイル（LastModified基準）のみを処理
    - 1つのCSVに複数インスタンス（Linux/Windows混在）対応
    - NON_COMPLIANTのインスタンスのみSSM APIでパッチ詳細を取得
    - OS別にWikiを更新（Linux用、Windows用）

    必要な環境変数:
    - BACKLOG_HOST: Backlogのホスト名（例: aws-plus.backlog.jp）
    - BACKLOG_WIKI_ID_LINUX: Linux用WikiページID
    - BACKLOG_WIKI_ID_WINDOWS: Windows用WikiページID
    - SSM_PARAMETER_NAME: APIキーのパラメータストアパス
    - S3_BUCKET: CSVが格納されているS3バケット名
    - S3_KEY_PREFIX: CSVのプレフィックス（オプション）
    """
    print(f"Event received: {json.dumps(event)}")

    # 環境変数を取得
    backlog_host = os.environ.get('BACKLOG_HOST')
    ssm_parameter_name = os.environ.get('SSM_PARAMETER_NAME')
    s3_bucket = os.environ.get('S3_BUCKET')
    s3_key_prefix = os.environ.get('S3_KEY_PREFIX', '')

    # 必須環境変数のチェック
    if not backlog_host:
        raise Exception("BACKLOG_HOST environment variable is required")
    if not ssm_parameter_name:
        raise Exception("SSM_PARAMETER_NAME environment variable is required")
    if not s3_bucket:
        raise Exception("S3_BUCKET environment variable is required")

    try:
        # 1. Backlog APIキーを取得
        api_key = get_backlog_api_key(ssm_parameter_name)
        print("Successfully retrieved Backlog API key from Parameter Store")

        # 2. S3から最新のCSVファイルを取得
        csv_key = get_latest_csv_from_s3(s3_bucket, s3_key_prefix)
        csv_filename = csv_key.split('/')[-1]

        # 3. CSVを取得してパース
        print(f"\nProcessing CSV: {csv_key}")
        csv_content = get_csv_from_s3(s3_bucket, csv_key)
        instances = parse_instance_summary_csv(csv_content)

        if not instances:
            raise Exception(f"No instances found in {csv_key}")

        # 4. OS種別ごとに処理してWikiを更新
        results = process_instances_by_os(api_key, backlog_host, instances, csv_filename)

        return {
            'statusCode': 200,
            'body': json.dumps({
                'message': f'Processed {csv_filename}, updated {len(results)} wiki pages',
                'csv_file': csv_filename,
                'results': results
            }, ensure_ascii=False)
        }

    except Exception as e:
        print(f"Error in lambda_handler: {e}")
        return {
            'statusCode': 500,
            'body': json.dumps({
                'error': str(e)
            }, ensure_ascii=False)
        }