#ifndef JSONPULL_H
#define JSONPULL_H

#include <stdio.h>
#include <sys/types.h>
#include <cassert>
#include <memory>
#include <string>
#include <vector>

typedef enum json_type {
	// These types can be returned by json_read()
	JSON_HASH,
	JSON_ARRAY,
	JSON_NUMBER,
	JSON_STRING,
	JSON_TRUE,
	JSON_FALSE,
	JSON_NULL,

	// These and JSON_HASH and JSON_ARRAY can be called back by json_read_with_separators()
	JSON_COMMA,
	JSON_COLON,

	// These are only used internally as expectations of what comes next
	JSON_ITEM,
	JSON_KEY,
	JSON_VALUE,
} json_type;

struct json_object;
struct json_pull;

typedef std::shared_ptr<json_object> json_object_ptr;
typedef std::shared_ptr<json_pull> json_pull_ptr;

// json_object is a small base type that just records the JSON type and
// the back-pointers to its parent and parser. The actual value payload
// lives in a type-specific subclass (json_number, json_string, json_array,
// json_hash), so that JSON_TRUE / JSON_FALSE / JSON_NULL nodes pay only
// the base-class cost and a JSON_HASH does not also drag along a string
// or a number field. Type-tagged accessor methods on the base class
// downcast and return references to the underlying subclass storage.
//
// Children are owned by their parent (via std::vector<json_object_ptr>
// inside json_array / json_hash); the raw `parent` and `parser`
// back-pointers stay valid as long as the node is attached to the tree.
// json_disconnect() splices a node out of its parent and walks the
// detached subtree clearing those back-pointers so the subtree can
// outlive the original parser.
//
// json_object intentionally has no virtual functions and no virtual
// destructor: subclasses are constructed via std::make_shared<json_xxx>(),
// and std::shared_ptr remembers the deleter from the original type, so
// destroying a shared_ptr<json_object> that actually points at a
// json_string still runs ~json_string(). Dispatch on `type` is what the
// rest of the code already does. The accessor methods assert at debug
// time that the type matches before downcasting.

struct json_object {
	json_object *parent = nullptr;
	json_pull *parser = nullptr;

	json_type type;
	int expect = 0;	 // used by the parser on JSON_ARRAY / JSON_HASH nodes

	json_object(json_type t) : type(t) {}
	json_object(json_type t, json_object *p, json_pull *pl) : parent(p), parser(pl), type(t) {}

	// Type-tagged accessors. Each one asserts that the receiver is of
	// the right kind, then downcasts to the storage in the appropriate
	// subclass. Inline so the assert and cast disappear at -O.
	inline std::string &string();
	inline const std::string &string() const;

	inline double &number();
	inline double number() const;
	inline unsigned long long &large_unsigned();
	inline unsigned long long large_unsigned() const;
	inline long long &large_signed();
	inline long long large_signed() const;

	inline std::vector<json_object_ptr> &array();
	inline const std::vector<json_object_ptr> &array() const;

	inline std::vector<json_object_ptr> &keys();
	inline const std::vector<json_object_ptr> &keys() const;
	inline std::vector<json_object_ptr> &values();
	inline const std::vector<json_object_ptr> &values() const;
};

struct json_number : json_object {
	double number_value = 0;
	unsigned long long large_unsigned_value = 0;
	long long large_signed_value = 0;

	json_number() : json_object(JSON_NUMBER) {}
	json_number(json_object *p, json_pull *pl) : json_object(JSON_NUMBER, p, pl) {}
};

struct json_string : json_object {
	std::string string_value;

	json_string() : json_object(JSON_STRING) {}
	json_string(json_object *p, json_pull *pl) : json_object(JSON_STRING, p, pl) {}
};

struct json_array : json_object {
	std::vector<json_object_ptr> array_value;

	json_array() : json_object(JSON_ARRAY) {}
	json_array(json_object *p, json_pull *pl) : json_object(JSON_ARRAY, p, pl) {}
};

struct json_hash : json_object {
	std::vector<json_object_ptr> keys_value;
	std::vector<json_object_ptr> values_value;

	json_hash() : json_object(JSON_HASH) {}
	json_hash(json_object *p, json_pull *pl) : json_object(JSON_HASH, p, pl) {}
};

