micro-build-server/BuildServer/lib/tasks/dotnetrewrite.js

"use strict";

const path = require("path");
const fs = require("fs");
const async = require("async");
const glob = require("glob");
const settings = require("../../settings");

module.exports = (params, processor) => ({
    "process": () => {
        if (processor.context.dotnetrewriterDone) {
            return processor.done();
        }

        processor.context.dotnetrewriterDone = true;

        const processAssemblyInfo = (appendInformationalVersion) => (originalContent, cb) => {
            let content = originalContent;

            if (!params.skipCodeSigning && !settings.skipCodeSigning) {
                content = content.replace(
                    /InternalsVisibleTo\s*\(\s*"([\w.]+)"\s*\)/g,
                    (match, p1) => `InternalsVisibleTo("${p1},PublicKey=${settings.codeSigningPublicKey}")`
                );
            }

            if (appendInformationalVersion) {
                content = `${content}\n[assembly: System.Reflection.AssemblyInformationalVersion("${processor.context.versionInfo}")]\n`;
            }

            return cb(null, content);
        };

        return glob("**/{InternalsVisible,AssemblyInfo}*.cs", { "cwd": processor.context.exported }, (err, files) => {
            if (err) {
                processor.onError(err);

                return processor.done();
            }

            processor.onInfo(`Found ${files.length} AssemblyInfo.cs files`);

            if (!files || !files.length) {
                processor.onWarn("No AssemblyInfo.cs found");

                return processor.done();
            }

            return async.parallel(files.map((file) => (callback) => async.waterfall([
                fs.readFile.bind(null, path.join(processor.context.exported, file), { "encoding": "utf8" }),
                processAssemblyInfo(file.toLowerCase().indexOf("assemblyinfo.cs") >= 0),
                fs.writeFile.bind(null, path.join(processor.context.exported, file))
            ], (err) => {
                if (err) {
                    processor.onError(`Unable to rewrite file ${file}: ${err}`);
                } else {
                    processor.onInfo(`Rewritten file ${file}`);
                }
                callback(err);
            })), processor.done.bind(processor));
        });
    }
});