/*

  * 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);

     }

 }