@exabugs/dynamodb-client 0.1.0

package/dist/scripts/repair-shadows.js

@@ -0,0 +1,190 @@
+#!/usr/bin/env tsx
+/**
+ * シャドーレコード修復スクリプト
+ *
+ * DynamoClient（IAM認証）を使用してレコードを更新し、シャドーレコードを再生成します。
+ *
+ * 使用方法:
+ *   pnpm --filter @exabugs/dynamodb-client tsx scripts/repair-shadows.ts --resource articles --dry-run
+ *   pnpm --filter @exabugs/dynamodb-client tsx scripts/repair-shadows.ts --resource articles --id C8CD6F0D81B2B030696E300752 --repair
+ *   pnpm --filter @exabugs/dynamodb-client tsx scripts/repair-shadows.ts --resource articles --repair
+ */
+import { DynamoClient } from '../client/index.iam.js';
+// 環境変数
+const ENV = process.env.ENV || 'dev';
+const REGION = process.env.AWS_REGION || 'us-east-1';
+// Records Lambda Function URLを取得
+let RECORDS_API_URL = process.env.RECORDS_API_URL || '';
+/**
+ * Records Lambda Function URLを取得
+ */
+async function getRecordsApiUrl() {
+    if (RECORDS_API_URL) {
+        return RECORDS_API_URL;
+    }
+    // AWS CLIで取得
+    const { execSync } = await import('child_process');
+    const functionName = `ainews-${ENV}-records`;
+    try {
+        const output = execSync(`aws lambda get-function-url-config --function-name ${functionName} --region ${REGION} --query 'FunctionUrl' --output text`, { encoding: 'utf-8' });
+        RECORDS_API_URL = output.trim();
+        return RECORDS_API_URL;
+    }
+    catch {
+        throw new Error(`Failed to get Function URL for ${functionName}`);
+    }
+}
+/**
+ * 本体レコードのシャドーキーと実際のシャドーレコードを比較
+ *
+ * 注: Records Lambda は __shadowKeys を HTTP レスポンスから除外するため、
+ * このスクリプトでは __shadowKeys の有無を確認できません。
+ * そのため、すべてのレコードを修復対象とします。
+ */
+async function checkShadowIntegrity() {
+    // Records Lambda は __shadowKeys を除外するため、常に修復対象とする
+    return {
+        needsRepair: true,
+        missingShadows: 0,
+        extraShadows: 0,
+    };
+}
+/**
+ * DynamoClient（IAM認証）を使用してシャドーを再生成
+ */
+async function repairRecordViaDynamoClient(collection, recordId) {
+    // updateOneを呼び出すと、Records Lambdaと同じロジックでシャドーレコードが再生成される
+    await collection.updateOne({ id: recordId }, { set: { updatedAt: new Date().toISOString() } });
+}
+/**
+ * 単一レコードを修復
+ */
+async function repairSingleRecord(collection, recordId, dryRun) {
+    console.log(`\n=== Checking record: ${recordId} ===`);
+    // 本体レコードを取得
+    console.log(`Fetching record with id: ${recordId}`);
+    const mainRecord = await collection.findOne({ id: recordId });
+    console.log(`findOne result:`, mainRecord);
+    if (!mainRecord) {
+        console.error(`Record not found: ${recordId}`);
+        return;
+    }
+    // 整合性チェック
+    const { needsRepair, missingShadows, extraShadows } = await checkShadowIntegrity();
+    // Records Lambdaは__shadowKeysをレスポンスから除外するため、
+    // 実際のシャドー数は確認できません
+    console.log(`Missing shadows: ${missingShadows}`);
+    console.log(`Extra shadows: ${extraShadows}`);
+    if (!needsRepair) {
+        console.log('✓ No repair needed');
+        return;
+    }
+    if (dryRun) {
+        console.log('\n[DRY RUN] Would repair this record via DynamoClient.updateOne()');
+        return;
+    }
+    // DynamoClient経由で修復
+    console.log('\nRepairing via DynamoClient.updateOne() with IAM auth...');
+    await repairRecordViaDynamoClient(collection, recordId);
+    console.log('✓ Repaired successfully');
+}
+/**
+ * 全レコードをスキャンして修復
+ */
+async function repairAllRecords(collection, dryRun) {
+    const stats = {
+        scanned: 0,
+        needsRepair: 0,
+        repaired: 0,
+        failed: 0,
+        errors: [],
+    };
+    console.log(`\n=== Scanning all records ===`);
+    console.log(`Mode: ${dryRun ? 'DRY RUN' : 'REPAIR'}`);
+    // 全レコードを取得
+    const cursor = collection.find({});
+    const records = await cursor.toArray();
+    console.log(`Found ${records.length} records`);
+    for (const mainRecord of records) {
+        stats.scanned++;
+        const recordId = mainRecord.id;
+        try {
+            const { needsRepair, missingShadows, extraShadows } = await checkShadowIntegrity();
+            if (needsRepair) {
+                stats.needsRepair++;
+                console.log(`\n[${stats.scanned}] Record ${recordId}: Missing=${missingShadows}, Extra=${extraShadows}`);
+                if (!dryRun) {
+                    await repairRecordViaDynamoClient(collection, recordId);
+                    stats.repaired++;
+                    console.log(`  ✓ Repaired`);
+                    // レート制限を避けるため少し待機
+                    await new Promise((resolve) => setTimeout(resolve, 100));
+                }
+            }
+        }
+        catch (error) {
+            stats.failed++;
+            stats.errors.push({ id: recordId, error: error.message });
+            console.error(`  ✗ Failed: ${error.message}`);
+        }
+    }
+    // 統計情報を表示
+    console.log('\n=== Repair Summary ===');
+    console.log(`Scanned: ${stats.scanned}`);
+    console.log(`Needs repair: ${stats.needsRepair}`);
+    console.log(`Repaired: ${stats.repaired}`);
+    console.log(`Failed: ${stats.failed}`);
+    if (stats.errors.length > 0) {
+        console.log('\nErrors:');
+        stats.errors.forEach(({ id, error }) => {
+            console.log(`  - ${id}: ${error}`);
+        });
+    }
+}
+/**
+ * メイン処理
+ */
+async function main() {
+    const args = process.argv.slice(2);
+    const resourceIndex = args.indexOf('--resource');
+    const idIndex = args.indexOf('--id');
+    const dryRun = args.includes('--dry-run');
+    const repair = args.includes('--repair');
+    if (resourceIndex === -1 || !args[resourceIndex + 1]) {
+        console.error('Usage: pnpm --filter @exabugs/dynamodb-client tsx scripts/repair-shadows.ts --resource <resource> [--id <id>] [--dry-run|--repair]');
+        process.exit(1);
+    }
+    const resource = args[resourceIndex + 1];
+    const recordId = idIndex !== -1 ? args[idIndex + 1] : undefined;
+    // Records Lambda Function URLを取得
+    const apiUrl = await getRecordsApiUrl();
+    console.log(`Records API: ${apiUrl}`);
+    console.log(`Resource: ${resource}`);
+    console.log(`Mode: ${dryRun ? 'DRY RUN' : repair ? 'REPAIR' : 'DRY RUN (default)'}`);
+    // DynamoClient（IAM認証）を作成
+    console.log(`\nCreating DynamoClient with IAM auth...`);
+    console.log(`Region: ${REGION}`);
+    console.log(`Endpoint: ${apiUrl}`);
+    const client = new DynamoClient(apiUrl, {
+        auth: {
+            region: REGION,
+        },
+        autoConnect: true,
+    });
+    console.log(`Client connected: ${client.isConnected()}\n`);
+    const collection = client
+        .db('ainews')
+        .collection(resource);
+    if (recordId) {
+        await repairSingleRecord(collection, recordId, !repair);
+    }
+    else {
+        await repairAllRecords(collection, !repair);
+    }
+    await client.close();
+}
+main().catch((error) => {
+    console.error('Fatal error:', error);
+    process.exit(1);
+});
+//# sourceMappingURL=repair-shadows.js.map

