commit ee848ea793c2156c0e6e9097b8adbbef257af8cb · knotbin.com/airport

+2 -1

.env.example

···

       1
       1
        
       # generate with `openssl ecparam --name secp256k1 --genkey --noout --outform DER | tail --bytes=+8 | head --bytes=32 | xxd --plain --cols 32`

     

       2
       2
       -
       COOKIE_SECRET=my_secret
     

       2
       2
       +
       COOKIE_SECRET=my_secret

     

       3
       3
       +
       MIGRATION_STATE=up

+74 -18

islands/MigrationProgress.tsx

···

       1
       1
        
       import { useEffect, useState } from "preact/hooks";

     

       2
       2
        
       

     

       3
       3
        
       /**

     

       4
       4
       +
        * The migration state info.

     

       5
       5
       +
        * @type {MigrationStateInfo}

     

       6
       6
       +
        */

     

       7
       7
       +
       interface MigrationStateInfo {

     

       8
       8
       +
         state: "up" | "issue" | "maintenance";

     

       9
       9
       +
         message: string;

     

       10
       10
       +
         allowMigration: boolean;

     

       11
       11
       +
       }

     

       12
       12
       +
       

     

       13
       13
       +
       /**

     

       4
       14
        
        * The migration progress props.

     

       5
       15
        
        * @type {MigrationProgressProps}

     

       6
       16
        
        */

     
···

       30
       40
        
        */

     

       31
       41
        
       export default function MigrationProgress(props: MigrationProgressProps) {

     

       32
       42
        
         const [token, setToken] = useState("");

     

       43
       43
       +
         const [migrationState, setMigrationState] = useState<MigrationStateInfo | null>(null);

     

       33
       44
        
       

     

       34
       45
        
         const [steps, setSteps] = useState<MigrationStep[]>([

     

       35
       46
        
           { name: "Create Account", status: "pending" },

     
···

       88
       99
        
             invite: props.invite,

     

       89
       100
        
           });

     

       90
       101
        
       

     

       91
       91
       -
           if (!validateParams()) {

     

       92
       92
       -
             console.log("Parameter validation failed");

     

       93
       93
       -
             return;

     

       94
       94
       -
           }

     

       102
       102
       +
           // Check migration state first

     

       103
       103
       +
           const checkMigrationState = async () => {

     

       104
       104
       +
             try {

     

       105
       105
       +
               const migrationResponse = await fetch("/api/migration-state");

     

       106
       106
       +
               if (migrationResponse.ok) {

     

       107
       107
       +
                 const migrationData = await migrationResponse.json();

     

       108
       108
       +
                 setMigrationState(migrationData);

     

       95
       109
        
       

     

       96
       96
       -
           startMigration().catch((error) => {

     

       97
       97
       -
             console.error("Unhandled migration error:", error);

     

       98
       98
       -
             updateStepStatus(

     

       99
       99
       -
               0,

     

       100
       100
       -
               "error",

     

       101
       101
       -
               error instanceof Error ? error.message : String(error),

     

       102
       102
       -
             );

     

       103
       103
       -
           });

     

       110
       110
       +
                 if (!migrationData.allowMigration) {

     

       111
       111
       +
                   updateStepStatus(0, "error", migrationData.message);

     

       112
       112
       +
                   return;

     

       113
       113
       +
                 }

     

       114
       114
       +
               }

     

       115
       115
       +
             } catch (error) {

     

       116
       116
       +
               console.error("Failed to check migration state:", error);

     

       117
       117
       +
               updateStepStatus(0, "error", "Unable to verify migration availability");

     

       118
       118
       +
               return;

     

       119
       119
       +
             }

     

       120
       120
       +
       

     

       121
       121
       +
             if (!validateParams()) {

     

       122
       122
       +
               console.log("Parameter validation failed");

     

       123
       123
       +
               return;

     

       124
       124
       +
             }

     

       125
       125
       +
       

     

       126
       126
       +
             startMigration().catch((error) => {

     

       127
       127
       +
               console.error("Unhandled migration error:", error);

     

       128
       128
       +
               updateStepStatus(

     

       129
       129
       +
                 0,

     

       130
       130
       +
                 "error",

     

       131
       131
       +
                 error.message || "Unknown error occurred",

     

       132
       132
       +
               );

     

       133
       133
       +
             });

     

       134
       134
       +
           };

     

       135
       135
       +
       

     

       136
       136
       +
           checkMigrationState();

     

       104
       137
        
         }, []);

     

       105
       138
        
       

     

       106
       139
        
         const getStepDisplayName = (step: MigrationStep, index: number) => {

     
···

       112
       145
        
               case 3: return "Migration Finalized";

     

       113
       146
        
             }

     

       114
       147
        
           }

     

       115
       115
       -
           

     

       148
       148
       +
       

     

       116
       149
        
           if (step.status === "in-progress") {

     

       117
       150
        
             switch (index) {

     

       118
       151
        
               case 0: return "Creating your new account...";

     

       119
       152
        
               case 1: return "Migrating your data...";

     

       120
       120
       -
               case 2: return step.name === "Enter the token sent to your email to complete identity migration" 

     

       121
       121
       -
                 ? step.name 

     

       153
       153
       +
               case 2: return step.name === "Enter the token sent to your email to complete identity migration"

     

       154
       154
       +
                 ? step.name

     

       122
       155
        
                 : "Migrating your identity...";

     

       123
       156
        
               case 3: return "Finalizing migration...";

     

       124
       157
        
             }

     
···

       132
       165
        
               case 3: return "Verifying migration completion...";

     

       133
       166
        
             }

     

       134
       167
        
           }

     

       135
       135
       -
           

     

       168
       168
       +
       

     

       136
       169
        
           return step.name;

     

       137
       170
        
         };

     

       138
       171
        
       

     
