src/main/java/org/apache/sling/servlets/resolver/internal/ScriptResourceResolver.java

/*
 * 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.servlets.resolver.internal;

import java.util.Iterator;
import java.util.Map;
import java.util.function.Supplier;

import org.apache.sling.api.resource.LoginException;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.resource.ResourceResolver;
import org.apache.sling.api.resource.ResourceWrapper;
import org.apache.sling.api.wrappers.IteratorWrapper;
import org.apache.sling.api.wrappers.ResourceResolverWrapper;
import org.apache.sling.servlets.resolver.internal.resource.MergingServletResourceProvider;
import org.apache.sling.spi.resource.provider.ResolveContext;
import org.apache.sling.spi.resource.provider.ResourceContext;
import org.apache.sling.spi.resource.provider.ResourceProvider;
import org.jetbrains.annotations.NotNull;

public class ScriptResourceResolver extends ResourceResolverWrapper {
    private final ResourceResolver resolver;
    private final Supplier<MergingServletResourceProvider> providerSupplier;

    public ScriptResourceResolver(ResourceResolver resolver, Supplier<MergingServletResourceProvider> provider) {
        super(resolver);
        this.resolver = resolver;
        this.providerSupplier = provider;
    }

    public static ScriptResourceResolver wrap(ResourceResolver scriptResourceResolver, Supplier<MergingServletResourceProvider> provider) {
        return new ScriptResourceResolver(scriptResourceResolver, provider);
    }

    @Override
    public Iterable<Resource> getChildren(Resource parent) {
        return () -> listChildren(parent);
    }

    @Override
    public Resource getResource(String scriptPath) {
        MergingServletResourceProvider provider = this.providerSupplier.get();

        if (provider == null) {
            return super.getResource(scriptPath);
        }
        else {
            return wrap(provider.getResource(new ResolveContext<Object>() {
                @Override
                public ResourceResolver getResourceResolver() {
                    return ScriptResourceResolver.this;
                }

                @Override
                public Object getProviderState() {
                    return null;
                }

                @Override
                public ResolveContext<?> getParentResolveContext() {
                    return null;
                }

                @Override
                public ResourceProvider<?> getParentResourceProvider() {
                    return new ResourceProvider<Object>() {
                        @Override
                        public Resource getResource(ResolveContext<Object> ctx, String path, ResourceContext resourceContext, Resource parent) {
                            return resolver.getResource(path);
                        }

                        @Override
                        public Iterator<Resource> listChildren(ResolveContext<Object> ctx, Resource parent) {
                            return resolver.listChildren(parent);
                        }
                    };
                }
            }, scriptPath));
        }
    }

    @Override
    public Resource getResource(Resource base, @NotNull String path) {
        if (!path.startsWith("/") && base != null) {
            path = String.format("%s/%s", base.getPath(), path);
        }
        return getResource(path);
    }

    @Override
    public Iterator<Resource> listChildren(Resource parent) {
        MergingServletResourceProvider provider = this.providerSupplier.get();
        if (provider == null) {
            return super.listChildren(parent);
        }
        else {
            return wrap(provider.listChildren(new ResolveContext<Object>() {
                @Override
                public ResourceResolver getResourceResolver() {
                    return ScriptResourceResolver.this;
                }

                @Override
                public Object getProviderState() {
                    return null;
                }

                public ResolveContext<?> getParentResolveContext() {
                    return null;
                }

                public ResourceProvider<?> getParentResourceProvider() {
                    return new ResourceProvider<Object>() {
                        @Override
                        public Resource getResource(ResolveContext<Object> ctx, String path, ResourceContext resourceContext, Resource parent) {
                            return resolver.getResource(path);
                        }

                        @Override
                        public Iterator<Resource> listChildren(ResolveContext<Object> ctx, Resource parent) {
                            return resolver.listChildren(parent);
                        }
                    };
                }
            }, unwrap(parent)));
        }
    }

    private Resource wrap(Resource resource) {
        if (resource != null && !(resource.getResourceResolver() instanceof ScriptResourceResolver)) {
            resource = new ScriptResourceResolverResourceWrapper(resource);
        }
        return resource;
    }

    private Iterator<Resource> wrap(Iterator<Resource> iter) {
        if (iter != null) {
            iter = new IteratorWrapper<Resource>(iter){
                @Override
                public Resource next() {
                    return wrap(super.next());
                }
            };
        }
        return iter;
    }

    private Resource unwrap(Resource resource) {
        if (resource instanceof ScriptResourceResolverResourceWrapper) {
            resource = ((ResourceWrapper) resource).getResource();
        }
        return resource;
    }

    @Override
    public ScriptResourceResolver clone(Map<String, Object> o) throws LoginException {
        return ScriptResourceResolver.wrap(resolver.clone(o), providerSupplier);
    }

    private class ScriptResourceResolverResourceWrapper extends ResourceWrapper {
        public ScriptResourceResolverResourceWrapper(Resource resource) {
            super(resource);
        }

        @Override
        public ResourceResolver getResourceResolver() {
            return ScriptResourceResolver.this;
        }
    }
}