/*

 * lftp - file transfer program

 *

 * Copyright (c) 1996-2013 by Alexander V. Lukyanov (lav@yars.free.net)

 *

 * 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 3 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, see <http://www.gnu.org/licenses/>.

 */

#include <config.h>

#include "bookmark.h"

#include <sys/stat.h>

#include <fcntl.h>

#include <unistd.h>

#include <stddef.h>

#include "trio.h"

#include "ResMgr.h"

#include "misc.h"

#include "url.h"

#define super KeyValueDB

Bookmark lftp_bookmarks;

ResDecl res_auto_sync("bmk:auto-sync","yes",ResMgr::BoolValidate,ResMgr::NoClosure);

Bookmark::Bookmark()

{

   const char *home=get_lftp_data_dir();

   if(home)

      bm_file.vset(home,"/bookmarks",NULL);

   bm_fd=-1;

   stamp=(time_t)-1;

}

Bookmark::~Bookmark()

{

   Close();

}

void Bookmark::Refresh()

{

   if(!bm_file)

      return;

   struct stat st;

   if((bm_fd==-1 ? stat(bm_file,&st) : fstat(bm_fd,&st)) == -1)

      return;

   if(st.st_mtime==stamp)

      return;

   Load();

}

void Bookmark::Load()

{

   super::Empty();

   if(!bm_file)

      return;

   if(bm_fd==-1)

   {

      bm_fd=open(bm_file,O_RDONLY);

      if(bm_fd==-1)

	 return;

      fcntl(bm_fd,F_SETFD,FD_CLOEXEC);

      if(Lock(F_RDLCK)==-1)

	 fprintf(stderr,"%s: lock for reading failed, trying to read anyway\n",bm_file.get());

   }

   struct stat st;

   fstat(bm_fd,&st);

   stamp=st.st_mtime;

   lseek(bm_fd,0,SEEK_SET);

   super::Read(dup(bm_fd));   // Read closes file

}

static bool auto_sync;

void Bookmark::PreModify()

{

   if(!bm_file)

      return;

   auto_sync=ResMgr::QueryBool("bmk:auto-sync",0);

   if(!auto_sync)

      return;

   Close();

   bm_fd=open(bm_file,O_RDWR|O_CREAT,0600);

   if(bm_fd==-1)

      return;

   if(Lock(F_WRLCK)==-1)

   {

      fprintf(stderr,"%s: lock for writing failed\n",bm_file.get());

      Close();

      return;

   }

   Refresh();

}

void Bookmark::PostModify()

{

   if(!bm_file)

      return;

   if(!auto_sync)

      return;

   // the file is already locked in PreModify.

   lseek(bm_fd,0,SEEK_SET);

#ifdef HAVE_FTRUNCATE

   if(ftruncate(bm_fd,0)==-1) // note the following statement

#endif

   close(open(bm_file,O_WRONLY|O_TRUNC));

   super::Write(bm_fd);

   bm_fd=-1;   // Write closes file

}

void Bookmark::UserSave()

{

   if(!bm_file)

      return;

   Close();

   bm_fd=open(bm_file,O_RDWR|O_CREAT|O_TRUNC,0600);

   if(bm_fd==-1)

      return;

   if(Lock(F_WRLCK)==-1)

   {

      fprintf(stderr,"%s: lock for writing failed\n",bm_file.get());

      Close();

      return;

   }

   super::Write(bm_fd);

   bm_fd=-1;   // Write closes file

}

void Bookmark::Add(const char *key,const char *value)

{

   PreModify();

   super::Add(key,value);

   PostModify();

}

void Bookmark::Remove(const char *key)

{

   PreModify();

   super::Remove(key);

   PostModify();

}

void Bookmark::Close()

{

   if(bm_fd!=-1)

   {

      close(bm_fd);

      bm_fd=-1;

   }

}

void Bookmark::AutoSync()

{

   if(ResMgr::QueryBool("bmk:auto-sync",0))

   {

      Refresh();

      Close();

   }

}

const char *Bookmark::Lookup(const char *key)

{

   AutoSync();

   return super::Lookup(key);

}

#if 0

void Bookmark::List()

{

   Refresh();

   Close();

   super::Write(dup(0));

}

#endif

char *Bookmark::Format()

{

   AutoSync();

   return super::Format();

}

char *Bookmark::FormatHidePasswords()

{

   AutoSync();

   return super::Format(url::hide_password);

}