···

       310
       343
        
                   );

     

       311
       344
        
                 }

     

       312
       345
        
                 console.log("Identity migration requested successfully");

     

       313
       313
       -
                 

     

       346
       346
       +
       

     

       314
       347
        
                 // Update step name to prompt for token

     

       315
       348
        
                 setSteps(prevSteps =>

     

       316
       349
        
                   prevSteps.map((step, i) =>

     
···

       515
       548
        
             console.log(`Verification: Status response status:`, res.status);

     

       516
       549
        
             const data = await res.json();

     

       517
       550
        
             console.log(`Verification: Status data for step ${stepNum + 1}:`, data);

     

       518
       518
       -
             

     

       551
       551
       +
       

     

       519
       552
        
             if (data.ready) {

     

       520
       553
        
               console.log(`Verification: Step ${stepNum + 1} is ready`);

     

       521
       554
        
               updateStepStatus(stepNum, "completed");

     
···

       548
       581
        
       

     

       549
       582
        
         return (

     

       550
       583
        
           <div class="space-y-8">

     

       584
       584
       +
             {/* Migration state alert */}

     

       585
       585
       +
             {migrationState && !migrationState.allowMigration && (

     

       586
       586
       +
               <div class={`p-4 rounded-lg border ${

     

       587
       587
       +
                 migrationState.state === "maintenance"

     

       588
       588
       +
                   ? "bg-yellow-50 border-yellow-200 text-yellow-800 dark:bg-yellow-900/20 dark:border-yellow-800 dark:text-yellow-200"

     

       589
       589
       +
                   : "bg-red-50 border-red-200 text-red-800 dark:bg-red-900/20 dark:border-red-800 dark:text-red-200"

     

       590
       590
       +
               }`}>

     

       591
       591
       +
                 <div class="flex items-center">

     

       592
       592
       +
                   <div class={`mr-3 ${

     

       593
       593
       +
                     migrationState.state === "maintenance" ? "text-yellow-600 dark:text-yellow-400" : "text-red-600 dark:text-red-400"

     

       594
       594
       +
                   }`}>

     

       595
       595
       +
                     {migrationState.state === "maintenance" ? "⚠️" : "🚫"}

     

       596
       596
       +
                   </div>

     

       597
       597
       +
                   <div>

     

       598
       598
       +
                     <h3 class="font-semibold mb-1">

     

       599
       599
       +
                       {migrationState.state === "maintenance" ? "Maintenance Mode" : "Service Unavailable"}

     

       600
       600
       +
                     </h3>

     

       601
       601
       +
                     <p class="text-sm">{migrationState.message}</p>

     

       602
       602
       +
                   </div>

     

       603
       603
       +
                 </div>

     

       604
       604
       +
               </div>

     

       605
       605
       +
             )}

     

       606
       606
       +
       

     

       551
       607
        
             <div class="space-y-4">

     

       552
       608
        
               {steps.map((step, index) => (

     

       553
       609
        
                 <div key={step.name} class={getStepClasses(step.status)}>

+59 -5

islands/MigrationSetup.tsx

···

       33
       33
        
       }

     

       34
       34
        
       

     

       35
       35
        
       /**

     

       36
       36
       +
        * The migration state info.

     

       37
       37
       +
        * @type {MigrationStateInfo}

     

       38
       38
       +
        */

     

       39
       39
       +
       interface MigrationStateInfo {

     

       40
       40
       +
         state: "up" | "issue" | "maintenance";

     

       41
       41
       +
         message: string;

     

       42
       42
       +
         allowMigration: boolean;

     

       43
       43
       +
       }

     

       44
       44
       +
       

     

       45
       45
       +
       /**

     

       36
       46
        
        * The migration setup component.

     

       37
       47
        
        * @param props - The migration setup props

     

       38
       48
        
        * @returns The migration setup component

     
···

       54
       64
        
         const [showConfirmation, setShowConfirmation] = useState(false);

     

       55
       65
        
         const [confirmationText, setConfirmationText] = useState("");

     

       56
       66
        
         const [passport, setPassport] = useState<UserPassport | null>(null);

     

       67
       67
       +
         const [migrationState, setMigrationState] = useState<MigrationStateInfo | null>(null);

     

       57
       68
        
       

     

       58
       69
        
         const ensureServiceUrl = (url: string): string => {

     

       59
       70
        
           if (!url) return url;

     
···

       70
       81
        
         useEffect(() => {

     

       71
       82
        
           if (!IS_BROWSER) return;

     

       72
       83
        
       

     

       73
       73
       -
           const fetchPassport = async () => {

     

       84
       84
       +
           const fetchInitialData = async () => {

     

       74
       85
        
             try {

     

       86
       86
       +
               // Check migration state first

     

       87
       87
       +
               const migrationResponse = await fetch("/api/migration-state");

     

       88
       88
       +
               if (migrationResponse.ok) {

     

       89
       89
       +
                 const migrationData = await migrationResponse.json();

     

       90
       90
       +
                 setMigrationState(migrationData);

     

       91
       91
       +
               }

     

       92
       92
       +
       

     

       93
       93
       +
               // Fetch user passport

     

       75
       94
        
               const response = await fetch("/api/me", {

     

       76
       95
        
                 credentials: "include",

     

       77
       96
        
               });

     
···

       92
       111
        
                 });

     

       93
       112
        
               }

     

       94
       113
        
             } catch (error) {

     

       95
       95
       -
               console.error("Failed to fetch passport:", error);

     

       114
       114
       +
               console.error("Failed to fetch initial data:", error);

     

       96
       115
        
             }

     

       97
       116
        
           };

     

       98
       117
        
       

     

       99
       99
       -
           fetchPassport();

     

       118
       118
       +
           fetchInitialData();

     

       100
       119
        
         }, []);

     

       101
       120
        
       

     

       102
       121
        
         const checkServerDescription = async (serviceUrl: string) => {

     
···

       144
       163
        
         const handleSubmit = (e: Event) => {

     

       145
       164
        
           e.preventDefault();

     

       146
       165
        
       

     

       166
       166
       +
           // Check migration state first

     

       167
       167
       +
           if (migrationState && !migrationState.allowMigration) {

     

       168
       168
       +
             setError(migrationState.message);

     

       169
       169
       +
             return;

     

       170
       170
       +
           }

     

       171
       171
       +
       

     

       147
       172
        
           if (!service || !handlePrefix || !email || !password) {

     

       148
       173
        
             setError("Please fill in all required fields");

     

       149
       174
        
             return;

     
···

       158
       183
        
         };

     

       159
       184
        
       

     

       160
       185
        
         const handleConfirmation = () => {

     

       186
       186
       +
           // Double-check migration state before proceeding

     

       187
       187
       +
           if (migrationState && !migrationState.allowMigration) {

     

       188
       188
       +
             setError(migrationState.message);

     

       189
       189
       +
             return;

     

       190
       190
       +
           }

     

       191
       191
       +
       

     

       161
       192
        
           if (confirmationText !== "MIGRATE") {

     

       162
       193
        
             setError("Please type 'MIGRATE' to confirm");

     

       163
       194
        
             return;

     
···

       189
       220
        
             <div class="absolute top-2 left-4 text-blue-500 text-sm font-mono">TERMINAL 1</div>

     

       190
       221
        
             <div class="absolute top-2 right-4 text-blue-500 text-sm font-mono">GATE M1</div>

     

       191
       222
        
       

     

       223
       223
       +
             {/* Migration state alert */}

     

       224
       224
       +
             {migrationState && !migrationState.allowMigration && (

     

       225
       225
       +
               <div class={`mb-6 mt-4 p-4 rounded-lg border ${

     

       226
       226
       +
                 migrationState.state === "maintenance"

     

       227
       227
       +
                   ? "bg-yellow-50 border-yellow-200 text-yellow-800 dark:bg-yellow-900/20 dark:border-yellow-800 dark:text-yellow-200"

     

       228
       228
       +
                   : "bg-red-50 border-red-200 text-red-800 dark:bg-red-900/20 dark:border-red-800 dark:text-red-200"

     

       229
       229
       +
               }`}>

     

       230
       230
       +
                 <div class="flex items-center">

     

       231
       231
       +
                   <div class={`mr-3 ${

     

       232
       232
       +
                     migrationState.state === "maintenance" ? "text-yellow-600 dark:text-yellow-400" : "text-red-600 dark:text-red-400"

     

       233
       233
       +
                   }`}>

     

       234
       234
       +
                     {migrationState.state === "maintenance" ? "⚠️" : "🚫"}

     

       235
       235
       +
                   </div>

     

       236
       236
       +
                   <div>

     

       237
       237
       +
                     <h3 class="font-semibold mb-1">

     

       238
       238
       +
                       {migrationState.state === "maintenance" ? "Maintenance Mode" : "Service Unavailable"}

     

       239
       239
       +
                     </h3>

     

       240
       240
       +
                     <p class="text-sm">{migrationState.message}</p>

     

       241
       241
       +
                   </div>

     

       242
       242
       +
                 </div>

     

       243
       243
       +
               </div>

     

       244
       244
       +
             )}

     

       245
       245
       +
       

     

       192
       246
        
             <div class="text-center mb-8 relative">

     

       193
       247
        
               <p class="text-gray-600 dark:text-gray-400 mt-4">Please complete your migration check-in</p>

     

       194
       248
        
               <div class="mt-2 text-sm text-gray-500 dark:text-gray-400 font-mono">FLIGHT: MIG-2024</div>

     
···

       226
       280
        
       

     

       227
       281
        
             <form onSubmit={handleSubmit} class="space-y-6">

     

       228
       282
        
               {error && (

     

       229
       229
       -
                 <div class="bg-red-50 dark:bg-red-900 rounded-lg">

     

       283
       283
       +
                 <div class="bg-red-50 dark:bg-red-900 rounded-lg ">

     

       230
       284
        
                   <p class="text-red-800 dark:text-red-200 flex items-center">

     

       231
       285
        
                     <svg class="w-5 h-5 mr-2" fill="none" stroke="currentColor" viewBox="0 0 24 24">

     

       232
       286
        
                       <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M12 9v2m0 4h.01m-6.938 4h13.856c1.54 0 2.502-1.667 1.732-3L13.732 4c-.77-1.333-2.694-1.333-3.464 0L3.34 16c-.77 1.333.192 3 1.732 3z"></path>

     
···

       398
       452
        
       

     

       399
       453
        
               <button

     

       400
       454
        
                 type="submit"

     

       401
       401
       -
                 disabled={isLoading}

     

       455
       455
       +
                 disabled={isLoading || Boolean(migrationState && !migrationState.allowMigration)}

     

       402
       456
        
                 class="w-full flex justify-center items-center py-3 px-4 rounded-md shadow-sm text-sm font-medium text-white bg-blue-600 hover:bg-blue-700 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-blue-500 disabled:opacity-50 disabled:cursor-not-allowed"

     

       403
       457
        
               >

     

       404
       458
        
                 <svg class="w-5 h-5 mr-2" fill="none" stroke="currentColor" viewBox="0 0 24 24">

+70

lib/migration-state.ts

···

       1
       1
       +
       /**

     

       2
       2
       +
        * Migration state types and utilities for controlling migration availability.

     

       3
       3
       +
        */

     

       4
       4
       +
       

     

       5
       5
       +
       export type MigrationState = "up" | "issue" | "maintenance";

     

       6
       6
       +
       

     

       7
       7
       +
       export interface MigrationStateInfo {

     

       8
       8
       +
         state: MigrationState;

     

       9
       9
       +
         message: string;

     

       10
       10
       +
         allowMigration: boolean;

     

       11
       11
       +
       }

     

       12
       12
       +
       

     

       13
       13
       +
       /**

     

       14
       14
       +
        * Get the current migration state from environment variables.

     

       15
       15
       +
        * @returns The migration state information

     

       16
       16
       +
        */

     

       17
       17
       +
       export function getMigrationState(): MigrationStateInfo {

     

       18
       18
       +
         const state = (Deno.env.get("MIGRATION_STATE") || "up").toLowerCase() as MigrationState;

     

       19
       19
       +
       

     

       20
       20
       +
         switch (state) {

     

       21
       21
       +
           case "issue":

     

       22
       22
       +
             return {

     

       23
       23
       +
               state: "issue",

     

       24
       24
       +
               message: "Migration services are temporarily unavailable as we investigate an issue. Please try again later.",

     

       25
       25
       +
               allowMigration: false,

     

       26
       26
       +
             };

     

       27
       27
       +
       

     

       28
       28
       +
           case "maintenance":

     

       29
       29
       +
             return {

     

       30
       30
       +
               state: "maintenance",

     

       31
       31
       +
               message: "Migration services are temporarily unavailable for scheduled maintenance. Please try again later.",

     

       32
       32
       +
               allowMigration: false,

     

       33
       33
       +
             };

     

       34
       34
       +
       

     

       35
       35
       +
           case "up":

     

       36
       36
       +
           default:

     

       37
       37
       +
             return {

     

       38
       38
       +
               state: "up",

     

       39
       39
       +
               message: "Migration services are operational.",

     

       40
       40
       +
               allowMigration: true,

     

       41
       41
       +
             };

     

       42
       42
       +
         }

     

       43
       43
       +
       }

     

       44
       44
       +
       

     

       45
       45
       +
       /**

     

       46
       46
       +
        * Check if migrations are currently allowed.

     

       47
       47
       +
        * @returns True if migrations are allowed, false otherwise

     

       48
       48
       +
        */

     

       49
       49
       +
       export function isMigrationAllowed(): boolean {

     

       50
       50
       +
         return getMigrationState().allowMigration;

     

       51
       51
       +
       }

     

       52
       52
       +
       

     

       53
       53
       +
       /**

     

       54
       54
       +
        * Get a user-friendly message for the current migration state.

     

       55
       55
       +
        * @returns The message to display to users

     

       56
       56
       +
        */

     

       57
       57
       +
       export function getMigrationStateMessage(): string {

     

       58
       58
       +
         return getMigrationState().message;

     

       59
       59
       +
       }

     

       60
       60
       +
       

     

       61
       61
       +
       /**

     

       62
       62
       +
        * Throw an error if migrations are not allowed.

     

       63
       63
       +
        * Used in API endpoints to prevent migration operations when disabled.

     

       64
       64
       +
        */

     

       65
       65
       +
       export function assertMigrationAllowed(): void {

     

       66
       66
       +
         const stateInfo = getMigrationState();

     

       67
       67
       +
         if (!stateInfo.allowMigration) {

     

       68
       68
       +
           throw new Error(stateInfo.message);

     

       69
       69
       +
         }

     

       70
       70
       +
       }

routes/api/migrate/create.ts

···

       2
       2
        
       import { setCredentialSession } from "../../../lib/cred/sessions.ts";

     

       3
       3
        
       import { Agent } from "@atproto/api";

     

       4
       4
        
       import { define } from "../../../utils.ts";

     

       5
       5
       +
       import { assertMigrationAllowed } from "../../../lib/migration-state.ts";

     

       5
       6
        
       

     

       6
       7
        
       /**

     

       7
       8
        
        * Handle account creation

     
···

       19
       20
        
         async POST(ctx) {

     

       20
       21
        
           const res = new Response();

     

       21
       22
        
           try {

     

       23
       23
       +
             // Check if migrations are currently allowed

     

       24
       24
       +
             assertMigrationAllowed();

     

       25
       25
       +
       

     

       22
       26
        
             const body = await ctx.req.json();

     

       23
       27
        
             const serviceUrl = body.service;

     

       24
       28
        
             const newHandle = body.handle;

+9 -5

routes/api/migrate/data/blobs.ts

···

       1
       1
        
       import { getSessionAgent } from "../../../../lib/sessions.ts";

     

       2
       2
        
       import { define } from "../../../../utils.ts";

     

       3
       3
       +
       import { assertMigrationAllowed } from "../../../../lib/migration-state.ts";

     

       3
       4
        
       

     

       4
       5
        
       export const handler = define.handlers({

     

       5
       6
        
         async POST(ctx) {

     

       6
       7
        
           const res = new Response();

     

       7
       8
        
           try {

     

       9
       9
       +
             // Check if migrations are currently allowed

     

       10
       10
       +
             assertMigrationAllowed();

     

       11
       11
       +
       

     

       8
       12
        
             console.log("Blob migration: Starting session retrieval");

     

       9
       13
        
             const oldAgent = await getSessionAgent(ctx.req);

     

       10
       14
        
             console.log("Blob migration: Got old agent:", !!oldAgent);

     
···

       56
       60
        
       

     

       57
       61
        
             const session = await oldAgent.com.atproto.server.getSession();

     

       58
       62
        
             const accountDid = session.data.did;

     

       59
       59
       -
             

     

       63
       63
       +
       

     

       60
       64
        
             do {

     

       61
       65
        
               const pageStartTime = Date.now();

     

       62
       66
        
               console.log(`[${new Date().toISOString()}] Counting blobs on page ${pageCount + 1}...`);

     
···

       69
       73
        
               const newBlobs = listedBlobs.data.cids.length;

     

       70
       74
        
               totalBlobs += newBlobs;

     

       71
       75
        
               const pageTime = Date.now() - pageStartTime;

     

       72
       72
       -
               

     

       76
       76
       +
       

     

       73
       77
        
               console.log(`[${new Date().toISOString()}] Found ${newBlobs} blobs on page ${pageCount + 1} in ${pageTime/1000} seconds, total so far: ${totalBlobs}`);

     

       74
       78
        
               migrationLogs.push(`[${new Date().toISOString()}] Found ${newBlobs} blobs on page ${pageCount + 1} in ${pageTime/1000} seconds, total so far: ${totalBlobs}`);

     

       75
       75
       -
               

     

       79
       79
       +
       

     

       76
       80
        
               pageCount++;

     

       77
       81
        
               blobCursor = listedBlobs.data.cursor;

     

       78
       82
        
             } while (blobCursor);

     
···

       172
       176
        
             return new Response(

     

       173
       177
        
               JSON.stringify({

     

       174
       178
        
                 success: true,

     

       175
       175
       -
                 message: failedBlobs.length > 0 

     

       179
       179
       +
                 message: failedBlobs.length > 0

     

       176
       180
        
                   ? `Blob migration completed with ${failedBlobs.length} failed blobs`

     

       177
       181
        
                   : "Blob migration completed successfully",

     

       178
       182
        
                 migratedBlobs,

     
···

       218
       222
        
             );

     

       219
       223
        
           }

     

       220
       224
        
         }

     

       221
       221
       -
       }); 
     

       225
       225
       +
       });

+7 -3

routes/api/migrate/data/prefs.ts

···

       1
       1
        
       import { getSessionAgent } from "../../../../lib/sessions.ts";

     

       2
       2
        
       import { define } from "../../../../utils.ts";

     

       3
       3
       +
       import { assertMigrationAllowed } from "../../../../lib/migration-state.ts";

     

       3
       4
        
       

     

       4
       5
        
       export const handler = define.handlers({

     

       5
       6
        
         async POST(ctx) {

     

       6
       7
        
           const res = new Response();

     

       7
       8
        
           try {

     

       9
       9
       +
             // Check if migrations are currently allowed

     

       10
       10
       +
             assertMigrationAllowed();

     

       11
       11
       +
       

     

       8
       12
        
             console.log("Preferences migration: Starting session retrieval");

     

       9
       13
        
             const oldAgent = await getSessionAgent(ctx.req);

     

       10
       14
        
             console.log("Preferences migration: Got old agent:", !!oldAgent);

     
···

       31
       35
        
             // Fetch preferences

     

       32
       36
        
             console.log(`[${new Date().toISOString()}] Fetching preferences from old account...`);

     

       33
       37
        
             migrationLogs.push(`[${new Date().toISOString()}] Fetching preferences from old account...`);

     

       34
       34
       -
             

     

       38
       38
       +
       

     

       35
       39
        
             const fetchStartTime = Date.now();

     

       36
       40
        
             const prefs = await oldAgent.app.bsky.actor.getPreferences();

     

       37
       41
        
             const fetchTime = Date.now() - fetchStartTime;

     

       38
       38
       -
             

     

       42
       42
       +
       

     

       39
       43
        
             console.log(`[${new Date().toISOString()}] Preferences fetched in ${fetchTime/1000} seconds`);

     

       40
       44
        
             migrationLogs.push(`[${new Date().toISOString()}] Preferences fetched in ${fetchTime/1000} seconds`);

     

       41
       45
        
       

     
···

       98
       102
        
             );

     

       99
       103
        
           }

     

       100
       104
        
         }

     

       101
       101
       -
       }); 
     

       105
       105
       +
       });

+7 -3

routes/api/migrate/data/repo.ts

···

       1
       1
        
       import { getSessionAgent } from "../../../../lib/sessions.ts";

     

       2
       2
        
       import { define } from "../../../../utils.ts";

     

       3
       3
       +
       import { assertMigrationAllowed } from "../../../../lib/migration-state.ts";

     

       3
       4
        
       

     

       4
       5
        
       export const handler = define.handlers({

     

       5
       6
        
         async POST(ctx) {

     

       6
       7
        
           const res = new Response();

     

       7
       8
        
           try {

     

       9
       9
       +
             // Check if migrations are currently allowed

     

       10
       10
       +
             assertMigrationAllowed();

     

       11
       11
       +
       

     

       8
       12
        
             console.log("Repo migration: Starting session retrieval");

     

       9
       13
        
             const oldAgent = await getSessionAgent(ctx.req);

     

       10
       14
        
             console.log("Repo migration: Got old agent:", !!oldAgent);

     
···

       34
       38
        
             // Get repo data from old account

     

       35
       39
        
             console.log(`[${new Date().toISOString()}] Fetching repo data from old account...`);

     

       36
       40
        
             migrationLogs.push(`[${new Date().toISOString()}] Fetching repo data from old account...`);

     

       37
       37
       -
             

     

       41
       41
       +
       

     

       38
       42
        
             const fetchStartTime = Date.now();

     

       39
       43
        
             const repoData = await oldAgent.com.atproto.sync.getRepo({

     

       40
       44
        
               did: accountDid,

     

       41
       45
        
             });

     

       42
       46
        
             const fetchTime = Date.now() - fetchStartTime;

     

       43
       43
       -
             

     

       47
       47
       +
       

     

       44
       48
        
             console.log(`[${new Date().toISOString()}] Repo data fetched in ${fetchTime/1000} seconds`);

     

       45
       49
        
             migrationLogs.push(`[${new Date().toISOString()}] Repo data fetched in ${fetchTime/1000} seconds`);

     

       46
       50
        
       

     
···

       105
       109
        
             );

     

       106
       110
        
           }

     

       107
       111
        
         }

     

       108
       108
       -
       }); 
     

       112
       112
       +
       });

routes/api/migrate/finalize.ts

···

       1
       1
        
       import { getSessionAgent } from "../../../lib/sessions.ts";

     

       2
       2
        
       import { define } from "../../../utils.ts";

     

       3
       3
       +
       import { assertMigrationAllowed } from "../../../lib/migration-state.ts";

     

       3
       4
        
       

     

       4
       5
        
       export const handler = define.handlers({

     

       5
       6
        
         async POST(ctx) {

     

       6
       7
        
           const res = new Response();

     

       7
       8
        
           try {

     

       9
       9
       +
             // Check if migrations are currently allowed

     

       10
       10
       +
             assertMigrationAllowed();

     

       11
       11
       +
       

     

       8
       12
        
             const oldAgent = await getSessionAgent(ctx.req);

     

       9
       13
        
             const newAgent = await getSessionAgent(ctx.req, res, true);

     

       10
       14

routes/api/migrate/identity/request.ts

···

       2
       2
        
         getSessionAgent,

     

       3
       3
        
       } from "../../../../lib/sessions.ts";

     

       4
       4
        
       import { define } from "../../../../utils.ts";

     

       5
       5
       +
       import { assertMigrationAllowed } from "../../../../lib/migration-state.ts";

     

       5
       6
        
       

     

       6
       7
        
       /**

     

       7
       8
        
        * Handle identity migration request

     
···

       14
       15
        
         async POST(ctx) {

     

       15
       16
        
           const res = new Response();

     

       16
       17
        
           try {

     

       18
       18
       +
             // Check if migrations are currently allowed

     

       19
       19
       +
             assertMigrationAllowed();

     

       20
       20
       +
       

     

       17
       21
        
             console.log("Starting identity migration request...");

     

       18
       22
        
             const oldAgent = await getSessionAgent(ctx.req);

     

       19
       23
        
             console.log("Got old agent:", {

routes/api/migrate/identity/sign.ts

···

       4
       4
        
       import { Secp256k1Keypair } from "npm:@atproto/crypto";

     

       5
       5
        
       import * as ui8 from "npm:uint8arrays";

     

       6
       6
        
       import { define } from "../../../../utils.ts";

     

       7
       7
       +
       import { assertMigrationAllowed } from "../../../../lib/migration-state.ts";

     

       7
       8
        
       

     

       8
       9
        
       /**

     

       9
       10
        
        * Handle identity migration sign

     
···

       16
       17
        
         async POST(ctx) {

     

       17
       18
        
           const res = new Response();

     

       18
       19
        
           try {

     

       20
       20
       +
             // Check if migrations are currently allowed

     

       21
       21
       +
             assertMigrationAllowed();

     

       19
       22
        
             const url = new URL(ctx.req.url);

     

       20
       23
        
             const token = url.searchParams.get("token");

     

       21
       24

+44

routes/api/migration-state.ts

···

       1
       1
       +
       import { getMigrationState } from "../../lib/migration-state.ts";

     

       2
       2
       +
       import { define } from "../../utils.ts";

     

       3
       3
       +
       

     

       4
       4
       +
       /**

     

       5
       5
       +
        * API endpoint to check the current migration state.

     

       6
       6
       +
        * Returns the migration state information including whether migrations are allowed.

     

       7
       7
       +
        */

     

       8
       8
       +
       export const handler = define.handlers({

     

       9
       9
       +
         GET(_ctx) {

     

       10
       10
       +
           try {

     

       11
       11
       +
             const stateInfo = getMigrationState();

     

       12
       12
       +
       

     

       13
       13
       +
             return new Response(

     

       14
       14
       +
               JSON.stringify({

     

       15
       15
       +
                 state: stateInfo.state,

     

       16
       16
       +
                 message: stateInfo.message,

     

       17
       17
       +
                 allowMigration: stateInfo.allowMigration,

     

       18
       18
       +
               }),

     

       19
       19
       +
               {

     

       20
       20
       +
                 status: 200,

     

       21
       21
       +
                 headers: {

     

       22
       22
       +
                   "Content-Type": "application/json",

     

       23
       23
       +
                 },

     

       24
       24
       +
               }

     

       25
       25
       +
             );

     

       26
       26
       +
           } catch (error) {

     

       27
       27
       +
             console.error("Error checking migration state:", error);

     

       28
       28
       +
       

     

       29
       29
       +
             return new Response(

     

       30
       30
       +
               JSON.stringify({

     

       31
       31
       +
                 state: "issue",

     

       32
       32
       +
                 message: "Unable to determine migration state. Please try again later.",

     

       33
       33
       +
                 allowMigration: false,

     

       34
       34
       +
               }),

     

       35
       35
       +
               {

     

       36
       36
       +
                 status: 500,

     

       37
       37
       +
                 headers: {

     

       38
       38
       +
                   "Content-Type": "application/json",

     

       39
       39
       +
                 },

     

       40
       40
       +
               }

     

       41
       41
       +
             );

     

       42
       42
       +
           }

     

       43
       43
       +
         },

     

       44
       44
       +
       });