inline std::string &json_object::string() {
	assert(type == JSON_STRING);
	return static_cast<json_string *>(this)->string_value;
}
inline const std::string &json_object::string() const {
	assert(type == JSON_STRING);
	return static_cast<const json_string *>(this)->string_value;
}

inline double &json_object::number() {
	assert(type == JSON_NUMBER);
	return static_cast<json_number *>(this)->number_value;
}
inline double json_object::number() const {
	assert(type == JSON_NUMBER);
	return static_cast<const json_number *>(this)->number_value;
}
inline unsigned long long &json_object::large_unsigned() {
	assert(type == JSON_NUMBER);
	return static_cast<json_number *>(this)->large_unsigned_value;
}
inline unsigned long long json_object::large_unsigned() const {
	assert(type == JSON_NUMBER);
	return static_cast<const json_number *>(this)->large_unsigned_value;
}
inline long long &json_object::large_signed() {
	assert(type == JSON_NUMBER);
	return static_cast<json_number *>(this)->large_signed_value;
}
inline long long json_object::large_signed() const {
	assert(type == JSON_NUMBER);
	return static_cast<const json_number *>(this)->large_signed_value;
}

inline std::vector<json_object_ptr> &json_object::array() {
	assert(type == JSON_ARRAY);
	return static_cast<json_array *>(this)->array_value;
}
inline const std::vector<json_object_ptr> &json_object::array() const {
	assert(type == JSON_ARRAY);
	return static_cast<const json_array *>(this)->array_value;
}

inline std::vector<json_object_ptr> &json_object::keys() {
	assert(type == JSON_HASH);
	return static_cast<json_hash *>(this)->keys_value;
}
inline const std::vector<json_object_ptr> &json_object::keys() const {
	assert(type == JSON_HASH);
	return static_cast<const json_hash *>(this)->keys_value;
}
inline std::vector<json_object_ptr> &json_object::values() {
	assert(type == JSON_HASH);
	return static_cast<json_hash *>(this)->values_value;
}
inline const std::vector<json_object_ptr> &json_object::values() const {
	assert(type == JSON_HASH);
	return static_cast<const json_hash *>(this)->values_value;
}

struct json_pull {
	const char *error = nullptr;  // points at a string literal; no allocation
	int line = 1;

	ssize_t (*read)(struct json_pull *, char *buf, size_t n) = nullptr;
	void *source = nullptr;
	std::vector<char> buffer;
	ssize_t buffer_tail = 0;
	ssize_t buffer_head = 0;

	// Stack of currently-open containers; the top is the innermost container
	// being parsed. Replaces the previous single `container` pointer / parent
	// walk, which previously required enable_shared_from_this<json_object>
	// on every json_object instance (16 extra bytes per node).
	std::vector<json_object_ptr> container_stack;
	json_object_ptr root;

	std::string number_buffer;
};

json_pull_ptr json_begin_file(FILE *f);
json_pull_ptr json_begin_string(const char *s);

json_pull_ptr json_begin(ssize_t (*read)(struct json_pull *, char *buffer, size_t n), void *source);

// json_end is now a thin convenience that resets the caller's json_pull_ptr.
// The parser (and any tree it still owns) is freed when the last shared_ptr
// to it is dropped, so calling json_end is optional if the json_pull_ptr will
// go out of scope on its own.
void json_end(json_pull_ptr &p);

typedef void (*json_separator_callback)(json_type type, json_pull *j, void *state);

json_object_ptr json_read_tree(json_pull_ptr j);
json_object_ptr json_read(json_pull_ptr j);
json_object_ptr json_read_separators(json_pull_ptr j, json_separator_callback cb, void *state);

// json_free now just resets the caller's json_object_ptr. The subtree is
// destroyed when the last shared_ptr to it is dropped (typically by also
// being removed from its parent or parser).
void json_free(json_object_ptr &j);

// Splice o out of its parent's array/object and clear parent/parser back-pointers
// throughout the detached subtree so it can outlive the original parser.
void json_disconnect(json_object_ptr j);

json_object_ptr json_hash_get(json_object_ptr o, const char *s);
json_object_ptr json_hash_get(json_object *o, const char *s);

std::string json_stringify(json_object_ptr o);

#endif