minilauncher-for-slippi/replay.c

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <time.h>
#include <sys/types.h>
#include <assert.h>
#include "replay.h"

Evas_Object* tab_replays = NULL;
char* replays_dir = "replays/";
Ecore_Exe* dolphin_replay_exe;
extern char* dolphin_replay_file, *game_path;
#ifndef DATA_DIR
#define DATA_DIR "data/"
#endif

struct replay
{
	char* filename;
	char* p1;
	char* p1code;
	char* p2;
	char* p2code;
	int game_state;
}* replays = NULL;
size_t replays_len = 0;

void
_launch_slippi_cb(void *data,
                  Evas_Object *obj EINA_UNUSED,
                  void *event_info EINA_UNUSED);
                  
Eina_Bool
_launch_restore_btn(void* data);

unsigned
dec_uint_be(unsigned char* arr, size_t len)
{
        unsigned res = 0;
        for (int i = len-1, j = 0; i >= 0; (--i, ++j))
        {
                res |= arr[i] << (j*8);
        }
        return res;
}

long
fsize(FILE* FP)
{
        long _ = ftell(FP);
        fseek(FP, 0, SEEK_END);
        long new_size = ftell(FP);
        fseek(FP, _, SEEK_SET);
        return new_size;
}

char*
sstrstr(char* haystack, char* needle, size_t length)
{
	size_t needle_length = strlen(needle);
	size_t i;
	for (i = 0; i < length; i++) {
		if (i + needle_length > length) {
			return NULL;
		}
		if (strncmp(&haystack[i], needle, needle_length) == 0) {
			return &haystack[i];
		}
	}
	return NULL;
}

char*
ubjson_search(unsigned char* buf, long length, char const* key, long* offset)
{
	size_t key_len = strlen(key);
	char* start = sstrstr(buf, key, length);
	if (!start) return NULL;
	start += key_len;
	// Just 8 bits for now...
	uint8_t value_len = start[0];
	start++;
	
	char* value = calloc(1, value_len + 1);
	memcpy(value, start, value_len);
	start += value_len;
	
	if (offset)
		*offset = (long)start - (long)buf;
	
	return value;
}

static int
parse_replay(struct replay* rpy)
{
	unsigned char buf[4096] = {0};
	char* filename;
	asprintf(&filename, "%s%s", replays_dir, rpy->filename);
	FILE* SLP = fopen(filename, "r");
	if (!SLP)
	{	
		perror("fopen");
		return 1;
	}
	// We used stat at some point, but a filename comparison is easier to write
	//assert(stat(filename, &(rpy->attr)) == 0);
	
	long data_skip;

	long filesize = fsize(SLP);
	
	// Skip begin header and raw data
	fseek(SLP, 11, SEEK_CUR);
	fread(buf, 1, 4, SLP);
	fseek(SLP, (data_skip = dec_uint_be(buf, 4)), SEEK_CUR);
	fseek(SLP, -6, SEEK_CUR);
	fread(buf, 1, 1, SLP);
	fseek(SLP, 5, SEEK_CUR);
	rpy->game_state = buf[0];
	
	// debug
	long length = filesize - ftell(SLP);
	// Sometimes, your computer will crash (this happened to me :^) and the file
	//  will of course, just be corrupted or F-ed up
	if (length > 4096)
	{
		goto abort;
	}
		
	fread(buf, 1, length, SLP);	
	long off;
	char* bufx = buf;
	char* p1name = ubjson_search(bufx, length, "netplaySU", &off);
	length -= off; bufx += off;
	char* p1code = ubjson_search(bufx, length, "codeSU", &off);
	length -= off; bufx += off;
	
	char* p2name = ubjson_search(bufx, length, "netplaySU", &off);
	length -= off; bufx += off;
	char* p2code = ubjson_search(bufx, length, "codeSU", &off);
	length -= off; bufx += off;
	
	rpy->p1 = p1name;
	rpy->p1code = p1code;
	rpy->p2 = p2name;
	rpy->p2code = p2code;
	
	//printf("Reading %s\n", filename);
	fclose(SLP);
	return 0;
abort:
	rpy->p1 = NULL;
	rpy->p1code = NULL;
	rpy->p2 = NULL;
	rpy->p2code = NULL;
	fclose(SLP);
	return 3;
}

