org.sonatype.nexus.error.reporting.bundle.NexusXmlAssembler.java Source code

Introduction

Here is the source code for org.sonatype.nexus.error.reporting.bundle.NexusXmlAssembler.java

Source

/*
 * Sonatype Nexus (TM) Open Source Version
 * Copyright (c) 2007-2012 Sonatype, Inc.
 * All rights reserved. Includes the third-party code listed at http://links.sonatype.com/products/nexus/oss/attributions.
 *
 * This program and the accompanying materials are made available under the terms of the Eclipse Public License Version 1.0,
 * which accompanies this distribution and is available at http://www.eclipse.org/legal/epl-v10.html.
 *
 * Sonatype Nexus (TM) Professional Version is available from Sonatype, Inc. "Sonatype" and "Sonatype Nexus" are trademarks
 * of Sonatype, Inc. Apache Maven is a trademark of the Apache Software Foundation. M2eclipse is a trademark of the
 * Eclipse Foundation. All other trademarks are the property of their respective owners.
 */
package org.sonatype.nexus.error.reporting.bundle;

import java.io.IOException;
import java.io.OutputStream;

import javax.inject.Inject;
import javax.inject.Named;

import com.google.common.io.ByteStreams;
import com.google.common.io.CharStreams;
import com.google.common.io.Files;
import com.google.common.io.OutputSupplier;
import org.codehaus.plexus.swizzle.IssueSubmissionException;
import org.codehaus.plexus.swizzle.IssueSubmissionRequest;
import org.codehaus.plexus.util.IOUtil;
import org.sonatype.nexus.configuration.application.NexusConfiguration;
import org.sonatype.nexus.configuration.model.Configuration;
import org.sonatype.nexus.configuration.model.ConfigurationHelper;
import org.sonatype.nexus.configuration.model.io.xpp3.NexusConfigurationXpp3Writer;
import org.sonatype.sisu.pr.bundle.Bundle;
import org.sonatype.sisu.pr.bundle.BundleAssembler;
import org.sonatype.sisu.pr.bundle.ManagedBundle;
import org.sonatype.sisu.pr.bundle.StorageManager;

/**
 * Adds nexus.xml configuration file to the error report bundle.
 * Passwords will be masked.
 */
@Named("nexus.xml")
public class NexusXmlAssembler extends AbstractXmlAssembler implements BundleAssembler {

    private ConfigurationHelper configHelper;

    private NexusConfiguration nexusConfig;

    private StorageManager storageManager;

    @Inject
    public NexusXmlAssembler(final ConfigurationHelper configHelper, final NexusConfiguration nexusConfig,
            final StorageManager storageManager) {
        this.configHelper = configHelper;
        this.nexusConfig = nexusConfig;
        this.storageManager = storageManager;
    }

    @SuppressWarnings("deprecation")
    @Override
    public boolean isParticipating(IssueSubmissionRequest request) {
        return nexusConfig.getConfigurationModel() != null;
    }

    @SuppressWarnings("deprecation")
    @Override
    public Bundle assemble(IssueSubmissionRequest request) throws IssueSubmissionException {
        OutputStream out = null;
        try {
            final ManagedBundle bundle = storageManager.createBundle("nexus.xml", "application/xml");
            final Configuration configuration = configHelper.maskPasswords(nexusConfig.getConfigurationModel());

            // No config ?
            if (configuration != null) {
                NexusConfigurationXpp3Writer writer = new NexusConfigurationXpp3Writer();
                out = bundle.getOutputStream();
                writer.write(out, configuration);
                out.close();
            } else {
                ByteStreams.write("Got no configuration from config helper".getBytes("utf-8"),
                        new OutputSupplier<OutputStream>() {
                            @Override
                            public OutputStream getOutput() throws IOException {
                                return bundle.getOutputStream();
                            }
                        });
            }
            return bundle;
        } catch (IOException e) {
            IOUtil.close(out);
            throw new IssueSubmissionException("Could not assemble nexus.xml: " + e.getMessage(), e);
        }
    }
}