/*
 * sparse/token_store.c
 *
 * Copyright (C) 2012 Oracle.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "lib.h"
#include "parse.h"
#include "allocate.h"

struct line {
	struct position pos;
	struct line *prev;
	struct token *token;
	struct line *next;
};

__ALLOCATOR(struct token, "token store", perm_token);
ALLOCATOR(line, "line of tokens");

static struct token *copy_token(struct token *token)
{
	struct token *new;

	new = __alloc_perm_token(0);
	memcpy(new, token, sizeof(*token));
	new->next = NULL;
	return new;
}

static struct line *cursor;

static void find_line(struct position pos)
{
	if (!cursor)
		return;
	if (pos.line == cursor->pos.line)
		return;
	if (pos.line < cursor->pos.line) {
		if (!cursor->prev)
			return;
		cursor = cursor->prev;
		find_line(pos);
		return;
	}
	if (!cursor->next)
		return;
	if (pos.line < cursor->next->pos.line)
		return;
	cursor = cursor->next;
	find_line(pos);
}

static void insert_into_line(struct token **current, struct token *new)
{
	if (!*current) {
		*current = new;
		return;
	}

	if (new->pos.pos < (*current)->pos.pos) {
		new->next = *current;
		*current = new;
		return;
	}

	if (new->pos.pos == (*current)->pos.pos)
		return;

	insert_into_line(&(*current)->next, new);
}

static void store_token(struct token *token)
{
	token = copy_token(token);

	find_line(token->pos);

	if (!cursor) {
		cursor = __alloc_line(0);
		cursor->pos = token->pos;
		cursor->token = token;
		return;
	}

	if (token->pos.line < cursor->pos.line) {
		cursor->prev = __alloc_line(0);
		cursor->prev->next = cursor;
		cursor = cursor->prev;
		cursor->pos = token->pos;
		cursor->token = token;
		return;
	}

	if (token->pos.line == cursor->pos.line) {
		insert_into_line(&cursor->token, token);
		return;
	}

	cursor->next = __alloc_line(0);
	cursor->next->prev = cursor;
	cursor = cursor->next;
	cursor->pos = token->pos;
	cursor->token = token;
}

void store_all_tokens(struct token *token)
{
	while (token_type(token) != TOKEN_STREAMEND) {
		store_token(token);
		token = token->next;
	}
}

struct token *first_token_from_line(struct position pos)
{
	find_line(pos);

	if (!cursor)
		return NULL;

	if (cursor->pos.stream != pos.stream)
		return NULL;
	if (cursor->pos.line != pos.line)
		return NULL;

	return cursor->token;
}

struct token *pos_get_token(struct position pos)
{
	struct token *token;

	token = first_token_from_line(pos);
	while (token) {
		if (pos.pos == token->pos.pos)
			return token;
		if (pos.pos < token->pos.pos)
			return NULL;
		token = token->next;
	}
	return NULL;
}

char *pos_ident(struct position pos)
{
	struct token *token;

	token = pos_get_token(pos);
	if (!token)
		return NULL;
	if (token_type(token) != TOKEN_IDENT)
		return NULL;
	return token->ident->name;
}