grive2/libgrive/src/protocol/AuthAgent.cc

104 lines
2.3 KiB
C++

/*
grive: an GPL program to sync a local directory with Google Drive
Copyright (C) 2012 Wan Wai Ho
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 version 2
of the License.
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 "AuthAgent.hh"
#include "http/Header.hh"
#include <cassert>
namespace gr {
using namespace http ;
AuthAgent::AuthAgent( const OAuth2& auth, std::auto_ptr<Agent> real_agent ) :
m_auth ( auth ),
m_agent ( real_agent )
{
assert( m_agent.get() != 0 ) ;
}
Header AuthAgent::AppendHeader( const Header& hdr ) const
{
Header h(hdr) ;
h.Add( "Authorization: Bearer " + m_auth.AccessToken() ) ;
h.Add( "GData-Version: 3.0" ) ;
return h ;
}
long AuthAgent::Put(
const std::string& url,
const std::string& data,
Receivable *dest,
const Header& hdr )
{
return m_agent->Put( url, data, dest, AppendHeader(hdr) ) ;
}
long AuthAgent::Put(
const std::string& url,
StdioFile& file,
Receivable *dest,
const Header& hdr )
{
return m_agent->Put( url, file, dest, AppendHeader(hdr) ) ;
}
long AuthAgent::Get(
const std::string& url,
Receivable *dest,
const Header& hdr )
{
return m_agent->Get( url, dest, AppendHeader(hdr) ) ;
}
long AuthAgent::Post(
const std::string& url,
const std::string& data,
Receivable *dest,
const Header& hdr )
{
return m_agent->Post( url, data, dest, AppendHeader(hdr) ) ;
}
long AuthAgent::Custom(
const std::string& method,
const std::string& url,
Receivable *dest,
const Header& hdr )
{
return m_agent->Custom( method, url, dest, AppendHeader(hdr) ) ;
}
std::string AuthAgent::RedirLocation() const
{
return m_agent->RedirLocation() ;
}
std::string AuthAgent::Escape( const std::string& str )
{
return m_agent->Escape( str ) ;
}
std::string AuthAgent::Unescape( const std::string& str )
{
return m_agent->Unescape( str ) ;
}
} // end of namespace