set the mtime of the local file to be the same as the remote

pull/40/head
Matchman Green 2012-04-28 01:52:02 +08:00
parent f0ca94d7e0
commit ceca83dc72
10 changed files with 124 additions and 17 deletions

View File

@ -16,7 +16,7 @@ add_executable( grive
src/protocol/HTTP.cc
src/protocol/Json.cc
src/protocol/OAuth2.cc
src/util/DateTime.hh
src/util/DateTime.cc
src/util/OS.cc
)

View File

@ -22,6 +22,8 @@
#include "protocol/HTTP.hh"
#include "protocol/Json.hh"
#include "protocol/OAuth2.hh"
#include "util/DateTime.hh"
#include "util/OS.hh"
// dependent libraries
#include <openssl/evp.h>
@ -180,10 +182,12 @@ void Drive::UpdateFile( const Json& entry )
if ( entry.Has( "docs$filename" ) )
{
// use title as the filename
std::string filename = entry["docs$filename"]["$t"].As<std::string>() ;
std::string url = entry["content"]["src"].As<std::string>() ;
std::string filename = entry["docs$filename"]["$t"].As() ;
std::string url = entry["content"]["src"].As() ;
std::string parent_href = Parent( entry ) ;
DateTime remote( entry["updated"]["$t"].As<std::string>() ) ;
bool changed = true ;
std::string path = "./" + filename ;
@ -194,11 +198,16 @@ void Drive::UpdateFile( const Json& entry )
if ( pit != m_coll.end() )
path = pit->Path() + "/" + filename ;
}
DateTime local = os::FileMTime( path ) ;
std::cout << "file time: " << entry["updated"]["$t"].As<std::string>() << " " << remote << " " << local << std::endl ;
// compare checksum first if file exists
std::ifstream ifile( path.c_str(), std::ios::binary | std::ios::out ) ;
if ( ifile && entry.Has("docs$md5Checksum") )
{
os::SetFileTime( path, remote ) ;
std::string remote_md5 = entry["docs$md5Checksum"]["$t"].As<std::string>() ;
std::string local_md5 = MD5( ifile.rdbuf() ) ;

View File

@ -138,6 +138,11 @@ void Json::Add( const std::string& key, const Json& json )
::json_object_object_add( m_json, key.c_str(), json.m_json ) ;
}
Json::Proxy Json::As() const
{
return Proxy( *this ) ;
}
template <>
std::string Json::As<std::string>() const
{
@ -243,5 +248,4 @@ bool Json::FindInArray( const std::string& key, const std::string& value, Json&
}
}
}

View File

@ -52,6 +52,14 @@ public :
template <typename T>
T As() const ;
struct Proxy
{
const Json& referring ;
explicit Proxy( const Json& j ) : referring( j ) { }
template <typename T> operator T() const { return referring.As<T>() ; }
} ;
Proxy As() const ;
template <typename T>
bool Is() const ;

View File

@ -20,7 +20,9 @@
#include "DateTime.hh"
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <iomanip>
#include <time.h>
@ -28,20 +30,32 @@ namespace gr {
DateTime::DateTime( ) :
m_sec ( 0 ),
m_usec ( 0 )
m_nsec ( 0 )
{
}
DateTime::DateTime( const std::string& iso ) :
m_sec ( 0 ),
m_usec ( 0 )
m_nsec ( 0 )
{
struct tm tp = {} ;
const char *r = ::strptime( iso.c_str(), "%Y-%m-%dT%H:%M:%S", &tp ) ;
// should be '.' followed by 3 digits and 'Z' (e.g. .123Z)
if ( r != 0 && r - iso.c_str() == 19 )
{
m_sec = ::timegm( &tp ) ;
// at least 3 digits is OK. we don't care the Z
if ( *r == '.' && ::strlen( r+1 ) >= 3 )
m_nsec = std::atoi( r+1 ) * 1000 * 1000 ;
}
}
DateTime::DateTime( std::time_t sec, unsigned long nsec ) :
m_sec ( sec ),
m_nsec ( nsec )
{
struct tm tp ;
const char *r = ::strptime( iso.c_str(), "%Y-%m-%dT%H:%M:%S.", &tp ) ;
m_sec = ::mktime( &tp ) ;
if ( r != 0 )
m_usec = std::atoi( r ) * 1000 ;
}
struct tm DateTime::Tm() const
@ -56,9 +70,26 @@ std::time_t DateTime::Sec( ) const
return m_sec ;
}
unsigned long DateTime::USec( ) const
unsigned long DateTime::NanoSec( ) const
{
return m_usec ;
return m_nsec ;
}
std::ostream& operator<<( std::ostream& os, const DateTime& dt )
{
struct tm tp = dt.Tm() ;
char buf[40] ;
strftime( buf, sizeof(buf), "%Y-%m-%dT%H:%M:%S", &tp ) ;
return os << buf << '.' << std::setw( 3 ) << std::setfill('0') << dt.NanoSec()/1000000 << 'Z' ;
}
struct timeval DateTime::Tv() const
{
timeval result ;
result.tv_sec = m_sec ;
result.tv_usec = m_nsec / 1000 ;
return result ;
}
} // end of namespace

