/*

 * Copyright (C) 2016  J. Ali Harlow <ali@juiblex.co.uk>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License along

 * with this program; if not, write to the Free Software Foundation, Inc.,

 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

#include <stdlib.h>

#include <string.h>

#include <errno.h>

#include <razor.h>

#include <glib.h>

#include <gio/gio.h>

#include "config.h"

#include "plover/plover.h"

#include "uri-handler.h"

#include "ascii-ctype.h"

static gboolean has_valid_scheme(const char *uri)

{

    /*

     * RFC 3986 defines valid schemes as:

     * scheme = ALPHA *( ALPHA / DIGIT / "+" / "-" / "." )

     */

    const char *s;

    if (!ascii_isalpha(*uri))

	return FALSE;

    for(s=uri+1;;s++)

	if (*s==':')

	    return TRUE;

	else if (!ascii_isalnum(*s) && *s!='+' && *s!='-' && *s!='.')

	    return FALSE;

}

static GFile *file_for_uri(const char *uri)

{

    GFile *file;

    if (!has_valid_scheme(uri))

    {

	g_warning("%s: Not a valid URI",uri);

	file=g_file_new_for_path(uri);

    }

    else

    {

	if (strstr(uri+1,"file:/"))

	    g_warning("%s: Implausible URI",uri);

	file=g_file_new_for_uri(uri);

    }

    return file;

}

int uri_mkdir(const char *uri,mode_t mode,struct razor_error **error)

{

    GFile *file;

    GFileInfo *info;

    int retval;

    GError *err=NULL;

    file=file_for_uri(uri);

    retval=!g_file_make_directory(file,NULL,&err);

    if (retval)

    {

	if (g_error_matches(err,G_IO_ERROR,G_IO_ERROR_EXISTS))

	{

	    info=g_file_query_info(file,G_FILE_ATTRIBUTE_STANDARD_TYPE,

	      G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS,NULL,NULL);

	    if (info)

	    {

		g_clear_error(&err);

		if (g_file_info_get_file_type(info)==G_FILE_TYPE_DIRECTORY)

		    retval=0;

		else

		    razor_set_error(error,RAZOR_POSIX_ERROR,EEXIST,uri,

		      "Not a directory");

		g_object_unref(info);

	    }

	}

	if (err)

	    plover_propagate_g_error(error,err);

    }

    g_object_unref(file);

    return retval;

}

int uri_unlink(const char *uri,struct razor_error **error)

{

    GFile *file;

    int retval;

    GError *err=NULL;

    file=file_for_uri(uri);

    retval=!g_file_delete(file,NULL,&err);

    if (retval)

	plover_propagate_g_error(error,err);

    g_object_unref(file);

    return retval;

}

int uri_move(const char *src_uri,const char *dst_uri,

  struct razor_error **error)

{

    GFile *src,*dst;

    int retval;

    GError *err=NULL;

    src=file_for_uri(src_uri);

    dst=file_for_uri(dst_uri);

    retval=!g_file_move(src,dst,G_FILE_COPY_OVERWRITE,NULL,NULL,NULL,&err);

    if (retval)

	plover_propagate_g_error(error,err);

    g_object_unref(src);

    g_object_unref(dst);

    return retval;

}

static void *uri_get_contents(const char *uri,size_t *length,int private,

  struct razor_error **error)

{

    GFile *file;

    void *addr;

    char *contents;

    gsize len;

    GError *err=NULL;

    file=file_for_uri(uri);

    if (!g_file_load_contents(file,NULL,&contents,&len,NULL,&err))

    {

	plover_propagate_g_error(error,err);

	addr=NULL;

    }

    else

    {

	addr=contents;

	if (length)

	    *length=len;

    }

    g_object_unref(file);

    return addr;

}

static int uri_free_contents(void *addr,size_t length)

{

    g_free(addr);

    return 0;

}

int uri_is_directory(const char *uri,struct razor_error **error)

{

    GFile *file;

    GFileInfo *info;

    int retval;

    GError *err=NULL;

    file=file_for_uri(uri);

    info=g_file_query_info(file,G_FILE_ATTRIBUTE_STANDARD_TYPE,

      G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS,NULL,&err);

    if (info)

    {

	if (g_file_info_get_file_type(info)==G_FILE_TYPE_DIRECTORY)

	    retval=1;

	else

	    retval=0;

	g_object_unref(info);

    }

    else

    {

	retval=-1;

	plover_propagate_g_error(error,err);

    }

    g_object_unref(file);

    return retval;

}

void *uri_opendir(const char *uri,struct razor_error **error)

{

    GFile *file;

    GFileEnumerator *dir;

    GError *err=NULL;

    file=file_for_uri(uri);

    dir=g_file_enumerate_children(file,G_FILE_ATTRIBUTE_STANDARD_NAME,

      G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS,NULL,&err);

    if (!dir)

	plover_propagate_g_error(error,err);

    g_object_unref(file);

    return dir;

}

char *uri_readdir(void *dp,struct razor_error **error)

{

    GFileEnumerator *dir=dp;

    char *name;

    GError *err=NULL;

    GFileInfo *info;

    info=g_file_enumerator_next_file(dir,NULL,&err);

    if (!info)

    {

	name=NULL;

	plover_propagate_g_error(error,err);

    }

    else

    {

	name=strdup(g_file_info_get_name(info));

	g_object_unref(info);

    }

    return name;

}

int uri_closedir(void *dp,struct razor_error **error)

{

    GFileEnumerator *dir=dp;

    int retval;

    GError *err=NULL;

    retval=!g_file_enumerator_close(dir,NULL,&err);

    if (retval)

	plover_propagate_g_error(error,err);

    g_object_unref(dir);

    return retval;

}

void plover__uri_handler_init(void)

{

    static gsize init=0;

    struct razor_uri_vtable uri_vtable={0,};

    if (g_once_init_enter(&init))

    {

	uri_vtable.structure_size=sizeof(uri_vtable);

	uri_vtable.mkdir=uri_mkdir;

	uri_vtable.unlink=uri_unlink;

	uri_vtable.move=uri_move;

	uri_vtable.get_contents=uri_get_contents;

	uri_vtable.free_contents=uri_free_contents;

	uri_vtable.is_directory=uri_is_directory;

	uri_vtable.opendir=uri_opendir;

	uri_vtable.readdir=uri_readdir;

	uri_vtable.closedir=uri_closedir;

	razor_uri_set_vtable(NULL,&uri_vtable,NULL);

	g_once_init_leave(&init,1);

    }

}