/*
 * 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.
 */

/* $Rev$ $Date$ */

/**
 * Test script evaluator.
 */

#include <assert.h>
#include <iostream>
#include <string>
#include <sstream>
#include "driver.hpp"

namespace tuscany {
namespace eval {

bool testEnv() {
    gc_pool pool;
    Env globalEnv = list<value>();
    Env env = extendEnvironment(mklist<value>("a"), mklist<value>(1), globalEnv, pool);
    defineVariable("x", env, env);
    assert(lookupVariableValue(value("x"), env) == env);
    assert(lookupVariableValue("a", env) == value(1));
    return true;
}

bool testEnvGC() {
    resetLambdaCounters();
    resetListCounters();
    resetValueCounters();
    testEnv();
    assert(countValues == 0);
    assert(countLambdas == 0);
    assert(countlists == 0);
    //printLambdaCounters();
    //printListCounters();
    //printValueCounters();
    return true;
}

bool testRead() {
    std::istringstream is("abcd");
    assert(readValue(is) == "abcd");

    std::istringstream is2("123");
    assert(readValue(is2) == value(123));

    std::istringstream is3("(abcd)");
    assert(readValue(is3) == mklist(value("abcd")));

    std::istringstream is4("(abcd xyz)");
    assert(readValue(is4) == mklist<value>("abcd", "xyz"));

    std::istringstream is5("(abcd (xyz tuv))");
    assert(readValue(is5) == mklist<value>("abcd", mklist<value>("xyz", "tuv")));

    return true;
}

bool testWrite() {
    const list<value> i = list<value>()
            << (list<value>() << "item" << "cart-53d67a61-aa5e-4e5e-8401-39edeba8b83b"
                << (list<value>() << "item"
                    << (list<value>() << "name" << "Apple")
                    << (list<value>() << "price" << "$2.99")))
            << (list<value>() << "item" << "cart-53d67a61-aa5e-4e5e-8401-39edeba8b83c"
                << (list<value>() << "item"
                    << (list<value>() << "name" << "Orange")
                    << (list<value>() << "price" << "$3.55")));
    const list<value> a = cons<value>("Feed", cons<value>("feed-1234", i));
    std::ostringstream os;
    writeValue(a, os);
    std::istringstream is(os.str());
    assert(readValue(is) == a);
    return true;
}

const std::string testSchemeNumber(
  "(define (testNumber) (if (= 1 1) (display \"testNumber ok\") (error \"testNumber\"))) "
  "(testNumber)");

const std::string testSchemeString(
  "(define (testString) (if (= \"abc\" \"abc\") (display \"testString ok\") (error \"testString\"))) "
  "(testString)");

const std::string testSchemeDefinition(
  "(define a \"abc\") (define (testDefinition) (if (= a \"abc\") (display \"testDefinition ok\") (error \"testDefinition\"))) "
  "(testDefinition)");

const std::string testSchemeIf(
  "(define (testIf) (if (= \"abc\" \"abc\") (if (= \"xyz\" \"xyz\") (display \"testIf ok\") (error \"testNestedIf\")) (error \"testIf\"))) "
  "(testIf)");

const std::string testSchemeCond(
  "(define (testCond) (cond ((= \"abc\" \"abc\") (display \"testCond ok\")) (else (error \"testIf\"))))"
  "(testCond)");

const std::string testSchemeBegin(
  "(define (testBegin) "
    "(begin "
        "(define a \"abc\") "
        "(if (= a \"abc\") (display \"testBegin1 ok\") (error \"testBegin\")) "
        "(define x \"xyz\") "
        "(if (= x \"xyz\") (display \"testBegin2 ok\") (error \"testBegin\")) "
      ") "
  ") "
  "(testBegin)");

const std::string testSchemeLambda(
  "(define sqrt (lambda (x) (* x x))) "
  "(define (testLambda) (if (= 4 (sqrt 2)) (display \"testLambda ok\") (error \"testLambda\"))) "
  "(testLambda)");

const std::string testSchemeForward(
  "(define (testLambda) (if (= 4 (sqrt 2)) (display \"testForward ok\") (error \"testForward\"))) "
  "(define sqrt (lambda (x) (* x x))) "
  "(testLambda)");

bool contains(const std::string& str, const std::string& pattern) {
    return str.find(pattern) != str.npos;
}

const std::string evalOutput(const std::string& scm) {
    std::istringstream is(scm);
    std::ostringstream os;
    evalDriverRun(is, os);
    return os.str();
}

bool testEval() {
    assert(contains(evalOutput(testSchemeNumber), "testNumber ok"));
    assert(contains(evalOutput(testSchemeString), "testString ok"));
    assert(contains(evalOutput(testSchemeDefinition), "testDefinition ok"));
    assert(contains(evalOutput(testSchemeIf), "testIf ok"));
    assert(contains(evalOutput(testSchemeCond), "testCond ok"));
    assert(contains(evalOutput(testSchemeBegin), "testBegin1 ok"));
    assert(contains(evalOutput(testSchemeBegin), "testBegin2 ok"));
    assert(contains(evalOutput(testSchemeLambda), "testLambda ok"));
    assert(contains(evalOutput(testSchemeForward), "testForward ok"));
    return true;
}

bool testEvalExpr() {
    gc_pool pool;
    const value exp = mklist<value>("+", 2, 3);
    Env env = setupEnvironment(pool);
    const value r = evalExpr(exp, env, pool);
    assert(r == value(5));
    return true;
}

bool testEvalRun() {
    evalDriverRun(std::cin, std::cout);
    return true;
}

const value mult(const list<value>& args) {
    const double x = car(args);
    const double y = cadr(args);
    return x * y;
}

const std::string testReturnLambda(
  "(define (testReturnLambda) * )");

const std::string testCallLambda(
  "(define (testCallLambda l x y) (l x y))");

bool testEvalLambda() {
    gc_pool pool;
    Env env = setupEnvironment(pool);

    const value trl = mklist<value>("testReturnLambda");
    std::istringstream trlis(testReturnLambda);
    const value trlv = evalScript(trl, trlis, env, pool);

    std::istringstream tclis(testCallLambda);
    const value tcl = cons<value>("testCallLambda", quotedParameters(mklist<value>(trlv, 2, 3)));
    const value tclv = evalScript(tcl, tclis, env, pool);
    assert(tclv == value(6));

    std::istringstream tcelis(testCallLambda);
    const value tcel = cons<value>("testCallLambda", quotedParameters(mklist<value>(primitiveProcedure(mult), 3, 4)));
    const value tcelv = evalScript(tcel, tcelis, env, pool);
    assert(tcelv == value(12));
    return true;
}

bool testEvalGC() {
    resetLambdaCounters();
    resetListCounters();
    resetValueCounters();
    testEval();
    testEvalExpr();
    testEvalLambda();
    assert(countValues == 0);
    assert(countLambdas == 0);
    assert(countlists == 0);
    //printLambdaCounters();
    //printListCounters();
    //printValueCounters();
    return true;
}

}
}

int main() {
    std::cout << "Testing..." << std::endl;

    tuscany::eval::testEnv();
    tuscany::eval::testEnvGC();
    tuscany::eval::testRead();
    tuscany::eval::testWrite();
    tuscany::eval::testEval();
    tuscany::eval::testEvalExpr();
    tuscany::eval::testEvalLambda();
    tuscany::eval::testEvalGC();

    std::cout << "OK" << std::endl;
    return 0;
}