/***************************************************************************
* Copyright (C) 2010
* by Dimok
*
* This software is provided 'as-is', without any express or implied
* warranty. In no event will the authors be held liable for any
* damages arising from the use of this software.
*
* Permission is granted to anyone to use this software for any
* purpose, including commercial applications, and to alter it and
* redistribute it freely, subject to the following restrictions:
*
* 1. The origin of this software must not be misrepresented; you
* must not claim that you wrote the original software. If you use
* this software in a product, an acknowledgment in the product
* documentation would be appreciated but is not required.
*
* 2. Altered source versions must be plainly marked as such, and
* must not be misrepresented as being the original software.
*
* 3. This notice may not be removed or altered from any source
* distribution.
*
* HTML_Stream Class
*
* for WiiXplorer 2010
***************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "HTML_Stream.h"
#include "networkops.h"
#include "http.h"

#define htmlstringcompare(text, cmp, pos) strncasecmp((const char*) &text[pos], (const char*) cmp, strlen((const char*) cmp))


HTML_Stream::HTML_Stream()
{
    HTML_File = NULL;
    position = 0;
    filesize = 0;
}

HTML_Stream::HTML_Stream( const char * url )
{
    HTML_File = NULL;
    position = 0;
    filesize = 0;

    LoadLink( url );
}

HTML_Stream::~HTML_Stream()
{
    if ( HTML_File )
        free( HTML_File );
}

bool HTML_Stream::LoadLink( const char * url )
{
    if ( !IsNetworkInit() )
        return false;

    struct block file = downloadfile( url );

    if ( !file.data || !file.size )
        return false;

    if ( HTML_File )
        free( HTML_File );

    HTML_File = ( char * ) file.data;
    filesize = file.size;
    position = 0;

    return true;
}

const char * HTML_Stream::FindStringStart( const char * string )
{
    if ( !HTML_File )
        return NULL;

    while ( ( u32 ) position < filesize )
    {
        if ( htmlstringcompare( HTML_File, string, position ) == 0 )
            break;

        position++;
    }

    return &HTML_File[position];
}

const char * HTML_Stream::FindStringEnd( const char * string )
{
    if ( !HTML_File )
        return NULL;

    while ( ( u32 ) position < filesize )
    {
        if ( htmlstringcompare( HTML_File, string, position ) == 0 )
            break;

        position++;
    }

    if ( ( u32 ) position >= filesize )
    {
        return NULL;
    }

    position += strlen( string );

    return &HTML_File[position];
}

char * HTML_Stream::CopyString( const char * stopat )
{
    if ( !stopat || !HTML_File )
        return NULL;

    u32 blocksize = 1024;
    u32 counter = 0;
    u32 allocatedsize = 0;

    char * outtext = ( char* ) malloc( blocksize );
    if ( !outtext )
        return NULL;

    allocatedsize = blocksize;
    memset( outtext, 0, blocksize );

    while ( ( htmlstringcompare( HTML_File, stopat, position ) != 0 ) && ( position + strlen( stopat ) < filesize ) )
    {
        if ( counter > blocksize )
        {
            blocksize += 1024;
            char * tmpblock = ( char* ) realloc( outtext, blocksize );
            if ( !tmpblock )
            {
                free( outtext );
                outtext = NULL;
                free( tmpblock );
                return NULL;
            }

            outtext = tmpblock;
        }

        outtext[counter] = HTML_File[position];
        position++;
        counter++;
    }

    outtext[counter] = '\0';
    outtext = ( char* ) realloc( outtext, counter + 1 );

    return outtext;
}

int HTML_Stream::Seek( u32 pos, int origin )
{
    if ( !HTML_File )
        return -1;

    switch ( origin )
    {
        case SEEK_SET:
            position = pos;
            break;
        case SEEK_CUR:
            position += pos;
            break;
        case SEEK_END:
            position = filesize + pos;
            break;
    }

    return 0;
}

void HTML_Stream::Rewind()
{
    if ( !HTML_File )
        return;

    position = 0;
}

int HTML_Stream::GetPosition()
{
    return position;
}