/*
	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 "OAuth2.hh"

#include "json/ValResponse.hh"

#include "http/CurlAgent.hh"
#include "http/Header.hh"
#include "util/log/Log.hh"

// for debugging
#include <iostream>

namespace gr {

const std::string token_url		= "https://accounts.google.com/o/oauth2/token" ;

OAuth2::OAuth2(
	const std::string& refresh_code,
	const std::string&	client_id,
	const std::string&	client_secret ) :
	m_refresh( refresh_code ),
	m_client_id( client_id ),
	m_client_secret( client_secret )
{
	Refresh( ) ;
}

OAuth2::OAuth2(
	const std::string&	client_id,
	const std::string&	client_secret ) :
	m_client_id( client_id ),
	m_client_secret( client_secret )
{
}

void OAuth2::Auth( const std::string&	auth_code )
{
	std::string post =
		"code="				+ auth_code +
		"&client_id="		+ m_client_id +
		"&client_secret="	+ m_client_secret +
		"&redirect_uri="	+ "urn:ietf:wg:oauth:2.0:oob" +
		"&grant_type=authorization_code" ;

	http::ValResponse  resp ;
	http::CurlAgent		http ;

	DisableLog dlog( log::debug ) ;
	http.Post( token_url, post, &resp, http::Header() ) ;

	Val jresp	= resp.Response() ;
	m_access	= jresp["access_token"].Str() ;
	m_refresh	= jresp["refresh_token"].Str() ;
}

std::string OAuth2::MakeAuthURL(
	const std::string&	client_id,
	const std::string&	state )
{
	http::CurlAgent h ;

	return "https://accounts.google.com/o/oauth2/auth"
		"?scope=" +
			h.Escape( "https://www.googleapis.com/auth/userinfo.email" )	+ "+" + 
			h.Escape( "https://www.googleapis.com/auth/userinfo.profile" )	+ "+" +
			h.Escape( "https://docs.google.com/feeds/" )					+ "+" + 
			h.Escape( "https://docs.googleusercontent.com/" )				+ "+" + 
			h.Escape( "https://spreadsheets.google.com/feeds/" )			+
		"&redirect_uri=urn:ietf:wg:oauth:2.0:oob"
		"&response_type=code"
		"&client_id=" + client_id ;
}

void OAuth2::Refresh( )
{
	std::string post =
		"refresh_token="	+ m_refresh +
		"&client_id="		+ m_client_id +
		"&client_secret="	+ m_client_secret +
		"&grant_type=refresh_token" ;

	http::ValResponse  resp ;
	http::CurlAgent		http ;
    
	DisableLog dlog( log::debug ) ;
	http.Post( token_url, post, &resp, http::Header() ) ;

	m_access	= resp.Response()["access_token"].Str() ;
}

std::string OAuth2::RefreshToken( ) const
{
	return m_refresh ;
}

std::string OAuth2::AccessToken( ) const
{
	return m_access ;
}

std::string OAuth2::HttpHeader( ) const
{
	return "Authorization: Bearer " + m_access ;
}

} // end of namespace