/* radare - LGPL - Copyright 2007-2020 - pancake, ret2libc */

#include <r_core.h>
#include <r_util.h>

R_API void r_graph_free_node_info(void *ptr) {
	if (ptr) {
		RGraphNodeInfo *info = ptr;
		free (info->body);
		free (info->title);
		free (info);
	}
}

R_API RGraphNodeInfo * R_NONNULL r_graph_create_node_info(const char *title, const char * R_NULLABLE body, ut64 offset) {
	RGraphNodeInfo *data = R_NEW0 (RGraphNodeInfo);
	data->title = strdup (title? title: "");
	data->body = body? strdup (body): NULL;
	data->offset = offset;
	return data;
}

R_API RGraphNode *r_graph_add_node_info(RGraph *graph, const char *title, const char *body, ut64 offset) {
	R_RETURN_VAL_IF_FAIL (graph, NULL);
	RGraphNodeInfo *data = r_graph_create_node_info (title, body, offset);
	RGraphNode *node = r_graph_add_nodef (graph, data, r_graph_free_node_info);
	if (!node) {
		r_graph_free_node_info (data);
	}
	return node;
}

R_API char *r_graph_drawable_to_dot(RGraph /*RGraphNodeInfo*/ *graph, const char *node_properties, const char *edge_properties) {
	RList *nodes = graph->nodes;
	RListIter *it;
	RGraphNode *node = NULL, *target = NULL;
	RStrBuf buf;
	r_strbuf_init (&buf);
	r_strbuf_appendf (&buf,
		"digraph code {\nrankdir=LR;\noutputorder=edgesfirst\ngraph [bgcolor=azure];\n"
		"edge [arrowhead=normal, color=\"#3030c0\" style=bold weight=2 %s];\n"
		"node [fillcolor=white, style=filled shape=box "
		"fontsize=\"8\" %s];\n",
		r_str_get (edge_properties),
		r_str_get (node_properties));

	r_list_foreach (nodes, it, node) {
		RGraphNodeInfo *print_node = (RGraphNodeInfo *)node->data;
		const char *body = print_node->body;
		if (!body || !*body) {
			r_strbuf_appendf (&buf, "%d [URL=\"%s\", color=\"lightgray\", label=\"%s\"]\n",
				node->idx, print_node->title, print_node->title);
		} else {
			r_strbuf_appendf (&buf, "%d [URL=\"%s\", color=\"lightgray\", label=\"%s\\n%s\"]\n",
				node->idx, print_node->title, print_node->title, body);
		}
		RGraphNode **itt;
		R_VEC_FOREACH (&node->out_nodes, itt) {
			target = *itt;
			r_strbuf_appendf (&buf, "%d -> %d\n", node->idx, target->idx);
		}
	}
	r_strbuf_append (&buf, "}\n");
	return r_strbuf_drain_nofree (&buf);
}

R_API void r_graph_drawable_to_json(RGraph /*RGraphNodeInfo*/ *graph, PJ *pj, bool use_offset) {
	RList *nodes = graph->nodes;
	RListIter *it;
	RGraphNode *node = NULL;
	if (!pj) {
		return;
	}
	pj_o (pj);
	pj_k (pj, "nodes");
	pj_a (pj);

	r_list_foreach (nodes, it, node) {
		RGraphNodeInfo *print_node = (RGraphNodeInfo *)node->data;
		pj_o (pj);
		pj_ki (pj, "id", node->idx);
		if (print_node->title) {
			pj_ks (pj, "title", print_node->title);
		}
		if (print_node->body) {
			pj_ks (pj, "body", print_node->body);
		}
		if (use_offset) {
			pj_kn (pj, "offset", print_node->offset);
		}
		pj_k (pj, "out_nodes");
		pj_a (pj);
		RGraphNode **itt;
		R_VEC_FOREACH (&node->out_nodes, itt) {
			pj_i (pj, (*itt)->idx);
		}
		pj_end (pj);
		pj_end (pj);
	}
	pj_end (pj);
	pj_end (pj);
}