View File

@ -21,6 +21,7 @@
#include <ctime>
#include <string>
#include <iosfwd>
namespace gr {
@ -29,15 +30,19 @@ class DateTime
public :
DateTime( ) ;
explicit DateTime( const std::string& iso ) ;
explicit DateTime( std::time_t sec, unsigned long nsec = 0 ) ;
std::time_t Sec( ) const ;
unsigned long USec( ) const ;
unsigned long NanoSec( ) const ;
struct tm Tm() const ;
struct timeval Tv() const ;
private :
std::time_t m_sec ;
unsigned long m_usec ;
unsigned long m_nsec ;
} ;
std::ostream& operator<<( std::ostream& os, const DateTime& dt ) ;
} // end of namespace

View File

@ -18,9 +18,13 @@
*/
#include "OS.hh"
#include "DateTime.hh"
#include <stdexcept>
// OS specific headers
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/types.h>
namespace gr { namespace os {
@ -30,4 +34,20 @@ void MakeDir( const std::string& dir )
mkdir( dir.c_str(), 0700 ) ;
}
DateTime FileMTime( const std::string& filename )
{
struct stat s = {} ;
if ( ::stat( filename.c_str(), &s ) != 0 )
throw std::runtime_error( "cannot get file attribute of " + filename ) ;
return DateTime( s.st_mtim.tv_sec, s.st_mtim.tv_nsec ) ;
}
void SetFileTime( const std::string& filename, const DateTime& t )
{
struct timeval tvp[2] = { t.Tv(), t.Tv() } ;
if ( ::utimes( filename.c_str(), tvp ) != 0 )
throw std::runtime_error( "cannot set file time" ) ;
}
} } // end of namespaces

View File

@ -28,7 +28,8 @@ class DateTime ;
namespace os
{
void MakeDir( const std::string& dir ) ;
DateTime FileMTime( const std::string& file ) ;
DateTime FileMTime( const std::string& filename ) ;
void SetFileTime( const std::string& filename, const DateTime& t ) ;
}
} // end of namespaces

View File

@ -21,6 +21,8 @@
#include "util/DateTime.hh"
#include <cstring>
namespace grut {
using namespace gr ;
@ -35,6 +37,27 @@ void DateTimeTest::TestParseIso( )
struct tm tp = subject.Tm() ;
CPPUNIT_ASSERT( tp.tm_year == 109 ) ;
CPPUNIT_ASSERT( tp.tm_sec == 39 ) ;
CPPUNIT_ASSERT_EQUAL( 804000000UL, subject.NanoSec() ) ;
}
void DateTimeTest::TestParseNoMillisec( )
{
DateTime subject( "2009-07-29T20:31:39Z" ) ;
CPPUNIT_ASSERT_EQUAL( 0UL, subject.NanoSec() ) ;
}
void DateTimeTest::TestParseInvalid( )
{
DateTime subject( "abcdefg" ) ;
CPPUNIT_ASSERT_EQUAL( static_cast<time_t>(0), subject.Sec() ) ;
CPPUNIT_ASSERT_EQUAL( 0UL, subject.NanoSec() ) ;
}
void DateTimeTest::TestOffByOne( )
{
DateTime subject( "2008-12-21T02:48:53.940Z" ) ;
struct tm tp = subject.Tm() ;
CPPUNIT_ASSERT_EQUAL( 21, tp.tm_mday ) ;
}
} // end of namespace grut

View File

@ -32,10 +32,16 @@ public :
// declare suit function
CPPUNIT_TEST_SUITE( DateTimeTest ) ;
CPPUNIT_TEST( TestParseIso ) ;
CPPUNIT_TEST( TestParseNoMillisec ) ;
CPPUNIT_TEST( TestOffByOne ) ;
CPPUNIT_TEST( TestParseInvalid ) ;
CPPUNIT_TEST_SUITE_END();
private :
void TestParseIso( ) ;
void TestParseNoMillisec( ) ;
void TestOffByOne( ) ;
void TestParseInvalid( ) ;
} ;
} // end of namespace