/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2015, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jboss.as.test.integration.domain.mixed;

import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OP_ADDR;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.PROFILE;

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.net.URISyntaxException;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.xml.namespace.QName;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;

import org.jboss.as.controller.Extension;
import org.jboss.as.controller.PathAddress;
import org.jboss.as.controller.ProcessType;
import org.jboss.as.controller.RunningMode;
import org.jboss.as.controller.RunningModeControl;
import org.jboss.as.controller.extension.ExtensionRegistry;
import org.jboss.as.controller.extension.RuntimeHostControllerInfoAccessor;
import org.jboss.as.subsystem.test.TestParser;
import org.jboss.dmr.ModelNode;
import org.jboss.staxmapper.XMLMapper;
import org.wildfly.build.configassembly.ConfigurationAssembler;
import org.wildfly.build.configassembly.SubsystemConfig;
import org.wildfly.build.configassembly.SubsystemInputStreamSources;
import org.wildfly.build.util.FileInputStreamSource;
import org.wildfly.build.util.InputStreamSource;

/**
 * Used by the domain adjuster to generate subsystem.xml files from existing templates for large/complex susbsystems.
 * Any extensions and socket binding groups must still be added manually.
 *
 *
 * @author Kabir Khan
 */
public class LegacySubsystemConfigurationUtil {

    private static final String SUBSYSTEM_OPEN = "<subsystem";
    private static final String SUBSYSTEM_CLOSE = "</subsystem>";
    private static final String TEST_NAMESPACE = "urn.org.jboss.test:1.0";

    final Extension extension;
    final String subsystemName;
    final String supplement;
    final String resourceName;

    public LegacySubsystemConfigurationUtil(Extension extension, String subsystemName, String supplement, String resourceName) {
        this.extension = extension;
        this.subsystemName = subsystemName;
        this.supplement = supplement;
        this.resourceName = resourceName;
    }

    public List<ModelNode> getSubsystemOperations() throws Exception {
        File file = createAssembly();
        String subsystemXml = extractSubsystemXml(file);
        List<ModelNode> list = parseSubsystemXml(subsystemXml);
        PathAddress profile = PathAddress.pathAddress(PROFILE, "full-ha");
        for (ModelNode op : list) {
            PathAddress address = PathAddress.pathAddress(op.get(OP_ADDR));
            op.get(OP_ADDR).set(profile.append(address).toModelNode());
        }
        return list;
    }

    private List<ModelNode> parseSubsystemXml(String subsystemXml) throws XMLStreamException {
        XMLMapper xmlMapper = XMLMapper.Factory.create();
        ExtensionRegistry extensionParsingRegistry = new ExtensionRegistry(ProcessType.HOST_CONTROLLER, new RunningModeControl(RunningMode.NORMAL), null, null, RuntimeHostControllerInfoAccessor.SERVER);
        TestParser testParser = new TestParser(subsystemName, extensionParsingRegistry);
        xmlMapper.registerRootElement(new QName(TEST_NAMESPACE, "test"), testParser);
        extension.initializeParsers(extensionParsingRegistry.getExtensionParsingContext("Test", xmlMapper));
        String xml = "<test xmlns=\"" + TEST_NAMESPACE + "\">" +
                subsystemXml +
                "</test>";
        final XMLStreamReader reader = XMLInputFactory.newInstance().createXMLStreamReader(new StringReader(xml));
        final List<ModelNode> operationList = new ArrayList<ModelNode>();
        xmlMapper.parseDocument(operationList, reader);
        return operationList;
    }


    private String extractSubsystemXml(File file) throws IOException {
        StringBuilder sb = new StringBuilder();
        BufferedReader reader = new BufferedReader(new FileReader(file));
        String line = reader.readLine();
        boolean inSusbsystem = false;
        while (line != null) {
            if (!inSusbsystem) {
                if (line.contains(SUBSYSTEM_OPEN)) {
                    inSusbsystem = true;
                    //This does not take into account things like <subsystem xmlns="..."/> but for simple subsystems
                    //like that this class should not be used anyway.
                    sb.append(line.substring(line.indexOf(SUBSYSTEM_OPEN)));
                }
            } else {
                if (!line.contains(SUBSYSTEM_CLOSE)) {
                    sb.append(line);
                } else {
                    sb.append(line.substring(0, line.indexOf(SUBSYSTEM_CLOSE) + SUBSYSTEM_CLOSE.length()));
                    break;
                }
            }
            line = reader.readLine();
        }
        return sb.toString();
    }

    private File createAssembly() throws IOException, XMLStreamException, URISyntaxException {
        final Map<String, Map<String, SubsystemConfig>> subsystemConfigs = new HashMap<>();
        final SubsystemInputStreamSources subsystemSources = createSubsystemInputStreamSources(subsystemConfigs);

        final File outputFile = createOutputFile();
        outputFile.delete();

        final URL url = this.getClass().getClassLoader().getResource("legacy-templates/test-template.xml");
        if (url == null) {
            throw new IllegalStateException("Can't find the template file");
        }
        final InputStreamSource template = new FileInputStreamSource(new File(url.toURI()));

        final ConfigurationAssembler assembler = new ConfigurationAssembler(subsystemSources, template, "server", subsystemConfigs, outputFile);
        assembler.assemble();
        return outputFile;
    }

    private SubsystemInputStreamSources createSubsystemInputStreamSources(Map<String, Map<String, SubsystemConfig>> subsystemConfigs) {
        final Map<String, URL> urls = new HashMap<>();
        addSubsystem(urls, subsystemConfigs, "messaging", "ha", "subsystem-templates/messaging.xml");

        return new SubsystemInputStreamSources() {
            @Override
            public InputStreamSource getInputStreamSource(String subsystem) {
                URL url = urls.get(subsystem);
                if (url == null) {
                    throw new IllegalArgumentException("No stream for " + subsystem);
                }
                return new InputStreamSource() {
                    @Override
                    public InputStream getInputStream() throws IOException {
                        return new BufferedInputStream(url.openStream());
                    }
                };
            }
        };
    }

    private void addSubsystem(Map<String, URL> urls, Map<String, Map<String, SubsystemConfig>> subsystemConfigs, String subsystem, String supplement, String resourceName) {
        URL url = this.getClass().getClassLoader().getResource(resourceName);

        if (url == null) {
            throw new IllegalArgumentException("Could not find " + resourceName);
        }

        urls.put(subsystem, url);
        Map<String, SubsystemConfig> config = Collections.singletonMap(subsystem, new SubsystemConfig(subsystem, supplement));
        subsystemConfigs.put("", config);
    }

    private File createOutputFile() throws IOException, URISyntaxException {
        final URL url = this.getClass().getProtectionDomain().getCodeSource().getLocation();
        final File target  = new File(url.toURI()).getParentFile();
        final File output = new File(target, subsystemName + "-legacy-xml");
        if (!output.exists()) {
            Files.createDirectories(Paths.get(output.getAbsolutePath()));
        }
        return new File(output, "test.xml");
    }}

@@ -4368,16 +4368,22 @@
 l, null,
+ null,
  Runtime

Use the updated ExtensionRegistry constructor