package/dist/scripts/repair-shadows.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"repair-shadows.js","sourceRoot":"","sources":["../../src/scripts/repair-shadows.ts"],"names":[],"mappings":";AAEA;;;;;;;;;GASG;AACH,OAAO,EAAmB,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEvE,OAAO;AACP,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,KAAK,CAAC;AACrC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,WAAW,CAAC;AAErD,iCAAiC;AACjC,IAAI,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,EAAE,CAAC;AAUxD;;GAEG;AACH,KAAK,UAAU,gBAAgB;IAC7B,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,aAAa;IACb,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;IACnD,MAAM,YAAY,GAAG,UAAU,GAAG,UAAU,CAAC;IAE7C,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CACrB,sDAAsD,YAAY,aAAa,MAAM,sCAAsC,EAC3H,EAAE,QAAQ,EAAE,OAAO,EAAE,CACtB,CAAC;QACF,eAAe,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAChC,OAAO,eAAe,CAAC;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,KAAK,CAAC,kCAAkC,YAAY,EAAE,CAAC,CAAC;IACpE,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,KAAK,UAAU,oBAAoB;IAKjC,kDAAkD;IAClD,OAAO;QACL,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,CAAC;QACjB,YAAY,EAAE,CAAC;KAChB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,2BAA2B,CACxC,UAA4E,EAC5E,QAAgB;IAEhB,wDAAwD;IACxD,MAAM,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;AACjG,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,kBAAkB,CAC/B,UAA4E,EAC5E,QAAgB,EAChB,MAAe;IAEf,OAAO,CAAC,GAAG,CAAC,0BAA0B,QAAQ,MAAM,CAAC,CAAC;IAEtD,YAAY;IACZ,OAAO,CAAC,GAAG,CAAC,4BAA4B,QAAQ,EAAE,CAAC,CAAC;IACpD,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,UAAU,CAAC,CAAC;IAE3C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QAC/C,OAAO;IACT,CAAC;IAED,UAAU;IACV,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,MAAM,oBAAoB,EAAE,CAAC;IAEnF,6CAA6C;IAC7C,mBAAmB;IACnB,OAAO,CAAC,GAAG,CAAC,oBAAoB,cAAc,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,kBAAkB,YAAY,EAAE,CAAC,CAAC;IAE9C,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QAClC,OAAO;IACT,CAAC;IAED,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,mEAAmE,CAAC,CAAC;QACjF,OAAO;IACT,CAAC;IAED,oBAAoB;IACpB,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,CAAC;IACzE,MAAM,2BAA2B,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;AACzC,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB,CAC7B,UAA4E,EAC5E,MAAe;IAEf,MAAM,KAAK,GAAgB;QACzB,OAAO,EAAE,CAAC;QACV,WAAW,EAAE,CAAC;QACd,QAAQ,EAAE,CAAC;QACX,MAAM,EAAE,CAAC;QACT,MAAM,EAAE,EAAE;KACX,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;IAC9C,OAAO,CAAC,GAAG,CAAC,SAAS,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;IAEtD,WAAW;IACX,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnC,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAEvC,OAAO,CAAC,GAAG,CAAC,SAAS,OAAO,CAAC,MAAM,UAAU,CAAC,CAAC;IAE/C,KAAK,MAAM,UAAU,IAAI,OAAO,EAAE,CAAC;QACjC,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,MAAM,QAAQ,GAAG,UAAU,CAAC,EAAE,CAAC;QAE/B,IAAI,CAAC;YACH,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,MAAM,oBAAoB,EAAE,CAAC;YAEnF,IAAI,WAAW,EAAE,CAAC;gBAChB,KAAK,CAAC,WAAW,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CACT,MAAM,KAAK,CAAC,OAAO,YAAY,QAAQ,aAAa,cAAc,WAAW,YAAY,EAAE,CAC5F,CAAC;gBAEF,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,2BAA2B,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;oBACxD,KAAK,CAAC,QAAQ,EAAE,CAAC;oBACjB,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;oBAE5B,kBAAkB;oBAClB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;gBAC3D,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,KAAK,CAAC,MAAM,EAAE,CAAC;YACf,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC1D,OAAO,CAAC,KAAK,CAAC,eAAe,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,UAAU;IACV,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IACzC,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,aAAa,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,WAAW,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAEvC,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACzB,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;YACrC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,KAAK,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,IAAI;IACjB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAEzC,IAAI,aAAa,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC,EAAE,CAAC;QACrD,OAAO,CAAC,KAAK,CACX,oIAAoI,CACrI,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAEhE,iCAAiC;IACjC,MAAM,MAAM,GAAG,MAAM,gBAAgB,EAAE,CAAC;IAExC,OAAO,CAAC,GAAG,CAAC,gBAAgB,MAAM,EAAE,CAAC,CAAC;IACtC,OAAO,CAAC,GAAG,CAAC,aAAa,QAAQ,EAAE,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,SAAS,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAErF,yBAAyB;IACzB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,CAAC,WAAW,MAAM,EAAE,CAAC,CAAC;IACjC,OAAO,CAAC,GAAG,CAAC,aAAa,MAAM,EAAE,CAAC,CAAC;IAEnC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,EAAE;QACtC,IAAI,EAAE;YACJ,MAAM,EAAE,MAAM;SACf;QACD,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,qBAAqB,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAE3D,MAAM,UAAU,GAAG,MAAM;SACtB,EAAE,CAAC,QAAQ,CAAC;SACZ,UAAU,CAAyC,QAAQ,CAAC,CAAC;IAEhE,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,kBAAkB,CAAC,UAAU,EAAE,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC;IAC1D,CAAC;SAAM,CAAC;QACN,MAAM,gBAAgB,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;AACvB,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}