tree_wasm.cpp

#include <emscripten/bind.h>
#include <sstream>
#include "BinaryTree.hpp"

using namespace emscripten;

// Wrapper class to expose BinaryTree<int> with JS-friendly methods
class BinaryTreeWrapper {
private:
    BinaryTree<int> tree;

public:
    BinaryTreeWrapper() : tree() {}

    void insert(int item) {
        tree.insert(item);
    }

    void remove(int item) {
        tree.remove(item);
    }

    bool find(int item) {
        return tree.find(item) != nullptr;
    }

    int nodeCount() {
        return tree.nodeCount();
    }

    int height() {
        return tree.height();
    }

    int leavesCount() {
        if (tree.nodeCount() == 0) return 0;
        return tree.leavesCount();
    }

    std::string preOrder() {
        std::ostringstream oss;
        tree.preOrder(oss);
        return oss.str();
    }

    std::string inOrder() {
        std::ostringstream oss;
        tree.inOrder(oss);
        return oss.str();
    }

    std::string postOrder() {
        std::ostringstream oss;
        tree.postOrder(oss);
        return oss.str();
    }
};

EMSCRIPTEN_BINDINGS(tree_module) {
    class_<BinaryTreeWrapper>("BinaryTree")
        .constructor<>()
        .function("insert", &BinaryTreeWrapper::insert)
        .function("remove", &BinaryTreeWrapper::remove)
        .function("find", &BinaryTreeWrapper::find)
        .function("nodeCount", &BinaryTreeWrapper::nodeCount)
        .function("height", &BinaryTreeWrapper::height)
        .function("leavesCount", &BinaryTreeWrapper::leavesCount)
        .function("preOrder", &BinaryTreeWrapper::preOrder)
        .function("inOrder", &BinaryTreeWrapper::inOrder)
        .function("postOrder", &BinaryTreeWrapper::postOrder);
}