int
order_filename_greatest(void const* _rpy1, void const* _rpy2)
{
	struct replay* rpy1 = _rpy1, * rpy2 = _rpy2;
	return -strcmp(rpy1->filename, rpy2->filename);
}

static int
recurse_replay_files()
{
	DIR *dp;
	struct dirent *ep;
	dp = opendir(replays_dir);
	if (dp != NULL)
	{
    	while ((ep = readdir(dp)) != NULL)
    	{
    		if (ep->d_name && ep->d_name[0] != '.')
    		{
    			replays = realloc(replays, sizeof(struct replay) * (replays_len + 2));
    			struct replay* rpy = replays + replays_len++;
    			rpy->filename = strdup(ep->d_name);

    			// We used to parse replays as is, but now we sort by date.
    			parse_replay(rpy);
			}
		}
		
		// Lol. Lmao, even.
		qsort(replays, replays_len, sizeof(struct replay),
		      order_filename_greatest);
		      
		closedir(dp);
		return 0;
	}
}

char const*
gameend2str(int code)
{
	switch (code)
	{
	case 0: return "Unresolved";
	case 1: return "TIME!";
	case 2: return "GAME!";
	case 3: return "Resolved.";
	case 7: return "No Contest.";
	}
}

Evas_Object*
replays_content(void* data, Evas_Object* obj, const char* part)
{
	char* c;
	if (strcmp(part, "elm.swallow.icon"))
	{
		Evas_Object* box = elm_box_add(obj);
		{
			elm_box_horizontal_set(box, EINA_TRUE);
			asprintf(&c, DATA_DIR "assets/stock/15_0.png");

			Evas_Object* ic = elm_icon_add(box);
			{
				elm_image_file_set(ic, c, NULL);
				elm_image_resizable_set(ic, EINA_FALSE, EINA_FALSE);
				evas_object_show(ic);
				elm_box_pack_end(box, ic);
			}
			Evas_Object* lbl = elm_label_add(box);
			{
				elm_object_text_set(lbl, " vs.  ");
				elm_object_scale_set(lbl, 1.2);
				evas_object_show(lbl);
				elm_box_pack_end(box, lbl);
			}
	        ic = elm_icon_add(box);
	        {
	        	elm_image_file_set(ic, c, NULL);
	        	elm_image_resizable_set(ic, EINA_FALSE, EINA_FALSE);
	        	evas_object_show(ic);
	        	elm_box_pack_end(box, ic);
			}
	        evas_object_show(box);
        }
        return box;
	}
	return NULL;
}

static char*
replays_strings(void* data, Evas_Object* obj, const char* part)
{
	long idx = data;
	char* c;
	// Check this is text for the part we're expecting
	if (strcmp(part, "elm.text") == 0)
	    /* DQ after join */
	    
	{
		if (replays[idx].p1 != NULL && replays[idx].p2 != NULL)
		{
			asprintf(&c, "%s (%s) Vs. %s (%s)",
		    	     replays[idx].p1, replays[idx].p1code,
		    	     replays[idx].p2, replays[idx].p2code);
			return c;
		}
		else if (replays[idx].p1 == NULL || replays[idx].p2 == NULL)
			printf("[Replay Warning] %s is a incomplete replay. Good idea to delete it.\n",
		       replays[idx].filename);
	}
	else if (strcmp(part, "elm.text.sub") == 0)
	{
		asprintf(&c, "%s  *  (3-0)", gameend2str(replays[idx].game_state));
		return c;
	}
	else
		return NULL;
}


static void
_launch_replay_cb(void *data, Evas_Object *obj EINA_UNUSED, void *event_info EINA_UNUSED)
{
	//if (data)
	//	elm_object_disabled_set(data, EINA_TRUE);
	
	//ecore_timer_add(3, _launch_restore_btn, data);
	
	Eina_Strbuf* exe = eina_strbuf_new();
	eina_strbuf_append_printf(exe, "%s -e %s -i play.json -b", dolphin_replay_file, game_path);
	dolphin_replay_exe = ecore_exe_run(eina_strbuf_release(exe), data);
}

