feat(cli): aded metadata transfers for .po files

Author: maxprilutskiy

.changeset/slimy-onions-collect.md

@@ -0,0 +1,5 @@
+---
+"lingo.dev": patch
+---
+
+metadata transfers for .po

packages/cli/src/cli/loaders/_types.ts

@@ -1,7 +1,14 @@
 export interface ILoaderDefinition<I, O, C> {
   init?(): Promise<C>;
   pull(locale: string, input: I, initCtx?: C): Promise<O>;
-  push(locale: string, data: O, originalInput: I | null, originalLocale: string): Promise<I>;
+  push(
+    locale: string,
+    data: O,
+    originalInput: I | null,
+    originalLocale: string,
+    pullInput: I | null,
+    pullOutput: O | null,
+  ): Promise<I>;
 }
 
 export interface ILoader<I, O, C = void> extends ILoaderDefinition<I, O, C> {

packages/cli/src/cli/loaders/_utils.ts

@@ -34,6 +34,8 @@ export function createLoader<I, O, C>(lDefinition: ILoaderDefinition<I, O, C>):
   const state = {
     defaultLocale: undefined as string | undefined,
     originalInput: undefined as I | undefined | null,
+    pullInput: undefined as I | undefined | null,
+    pullOutput: undefined as O | undefined | null,
     initCtx: undefined as C | undefined,
   };
   return {
@@ -62,7 +64,11 @@ export function createLoader<I, O, C>(lDefinition: ILoaderDefinition<I, O, C>):
         state.originalInput = input || null;
       }
 
-      return lDefinition.pull(locale, input, state.initCtx);
+      state.pullInput = input;
+      const result = await lDefinition.pull(locale, input, state.initCtx);
+      state.pullOutput = result;
+
+      return result;
     },
     async push(locale, data) {
       if (!state.defaultLocale) {
@@ -72,7 +78,14 @@ export function createLoader<I, O, C>(lDefinition: ILoaderDefinition<I, O, C>):
         throw new Error("Cannot push data without pulling first");
       }
 
-      const pushResult = await lDefinition.push(locale, data, state.originalInput, state.defaultLocale);
+      const pushResult = await lDefinition.push(
+        locale,
+        data,
+        state.originalInput,
+        state.defaultLocale,
+        state.pullInput!,
+        state.pullOutput!,
+      );
       return pushResult;
     },
   };

packages/cli/src/cli/loaders/po/index.ts

@@ -40,16 +40,31 @@ export function createPoDataLoader(params: PoLoaderParams): ILoader<string, PoTr
       return result;
     },
 
-    async push(locale, data, originalInput) {
+    async push(locale, data, originalInput, originalLocale, pullInput) {
       // Parse each section to maintain structure
-      const sections = originalInput?.split("\n\n").filter(Boolean) || [];
-      const result = sections
+      const currentSections = pullInput?.split("\n\n").filter(Boolean) || [];
+      const originalSections = originalInput?.split("\n\n").filter(Boolean) || [];
+      const result = originalSections
         .map((section) => {
           const sectionPo = gettextParser.po.parse(section);
           const contextKey = _.keys(sectionPo.translations)[0];
           const entries = sectionPo.translations[contextKey];
           const msgid = Object.keys(entries).find((key) => entries[key].msgid);
-          if (!msgid) return section;
+          if (!msgid) {
+            // If the section is empty, try to find it in the current sections
+            const currentSection = currentSections.find((cs) => {
+              const csPo = gettextParser.po.parse(cs);
+              const csContextKey = _.keys(csPo.translations)[0];
+              const csEntries = csPo.translations[csContextKey];
+              const csMsgid = Object.keys(csEntries).find((key) => csEntries[key].msgid);
+              return csMsgid === msgid;
+            });
+
+            if (currentSection) {
+              return currentSection;
+            }
+            return section;
+          }
           if (data[msgid]) {
             const updatedPo = _.merge({}, sectionPo, {
               translations: {