/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.sling.feature.resolver;

import org.apache.sling.feature.Application;
import org.apache.sling.feature.Artifact;
import org.apache.sling.feature.Feature;
import org.apache.sling.feature.analyser.Analyser;
import org.apache.sling.feature.io.ArtifactManager;
import org.apache.sling.feature.io.ArtifactManagerConfig;
import org.apache.sling.feature.io.json.FeatureJSONReader;
import org.apache.sling.feature.io.json.FeatureJSONReader.SubstituteVariables;
import org.apache.sling.feature.scanner.BundleDescriptor;
import org.apache.sling.feature.scanner.Scanner;
import org.apache.sling.feature.scanner.impl.BundleDescriptorImpl;
import org.junit.Ignore;
import org.junit.Test;

import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.ArrayList;
import java.util.List;

import static org.junit.Assert.fail;

public class AnalyserTest {
    @Test
    public void testAnalyserWithCompleteFeature() throws Exception {
        final Scanner scanner = new Scanner(new ArtifactManagerConfig());
        final Analyser analyser = new Analyser(scanner);
        try ( final Reader reader = new InputStreamReader(AnalyserTest.class.getResourceAsStream("/feature_complete.json"),
                "UTF-8") ) {
            Feature feature = FeatureJSONReader.read(reader, "feature", SubstituteVariables.RESOLVE);

            Application app = ApplicationResolverAssembler.assembleApplication(null, ArtifactManager.getArtifactManager(new ArtifactManagerConfig()),
                    getTestResolver(), feature);

            analyser.analyse(app);
        }
    }

    @Test
    public void testAnalyserWithInCompleteFeature() throws Exception {
        final Scanner scanner = new Scanner(new ArtifactManagerConfig());
        final Analyser analyser = new Analyser(scanner);
        try ( final Reader reader = new InputStreamReader(AnalyserTest.class.getResourceAsStream("/feature_incomplete.json"),
                "UTF-8") ) {
            Feature feature = FeatureJSONReader.read(reader, "feature", SubstituteVariables.RESOLVE);

            Application app = ApplicationResolverAssembler.assembleApplication(null, ArtifactManager.getArtifactManager(new ArtifactManagerConfig()),
                    getTestResolver(), feature);

            try {
                analyser.analyse(app);

                fail("Expected an exception");
            }
            catch (Exception ex) {
                // Pass
            }
        }
    }

    private FeatureResolver getTestResolver() {
        return new FeatureResolver() {
            @Override
            public void close() throws Exception {
            }

            @Override
            public List<FeatureResource> orderResources(List<Feature> features) {
                try {
                    // Just return the resources in the same order as they are listed in the features
                    List<FeatureResource> l = new ArrayList<>();

                    for (Feature f : features) {
                        for (Artifact a : f.getBundles()) {
                            BundleDescriptor bd = getBundleDescriptor(ArtifactManager.getArtifactManager(new ArtifactManagerConfig()), a);
                            l.add(new TestBundleResourceImpl(bd, f));
                        }
                    }

                    return l;
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }

            private BundleDescriptor getBundleDescriptor(ArtifactManager artifactManager, Artifact b) throws IOException {
                final File file = artifactManager.getArtifactHandler(b.getId().toMvnUrl()).getFile();
                if ( file == null ) {
                    throw new IOException("Unable to find file for " + b.getId());
                }

                return new BundleDescriptorImpl(b, file, -1);
            }
        };
    }
}
