package io.jooby;

import org.jooby.funzy.Throwing;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.nio.ByteBuffer;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.concurrent.Executor;

public interface Context {

  class Dispatched extends RuntimeException {
    public final Executor executor;

    public Dispatched(@Nullable Executor executor) {
      this.executor = executor;
    }
  }

  /**
   * **********************************************************************************************
   * **** Request methods *************************************************************************
   * **********************************************************************************************
   */
  @Nonnull QueryString query();

  @Nonnull Value query(String name);

  @Nonnull String method();

  @Nonnull String path();

  @Nonnull Route route();

  @Nonnull Context route(@Nonnull Route route);

  boolean isInIoThread();

  default @Nonnull Context dispatch() {
    return dispatch(null);
  }

  @Nonnull Context dispatch(@Nullable Executor executor);

  @Nonnull Context detach();

  /**
   * **********************************************************************************************
   * **** Response methods *************************************************************************
   * **********************************************************************************************
   */

  @Nonnull Context reset();

  @Nonnull OutputStream outputStream();

  default @Nonnull Context outputStream(@Nonnull Throwing.Consumer<OutputStream> callback) {
    try (OutputStream w = outputStream()) {
      callback.accept(w);
    } catch (Throwable x) {
      throw Throwing.sneakyThrow(x);
    } finally {
      end();
    }
    return this;
  }

  default @Nonnull Writer writer() {
    return writer(StandardCharsets.UTF_8);
  }

  default @Nonnull Context writer(@Nonnull Throwing.Consumer<Writer> writer) {
    return writer(StandardCharsets.UTF_8, writer);
  }

  default @Nonnull Context writer(@Nonnull Charset charset, @Nonnull Throwing.Consumer<Writer> writer) {
    try (Writer w = writer(charset)) {
      writer.accept(w);
    } catch (Throwable x) {
      throw Throwing.sneakyThrow(x);
    } finally {
      end();
    }
    return this;
  }

  default @Nonnull Writer writer(@Nonnull Charset charset) {
    return new OutputStreamWriter(outputStream(), charset);
  }

  @Nonnull Context after(@Nonnull Route.After after);

  default @Nonnull Context header(@Nonnull String name, int value) {
    return header(name, Integer.toString(value));
  }

  @Nonnull Context header(@Nonnull String name, @Nonnull String value);

  int status();

  @Nonnull Context status(int status);

  @Nonnull Context length(long length);

  @Nonnull Context type(@Nonnull String contentType);

  default @Nonnull Context write(@Nonnull String chunk) {
    return write(chunk, StandardCharsets.UTF_8);
  }

  @Nonnull Context write(@Nonnull String chunk, @Nonnull Charset charset);

  @Nonnull Context write(byte[] chunk);

  @Nonnull Context write(@Nonnull ByteBuffer chunk);

  default @Nonnull Context send(@Nonnull String response) {
    return send(response, StandardCharsets.UTF_8);
  }

  @Nonnull Context send(@Nonnull String response, @Nonnull Charset charset);

  @Nonnull Context send(@Nonnull byte[] chunk);

  @Nonnull Context send(@Nonnull ByteBuffer chunk);

  @Nonnull Context end();

  boolean committed();
}