package extra;

import static avian.Continuations.shift;

import avian.Function;
import avian.Continuations;

import java.util.concurrent.Callable;

public class ComposableContinuations {
  private static void expect(boolean v) {
    if (! v) throw new RuntimeException();
  }

  public static void main(String[] args) throws Exception {
    expect(2 * Continuations.<Integer,Integer>reset(new Callable<Integer>() {
          public Integer call() throws Exception {
            return 1 + shift
              (new Function<Function<Integer,Integer>,Integer>() {
                public Integer call(Function<Integer,Integer> continuation)
                  throws Exception
                {
                  return continuation.call(5);
                }
              });
          }
        }) == 12);

    expect(1 + Continuations.<Integer,Integer>reset(new Callable<Integer>() {
          public Integer call() throws Exception {
            return 2 * shift
              (new Function<Function<Integer,Integer>,Integer>() {
                public Integer call(Function<Integer,Integer> continuation)
                  throws Exception
                {
                  return continuation.call(continuation.call(4));
                }
              });
          }
        }) == 17);

    expect
      (equal
       (Continuations.<Cell<Integer>,Cell<Integer>>reset
        (new Callable<Cell<Integer>>() {
          public Cell<Integer> call() throws Exception {
            shift(new Function<Function<Cell<Integer>,Cell<Integer>>,
                  Cell<Integer>>()
                  {
                    public Cell<Integer> call
                      (Function<Cell<Integer>,Cell<Integer>> continuation)
                      throws Exception
                    {
                      return cons(1, continuation.call(null));
                    }
                  });

            shift(new Function<Function<Cell<Integer>,Cell<Integer>>,
                  Cell<Integer>>()
                  {
                    public Cell<Integer> call
                      (Function<Cell<Integer>,Cell<Integer>> continuation)
                      throws Exception
                    {
                      return cons(2, continuation.call(null));
                    }
                  });

            return null;
          }
        }), cons(1, cons(2, null))));

    expect
      (equal
       (Continuations.<String,Integer>reset
        (new Callable<String>() {
          public String call() throws Exception {
            return new String
              (shift(new Function<Function<byte[],String>,Integer>() {
                  public Integer call(Function<byte[],String> continuation)
                    throws Exception
                  {
                    return Integer.parseInt
                      (continuation.call(new byte[] { 0x34, 0x32 }));
                  }
                }), "UTF-8");
          }
        }), 42));
  }

  private static <T> boolean equal(T a, T b) {
    return (a == null && b == null) || (a != null && a.equals(b));
  }
  
  private static <Car> boolean equal(Cell<Car> a, Cell<Car> b) {
    while (a != null) {
      if (b == null || (! equal(a.car, b.car))) {
        return false;
      }
      a = a.cdr;
      b = b.cdr;
    }

    return b == null;
  }

  private static <Car> Cell<Car> cons(Car car, Cell<Car> cdr) {
    return new Cell(car, cdr);
  }

  private static class Cell<Car> {
    public final Car car;
    public final Cell<Car> cdr;

    public Cell(Car car, Cell<Car> cdr) {
      this.car = car;
      this.cdr = cdr;
    }

    public String toString() {
      StringBuilder sb = new StringBuilder();
      sb.append("(");
      for (Cell<Car> c = this; c != null; c = c.cdr) {
        sb.append(c.car);
        if (c.cdr != null) {
          sb.append(", ");
        }
      }
      sb.append(")");
      return sb.toString();
    }
  }
}