/*

 * Copyright (c) 2014, 2017, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code 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 General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package com.oracle.mxtool.junit;

import java.lang.reflect.Modifier;

import java.util.ArrayList;

import java.util.LinkedHashSet;

import java.util.List;

import java.util.Set;

import org.junit.runner.Description;

import org.junit.runner.Request;

import org.junit.runner.notification.Failure;

public final class MxJUnitRequest {

    private final Request request;

    final Set<Class<?>> classes;

    final String methodName;

    final List<Failure> missingClasses;

    private MxJUnitRequest(Request request, Set<Class<?>> classes, String methodName, List<Failure> missingClasses) {

        this.request = request;

        this.classes = classes;

        this.methodName = methodName;

        this.missingClasses = missingClasses;

    }

    public Request getRequest() {

        return request;

    }

    public static final class BuilderException extends Exception {

        private static final long serialVersionUID = 1L;

        private BuilderException(String msg) {

            super(msg);

        }

    }

    public static class Builder {

        private final Set<Class<?>> classes = new LinkedHashSet<>();

        private String methodName = null;

        private final List<Failure> missingClasses = new ArrayList<>();

        protected Class<?> resolveClass(String name) throws ClassNotFoundException {

            return Class.forName(name, false, Builder.class.getClassLoader());

        }

        public void addTestSpec(String arg) throws BuilderException {

            String className;

            /*

             * Entries of the form class#method are handled specially. Only one can be specified on

             * the command line as there's no obvious way to build a runner for multiple ones.

             */

            if (methodName != null) {

                throw new BuilderException("Only a single class and method can be specified: " + arg);

            } else if (arg.contains("#")) {

                String[] pair = arg.split("#");

                if (pair.length != 2) {

                    throw new BuilderException("Malformed class and method request: " + arg);

                } else if (!classes.isEmpty()) {

                    throw new BuilderException("Only a single class and method can be specified: " + arg);

                } else {

                    methodName = pair[1];

                    className = pair[0];

                }

            } else {

                className = arg;

            }

            try {

                Class<?> cls = resolveClass(className);

                if ((cls.getModifiers() & Modifier.ABSTRACT) == 0) {

                    classes.add(cls);

                }

            } catch (ClassNotFoundException e) {

                Description description = Description.createSuiteDescription(className);

                Failure failure = new Failure(description, e);

                missingClasses.add(failure);

            }

        }

        public MxJUnitRequest build() {

            Request request;

            if (methodName == null) {

                request = Request.classes(classes.toArray(new Class<?>[0]));

            } else {

                request = Request.method(classes.iterator().next(), methodName);

            }

            return new MxJUnitRequest(request, classes, methodName, missingClasses);

        }

    }

}