/*
 * 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.testing.mock.sling.services;

import java.net.URL;
import java.util.Collections;
import java.util.Set;
import java.util.UUID;

import org.apache.sling.settings.SlingSettingsService;
import org.jetbrains.annotations.NotNull;

/**
 * Mock implementation of {@link SlingSettingsService}.
 */
public final class MockSlingSettingService implements SlingSettingsService {

    private Set<String> runModes;
    private String slingId;

    /**
     * Instantiate with no default run modes.
     */
    public MockSlingSettingService() {
        this(Collections.emptySet());
    }

    /**
     * Instantiate with given run modes
     * @param defaultRunModes Run modes
     */
    public MockSlingSettingService(@NotNull Set<String> defaultRunModes) {
        this.runModes = defaultRunModes;
        this.slingId = UUID.randomUUID().toString();
    }

    @Override
    public Set<String> getRunModes() {
        return Collections.unmodifiableSet(this.runModes);
    }

    /**
     * @param runModes Rund modes
     */
    public void setRunModes(@NotNull Set<String> runModes) {
        this.runModes = runModes;
    }

    @Override
    public String getSlingId() {
        return slingId;
    }

    // --- unsupported operations ---
    @Override
    public String getAbsolutePathWithinSlingHome(String relativePath) {
        throw new UnsupportedOperationException();
    }

    @Override
    public String getSlingHomePath() {
        throw new UnsupportedOperationException();
    }

    @Override
    public URL getSlingHome() {
        throw new UnsupportedOperationException();
    }

    @Override
    public String getSlingName() {
        throw new UnsupportedOperationException();
    }

    @Override
    public String getSlingDescription() {
        throw new UnsupportedOperationException();
    }

    // part of Sling Setting Service 1.4.2
    /**
     * @param spec run mode spec
     * @return number of matching run modes
     */
    public int getBestRunModeMatchCountFromSpec(String spec) {
        throw new UnsupportedOperationException();
    }

}
