RatpackDslClosures.java

/*
 * Copyright 2015 the original author or authors.
 *
 * Licensed 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 ratpack.groovy.internal.capture;

import groovy.lang.Closure;
import org.codehaus.groovy.runtime.ComposedClosure;
import ratpack.func.Block;
import ratpack.func.Function;
import ratpack.groovy.Groovy;
import ratpack.groovy.internal.ClosureUtil;
import ratpack.util.Exceptions;
import ratpack.util.internal.Paths2;

import java.nio.charset.StandardCharsets;
import java.nio.file.Path;

public class RatpackDslClosures {

  private Closure<?> handlers;
  private Closure<?> bindings;
  private Closure<?> serverConfig;
  private Path thisScript;
  private boolean closed;

  public RatpackDslClosures(Path thisScript) {
    this.handlers = Closure.IDENTITY;
    this.bindings = Closure.IDENTITY;
    this.serverConfig = Closure.IDENTITY;
    this.thisScript = thisScript;
  }

  public Closure<?> getHandlers() {
    return handlers;
  }

  public Closure<?> getBindings() {
    return bindings;
  }

  public Closure<?> getServerConfig() {
    return serverConfig;
  }

  public void setHandlers(Closure<?> handlers) {
    assertAtTopLevelOfRatpackDsl("handlers");
    this.handlers = new ComposedClosure<>(handlers, this.handlers);
  }

  public void setBindings(Closure<?> bindings) {
    assertAtTopLevelOfRatpackDsl("bindings");
    this.bindings = new ComposedClosure<>(bindings, this.bindings);
  }

  public void setServerConfig(Closure<?> serverConfig) {
    assertAtTopLevelOfRatpackDsl("serverConfig");
    this.serverConfig = new ComposedClosure<>(serverConfig, this.serverConfig);
  }

  public void include(Path path) {
    assertAtTopLevelOfRatpackDsl("include");
    Exceptions.uncheck(() -> {
      Path resolvedPath = path;
      if (thisScript != null && !path.isAbsolute()) {
        resolvedPath = thisScript.resolveSibling(path);
      }
      String script = Paths2.readText(resolvedPath, StandardCharsets.UTF_8);
      RatpackDslClosures closures = new RatpackDslScriptCapture(false, new String[]{}, RatpackDslBacking::new).apply(resolvedPath, script);
      this.setServerConfig(closures.getServerConfig());
      this.setBindings(closures.getBindings());
      this.setHandlers(closures.getHandlers());
    });
  }

  private void assertAtTopLevelOfRatpackDsl(String methodName) {
    if (closed) {
      throw new IllegalStateException(methodName + " {} DSL method can only be used at the top level of the ratpack {} block");
    }
  }

  public static RatpackDslClosures capture(Function<? super RatpackDslClosures, ? extends Groovy.Ratpack> function, Path script, Block action) throws Exception {
    RatpackDslClosures closures = new RatpackDslClosures(script);
    Groovy.Ratpack receiver = function.apply(closures);
    RatpackScriptBacking.withBacking(closure -> ClosureUtil.configureDelegateFirst(receiver, closure), action);
    closures.closed = true;
    return closures;
  }

}