Add installationMethod property to package.json

References yarnpkg#1139, yarnpkg#942, yarnpkg#1429, yarnpkg#1138

package.json

@@ -1,5 +1,6 @@
 {
   "name": "yarn",
+  "installationMethod": "unknown",
   "version": "0.16.2",
   "license": "BSD-2-Clause",
   "preferGlobal": true,

resources/win-chocolatey/tools/chocolateyinstall.ps1.in

@@ -13,3 +13,16 @@ $packageArgs = @{
 }
 
 Install-ChocolateyPackage @packageArgs
+
+# Update Yarn's package.json file so it can tell that it was installed via Chocolatey.
+if (Test-Path "${env:ProgramFiles(x86)}\Yarn\package.json") {
+  $path = "${env:ProgramFiles(x86)}\Yarn\package.json"
+} else {
+  $path = "$env:ProgramFiles\Yarn\package.json"
+}
+$script = @"
+  `$package_manifest = Get-Content -Path '$path' | ConvertFrom-Json
+  `$package_manifest.installationMethod = 'choco'
+  ConvertTo-Json -InputObject `$package_manifest | Set-Content -Path '$path'
+"@
+Start-ChocolateyProcessAsAdmin -Statements $script

resources/winsetup/YarnSetup.wixproj

@@ -49,6 +49,8 @@
     <!-- Delete .tar.gz from dist path, leaving only the files -->
     <Delete Files="$(YarnDistPath)\*.tar.gz" />
 
+    <Exec Command="node $(YarnDistPath)\..\scripts\set-installation-method.js $([System.IO.Path]::GetFullPath($(YarnDistPath)))\package.json msi" />
+
     <!-- Get Yarn version -->
     <Exec Command="node $(YarnDistPath)\bin\yarn.js --version" ConsoleToMSBuild="true">
       <Output TaskParameter="ConsoleOutput" PropertyName="YarnVersion" />

scripts/build-deb.sh

@@ -32,6 +32,7 @@ rm -rf $PACKAGE_TMPDIR
 mkdir -p $PACKAGE_TMPDIR/
 umask 0022 # Ensure permissions are correct (0755 for dirs, 0644 for files)
 tar zxf $TARBALL_NAME -C $PACKAGE_TMPDIR/
+PACKAGE_TMPDIR_ABSOLUTE=$(readlink -f $PACKAGE_TMPDIR)
 
 # Create Linux package structure
 mkdir -p $PACKAGE_TMPDIR/usr/share/yarn/
@@ -67,10 +68,12 @@ FPM="fpm --input-type dir --chdir $PACKAGE_TMPDIR --name yarn --version $VERSION
   `"--url https://yarnpkg.com/ --license BSD --description '$(cat resources/debian/description)'"
 
 ##### Build RPM (CentOS, Fedora) package
+./scripts/set-installation-method.js $PACKAGE_TMPDIR_ABSOLUTE/usr/share/yarn/package.json rpm
 eval "$FPM --output-type rpm  --architecture noarch --depends nodejs --category 'Development/Languages' ."
 mv *.rpm $OUTPUT_DIR
 
 ##### Build DEB (Debian, Ubuntu) package
+./scripts/set-installation-method.js $PACKAGE_TMPDIR_ABSOLUTE/usr/share/yarn/package.json deb
 mkdir -p $PACKAGE_TMPDIR/DEBIAN
 mkdir -p $PACKAGE_TMPDIR/usr/share/lintian/overrides/
 cp resources/debian/lintian-overrides $PACKAGE_TMPDIR/usr/share/lintian/overrides/yarn

scripts/build-dist.sh

@@ -16,6 +16,7 @@ rm -rf pack.tgz
 # Change this to "yarn install --production" once #1115 is fixed
 npm install --production
 rm -rf node_modules/*/test node_modules/*/dist
+../scripts/set-installation-method.js $(readlink -f package.json) tar
 cd ..
 
 tar -cvzf artifacts/yarn-v`dist/bin/yarn --version`.tar.gz dist/*

scripts/set-installation-method.js

@@ -0,0 +1,12 @@
+#!/usr/bin/env node
+/**
+ * Sets the installationMethod field in package.json. Useful for setting it in
+ * shell scripts.
+ */
+
+const fs = require('fs');
+
+const packageManifestFilename = process.argv[2];
+const packageManifest = require(packageManifestFilename);
+packageManifest.installationMethod = process.argv[3];
+fs.writeFileSync(packageManifestFilename, JSON.stringify(packageManifest, null, 2) + "\n");