fix(cli): add publish behaviour on run and rollback (#348)

Author: alexjoverm

packages/cli/src/commands/migrations/rollback/actions.ts

@@ -8,6 +8,8 @@ export interface RollbackDataStory {
   storyId: number;
   name: string;
   content: StoryContent;
+  published?: boolean;
+  unpublished_changes?: boolean;
 }
 
 export interface RollbackData {
@@ -20,8 +22,13 @@ export interface RollbackData {
  * @param options.space - The space ID
  * @param options.path - Base path for saving rollback data
  * @param options.story - Story with their original content
+ * @param options.story.id - Story ID
+ * @param options.story.name - Story name
+ * @param options.story.content - Story content
+ * @param options.story.published - Story publication status
+ * @param options.story.unpublished_changes - Story unpublished changes status
  * @param options.migrationTimestamp - The timestamp when the migration started
- * @param options.migrationFile - Name of the migration file being applied
+ * @param options.migrationNames - Names of the migration files being applied
  */
 export async function saveRollbackData({
   space,
@@ -32,14 +39,16 @@ export async function saveRollbackData({
 }: {
   space: string;
   path: string;
-  story: { id: number; name: string; content: StoryContent };
+  story: { id: number; name: string; content: StoryContent; published?: boolean; unpublished_changes?: boolean };
   migrationTimestamp: number;
   migrationNames: string[];
 }): Promise<void> {
   const rollbackData: RollbackDataStory = {
     storyId: story.id,
     name: story.name,
     content: story.content,
+    published: story.published,
+    unpublished_changes: story.unpublished_changes,
   };
 
   // Resolve the path for rollbacks

packages/cli/src/commands/migrations/rollback/index.ts

@@ -53,14 +53,26 @@ migrationsCommand.command('rollback [migrationFile]')
       for (const story of rollbackData.stories) {
         const spinner = new Spinner({ verbose }).start(`Restoring story ${chalk.hex(colorPalette.PRIMARY)(story.name || story.storyId)}...`);
         try {
-          await updateStory(space, story.storyId, {
+          const payload: any = {
             story: {
               content: story.content,
               id: story.storyId,
               name: story.name,
             },
             force_update: '1',
-          });
+          };
+
+          // Only restore publication status if it was saved in the rollback data
+          // For backwards compatibility, we check if the published status exists in the rollback data
+          if (story.published !== undefined && story.unpublished_changes !== undefined) {
+            // If the story was originally published without changes, publish it
+            if (story.published && !story.unpublished_changes) {
+              payload.publish = 1;
+            }
+            // Otherwise, don't publish (let it remain as draft or with unpublished changes)
+          }
+
+          await updateStory(space, story.storyId, payload);
           spinner.succeed(`Restored story ${chalk.hex(colorPalette.PRIMARY)(story.name || story.storyId)}`);
         }
         catch (error) {

packages/cli/src/commands/migrations/run/streams/migrations-transform.ts

@@ -83,7 +83,7 @@ export class MigrationStream extends Transform {
     return migrationFunction;
   }
 
-  private async processStory(story: Story): Promise<Array<{ storyId: number; name: string | undefined; content: StoryContent }>> {
+  private async processStory(story: Story): Promise<Array<{ storyId: number; name: string | undefined; content: StoryContent; published?: boolean; unpublished_changes?: boolean }>> {
     // Filter migrations based on component name if provided
     const relevantMigrations = this.options.componentName
       ? this.options.migrationFiles.filter((file) => {
@@ -101,7 +101,7 @@ export class MigrationStream extends Transform {
       return [];
     }
 
-    const successfulResults: Array<{ storyId: number; name: string | undefined; content: StoryContent }> = [];
+    const successfulResults: Array<{ storyId: number; name: string | undefined; content: StoryContent; published?: boolean; unpublished_changes?: boolean }> = [];
 
     // Process each relevant migration
     const result = await this.applyMigrationsToStory(story, relevantMigrations);
@@ -112,7 +112,7 @@ export class MigrationStream extends Transform {
     return successfulResults;
   }
 
-  private async applyMigrationsToStory(story: Story, migrationFiles: MigrationFile[]): Promise<{ storyId: number; name: string | undefined; content: StoryContent } | null> {
+  private async applyMigrationsToStory(story: Story, migrationFiles: MigrationFile[]): Promise<{ storyId: number; name: string | undefined; content: StoryContent; published?: boolean; unpublished_changes?: boolean } | null> {
     const migrationNames = migrationFiles.map(f => f.name);
 
     try {
@@ -144,7 +144,13 @@ export class MigrationStream extends Transform {
         await saveRollbackData({
           space: this.options.space,
           path: this.options.path,
-          story: { id: story.id, name: story.name || '', content: story.content as StoryContent },
+          story: {
+            id: story.id,
+            name: story.name || '',
+            content: story.content as StoryContent,
+            published: story.published,
+            unpublished_changes: story.unpublished_changes,
+          },
           migrationTimestamp: this.timestamp,
           migrationNames,
         });
@@ -160,6 +166,8 @@ export class MigrationStream extends Transform {
           storyId: story.id,
           name: story.name,
           content: storyContent,
+          published: story.published,
+          unpublished_changes: story.unpublished_changes,
         };
       }
       else if (processed && !contentChanged) {