#include "ast.h"
#include <stdio.h>

void print_spaces( int n ) {
    printf("\n");
    for(int i = 0; i < n; ++i) {
        printf(" ");
    }
}

static void print_native_type( enum native_types type ) {
    switch( type ) {
        case INT:
            printf("INT");
            break;
        case CHAR:
            printf("CHAR");
            break;
        case FLOAT:
            printf("FLOAT");
            break;
        case BOOL:
            printf("BOOL");
            break;
        default:
            fprintf(stderr, "uknown type\n");
            exit(-1);
    }
}

static void print_type( union type* type) {
    if( type != NULL ) {
        if( type->tag == SINGLE ) {
            print_native_type( type->single.type );
        }
        else {
            printf("[");
            print_type( type->seq.next );
            printf("]");
        }
    }
}

void print_var( int n_spaces, struct var *v ) {
    if( v != NULL ) {
        print_spaces( n_spaces );
        print_type( v->type );
        printf(" %s\n", v->name );
        print_var( n_spaces, v->next );
    }
}

void print_params( struct param* param ) {
    if( param == NULL ) return;
    printf("%s ", param->name );
    print_type( param->type );
    if( param->next != NULL ) {
        printf(", ");
        print_params( param->next );
    }
}

void print_func( int n_spaces, struct func* f ) {
    if( f != NULL ) {
        print_spaces( n_spaces );
        printf("FUNC< ");
        print_type( f->type );
        printf(" > ");
        printf(" %s, ", f->name );
        printf("PARAMS = {");
        print_params( f->param );
        printf("}\n");

        print_func( n_spaces, f->next );
    }
}

void print_tree() {
    print_var( 4, GLOBAL_TREE->vars );
    print_func( 4, GLOBAL_TREE->funcs );
}