static void
_item_select_cb(void *data, Evas_Object *obj, void *event_info)
{
	long idx = data;
	char* jason;
	asprintf(&jason, "{\"replay\":\"%s%s\"}\n", replays_dir, replays[idx].filename);
	
	// Write to file
	FILE* outfile = fopen("play.json", "w");
	fwrite(jason, 1, strlen(jason), outfile);
	fclose(outfile);
	
	_launch_replay_cb(obj, obj, NULL);
	
	free(jason);
}

static void
_dd_exp(void *data, Evas_Object *obj EINA_UNUSED, void *event_info)
{
   	long idx = (long)data;
   	Elm_Object_Item *glit = event_info;
   	Evas_Object* gl = elm_object_item_widget_get(glit);
   
   	Elm_Genlist_Item_Class* _itc_sub = elm_genlist_item_class_new();
	_itc_sub->item_style = "default";
	_itc_sub->func.text_get = NULL;
	_itc_sub->func.content_get = NULL;
	_itc_sub->func.state_get = NULL;
	_itc_sub->func.del = NULL;

	for (int i = 0; i < 2; ++i)
	{
   	elm_genlist_item_append(gl, _itc_sub,
                           (void *)(long) (idx)/* item data */,
                           glit/* parent */,
                           ELM_GENLIST_ITEM_NONE, NULL/* func */,
                           NULL/* func data */);
	}
}


static void
_dd_con(void *data EINA_UNUSED, Evas_Object *obj EINA_UNUSED, void *event_info)
{
   Elm_Object_Item *glit = event_info;
   elm_genlist_item_subitems_clear(glit);
}

static void
_dd_exp_req(void *data EINA_UNUSED, Evas_Object *obj EINA_UNUSED, void *event_info)
{
   Elm_Object_Item *glit = event_info;
   elm_genlist_item_expanded_set(glit, EINA_TRUE);
}

static void
_dd_con_req(void *data EINA_UNUSED, Evas_Object *obj EINA_UNUSED, void *event_info)
{
   Elm_Object_Item *glit = event_info;
   elm_genlist_item_expanded_set(glit, EINA_FALSE);
}

Evas_Object*
tab_replays_setup(Evas_Object* parent)
{
	tab_replays = elm_genlist_add(parent);
	evas_object_smart_callback_add(tab_replays, "expand,request", _dd_exp_req, tab_replays);
  	evas_object_smart_callback_add(tab_replays, "contract,request", _dd_con_req, tab_replays);
  	evas_object_smart_callback_add(tab_replays, "expanded", _dd_exp, tab_replays);
  	evas_object_smart_callback_add(tab_replays, "contracted", _dd_con, tab_replays);

  	   elm_genlist_tree_effect_enabled_set(tab_replays, EINA_TRUE);

	elm_genlist_homogeneous_set(tab_replays, EINA_TRUE);
	evas_object_resize(tab_replays, 40, 40);
	Elm_Genlist_Item_Class *_itc = elm_genlist_item_class_new();
	_itc->item_style = "double_label";
	_itc->func.text_get = replays_strings;
	_itc->func.content_get = replays_content;
	_itc->func.state_get = NULL;
	_itc->func.del = NULL;
	
   	recurse_replay_files();
   	
	for (int i = 0; i < replays_len; ++i)
	{
		Evas_Object* root = elm_genlist_item_append(tab_replays,
			_itc,
			i,                    // Item data
			NULL,                    // Parent item for trees, NULL if none
			ELM_GENLIST_ITEM_TREE,   // Item type; this is the common one
			_item_select_cb,                    // Callback on selection of the item
			(long)i                     // Data for that callback function
		);
	}
	
   	evas_object_size_hint_align_set(tab_replays, EVAS_HINT_FILL, EVAS_HINT_FILL);
   	return tab_replays;
}