feat(package): .serialize() also writes package-lock version

core/package/__tests__/core-package.test.js

@@ -2,11 +2,15 @@
 
 jest.mock("load-json-file");
 jest.mock("write-pkg");
+jest.mock("write-json-file");
+jest.mock("fs");
 
 const os = require("os");
 const path = require("path");
 const loadJsonFile = require("load-json-file");
 const writePkg = require("write-pkg");
+const writeJsonFile = require("write-json-file");
+const fs = require("fs");
 
 // file under test
 const Package = require("..");
@@ -29,6 +33,13 @@ describe("Package", () => {
     });
   });
 
+  describe("get .lockFileLocation", () => {
+    it("should return the package-lock location", () => {
+      const pkg = factory({ name: "lock-file-location" });
+      expect(pkg.lockFileLocation).toBe(path.normalize("/root/path/to/lock-file-location/package-lock.json"));
+    });
+  });
+
   describe("get .resolved", () => {
     it("returns npa.Result relative to rootPath, always posix", () => {
       const pkg = factory({ name: "get-resolved" });
@@ -257,8 +268,11 @@ describe("Package", () => {
   });
 
   describe(".serialize()", () => {
-    it("writes changes to disk", async () => {
-      writePkg.mockImplementation(() => Promise.resolve());
+    it("writes changes to package.json to disk", async () => {
+      writePkg.mockImplementationOnce(() => Promise.resolve());
+      fs.existsSync.mockImplementationOnce(() => false);
+      writeJsonFile.mockImplementationOnce(() => Promise.resolve());
+      loadJsonFile.mockImplementationOnce(() => Promise.resolve());
 
       const pkg = factory({ name: "serialize-me" });
       const result = await pkg.set("woo", "hoo").serialize();
@@ -271,6 +285,40 @@ describe("Package", () => {
           woo: "hoo",
         })
       );
+      expect(fs.existsSync).toHaveBeenCalledWith("/root/path/to/serialize-me/package-lock.json");
+      expect(loadJsonFile).not.toHaveBeenCalled();
+      expect(writeJsonFile).not.toHaveBeenCalled();
+    });
+
+    it("writes changes to package-lock to disk", async () => {
+      writePkg.mockImplementationOnce(() => Promise.resolve());
+      fs.existsSync.mockImplementationOnce(() => true);
+      writeJsonFile.mockImplementationOnce(() => Promise.resolve());
+      loadJsonFile.mockImplementationOnce(() => Promise.resolve({ name: "something" }));
+
+      const pkg = factory({ name: "serialize-me-lock", version: "1.0.0" });
+      expect(pkg.version).toBe("1.0.0");
+
+      const result = await pkg.set("version", "2.0.0").serialize();
+      expect(result).toBe(pkg);
+      expect(result.version).toBe("2.0.0");
+
+      expect(writePkg).toHaveBeenLastCalledWith(
+        pkg.manifestLocation,
+        expect.objectContaining({
+          name: "serialize-me-lock",
+          version: "2.0.0",
+        })
+      );
+      expect(fs.existsSync).toHaveBeenCalledWith("/root/path/to/serialize-me-lock/package-lock.json");
+      expect(loadJsonFile).toHaveBeenCalledWith("/root/path/to/serialize-me-lock/package-lock.json");
+      expect(writeJsonFile).toHaveBeenCalledWith(
+        "/root/path/to/serialize-me-lock/package-lock.json",
+        expect.objectContaining({
+          version: "2.0.0",
+        }),
+        { detectIndent: true }
+      );
     });
   });
 });

core/package/index.js

@@ -4,6 +4,9 @@ const npa = require("npm-package-arg");
 const path = require("path");
 const loadJsonFile = require("load-json-file");
 const writePkg = require("write-pkg");
+const fs = require("fs");
+const writeJsonFile = require("write-json-file");
+const pWaterfall = require("p-waterfall");
 
 // symbol used to "hide" internal state
 const PKG = Symbol("pkg");
@@ -30,6 +33,12 @@ function shallowCopy(json) {
   }, {});
 }
 
+// determine if a package-lock.json file exists
+function hasPackageLock(location) {
+  const packageLock = path.resolve(location);
+  return fs.existsSync(packageLock);
+}
+
 class Package {
   constructor(pkg, location, rootPath = location) {
     // npa will throw an error if the name is invalid
@@ -77,6 +86,9 @@ class Package {
       manifestLocation: {
         value: path.join(location, "package.json"),
       },
+      lockFileLocation: {
+        value: path.join(location, "package-lock.json"),
+      },
       nodeModulesLocation: {
         value: path.join(location, "node_modules"),
       },
@@ -181,7 +193,27 @@ class Package {
    * @returns {Promise} resolves when write finished
    */
   serialize() {
-    return writePkg(this.manifestLocation, this[PKG]).then(() => this);
+    return writePkg(this.manifestLocation, this[PKG])
+      .then(() => this.writePkgLock())
+      .then(() => this);
+  }
+
+  /**
+   * Write package-lock changes to disk
+   * @returns {Promise} resolves when write finished
+   */
+  writePkgLock() {
+    // We only care about version in the lock file; a package's name never changes
+    const lockChanges = { version: this[PKG].version };
+
+    const tasks = [
+      () => loadJsonFile(this.lockFileLocation),
+      lockFileData => Promise.resolve(Object.assign({}, lockFileData, lockChanges)),
+      updatedLockFileData =>
+        writeJsonFile(this.lockFileLocation, updatedLockFileData, { detectIndent: true }),
+    ];
+
+    return hasPackageLock(this.lockFileLocation) ? pWaterfall(tasks) : Promise.resolve();
   }
 
   /**

core/package/package.json

@@ -33,6 +33,8 @@
   "dependencies": {
     "load-json-file": "^4.0.0",
     "npm-package-arg": "^6.1.0",
+    "p-waterfall": "2.1.0",
+    "write-json-file": "4.1.0",
     "write-pkg": "^3.1.0"
   }
 }