mirror of
https://github.com/GothenburgBitFactory/taskwarrior.git
synced 2025-07-07 20:06:36 +02:00
286 lines
8.3 KiB
C++
286 lines
8.3 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
// taskwarrior - a command line task list manager.
|
|
//
|
|
// Copyright 2006-2012, Paul Beckingham, Federico Hernandez.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included
|
|
// in all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
// SOFTWARE.
|
|
//
|
|
// http://www.opensource.org/licenses/mit-license.php
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <inttypes.h>
|
|
#include <Context.h>
|
|
#include <Socket.h>
|
|
#include <Color.h>
|
|
#include <text.h>
|
|
#include <i18n.h>
|
|
#include <CmdSync.h>
|
|
|
|
extern Context context;
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
CmdSync::CmdSync ()
|
|
{
|
|
_keyword = "synchronize";
|
|
_usage = "task synchronize";
|
|
_description = STRING_CMD_SYNC_USAGE;
|
|
_read_only = false;
|
|
_displays_id = false;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
int CmdSync::execute (std::string& output)
|
|
{
|
|
int status = 0;
|
|
Timer timer_sync;
|
|
timer_sync.start ();
|
|
|
|
std::stringstream out;
|
|
|
|
// If no server is set up, quit.
|
|
std::string connection = context.config.get ("taskd.server");
|
|
if (connection == "" ||
|
|
connection.rfind (':') == std::string::npos)
|
|
throw std::string (STRING_CMD_SYNC_NO_SERVER);
|
|
|
|
// Obtain credentials.
|
|
std::string credentials_string = context.config.get ("taskd.credentials");
|
|
if (credentials_string == "")
|
|
throw std::string (STRING_CMD_SYNC_BAD_CRED);
|
|
|
|
std::vector <std::string> credentials;
|
|
split (credentials, credentials_string, "/");
|
|
if (credentials.size () != 3)
|
|
throw std::string (STRING_CMD_SYNC_BAD_CRED);
|
|
|
|
// Read backlog.data.
|
|
std::string payload = "";
|
|
File backlog (context.config.get ("data.location") + "/backlog.data");
|
|
if (backlog.exists ())
|
|
backlog.read (payload);
|
|
|
|
// Count the number of tasks being uploaded.
|
|
int upload_count = 0;
|
|
{
|
|
std::vector <std::string> lines;
|
|
split (lines, payload, "\n");
|
|
std::vector <std::string>::iterator i;
|
|
for (i = lines.begin (); i != lines.end (); ++i)
|
|
if ((*i)[0] == '[')
|
|
++upload_count;
|
|
}
|
|
|
|
// Send 'sync' + payload.
|
|
Msg request;
|
|
request.set ("protocol", "v1");
|
|
request.set ("type", "sync");
|
|
request.set ("org", credentials[0]);
|
|
request.set ("user", credentials[1]);
|
|
request.set ("key", credentials[2]);
|
|
|
|
// TODO Add the other necessary header fields.
|
|
|
|
request.setPayload (payload);
|
|
|
|
out << format (STRING_CMD_SYNC_PROGRESS, connection)
|
|
<< "\n";
|
|
|
|
Msg response;
|
|
if (send (connection, request, response))
|
|
{
|
|
std::string code = response.get ("code");
|
|
if (code == "200")
|
|
{
|
|
Color colorAdded (context.config.get ("color.sync.added"));
|
|
Color colorChanged (context.config.get ("color.sync.changed"));
|
|
|
|
int download_count = 0;
|
|
payload = response.getPayload ();
|
|
std::vector <std::string> lines;
|
|
split (lines, payload, '\n');
|
|
|
|
// TODO This is not necessary if only a synch key was received.
|
|
// Load all tasks.
|
|
context.tdb2.all_tasks ();
|
|
|
|
std::string synch_key = "";
|
|
std::vector <std::string>::iterator line;
|
|
for (line = lines.begin (); line != lines.end (); ++line)
|
|
{
|
|
if ((*line)[0] == '[')
|
|
{
|
|
++download_count;
|
|
|
|
Task from_server (*line);
|
|
std::string uuid = from_server.get ("uuid");
|
|
|
|
// Is it a new task from the server, or an update to an existing one?
|
|
Task dummy;
|
|
if (context.tdb2.get (uuid, dummy))
|
|
{
|
|
out << " "
|
|
<< colorChanged.colorize (
|
|
format (STRING_CMD_SYNC_MOD,
|
|
uuid,
|
|
from_server.get ("description")))
|
|
<< "\n";
|
|
context.tdb2.modify (from_server, false);
|
|
}
|
|
else
|
|
{
|
|
out << " "
|
|
<< colorAdded.colorize (
|
|
format (STRING_CMD_SYNC_ADD,
|
|
uuid,
|
|
from_server.get ("description")))
|
|
<< "\n";
|
|
context.tdb2.add (from_server, false);
|
|
}
|
|
}
|
|
else if (*line != "")
|
|
{
|
|
synch_key = *line;
|
|
context.debug ("Synch key " + synch_key);
|
|
}
|
|
|
|
// Otherwise line is blank, so ignore it.
|
|
}
|
|
|
|
// Only update everything if there is a new synch_key. No synch_key means
|
|
// something horrible happened on the other end of the wire.
|
|
if (synch_key != "")
|
|
{
|
|
// Truncate backlog.data, save new synch_key.
|
|
context.tdb2.backlog._file.truncate ();
|
|
context.tdb2.backlog.clear_tasks ();
|
|
context.tdb2.backlog.clear_lines ();
|
|
context.tdb2.backlog.add_line (synch_key + "\n");
|
|
|
|
// Commit all changes.
|
|
context.tdb2.commit ();
|
|
|
|
// Present a clear status message.
|
|
if (upload_count == 0 && download_count == 0)
|
|
context.footnote (STRING_CMD_SYNC_SUCCESS0);
|
|
else if (upload_count == 0 && download_count > 0)
|
|
context.footnote (format (STRING_CMD_SYNC_SUCCESS2, download_count));
|
|
else if (upload_count > 0 && download_count == 0)
|
|
context.footnote (format (STRING_CMD_SYNC_SUCCESS1, upload_count));
|
|
else if (upload_count > 0 && download_count > 0)
|
|
context.footnote (format (STRING_CMD_SYNC_SUCCESS3, upload_count, download_count));
|
|
}
|
|
}
|
|
else if (code == "201")
|
|
{
|
|
context.footnote (STRING_CMD_SYNC_SUCCESS_NOP);
|
|
}
|
|
else if (code == "430")
|
|
{
|
|
context.error (STRING_CMD_SYNC_FAIL_ACCOUNT);
|
|
status = 2;
|
|
}
|
|
else
|
|
{
|
|
context.error (format (STRING_CMD_SYNC_FAIL_ERROR,
|
|
code,
|
|
response.get ("status")));
|
|
status = 2;
|
|
}
|
|
|
|
// Display all errors returned. This is required by the server protocol.
|
|
std::string to_be_displayed = response.get ("messages");
|
|
if (to_be_displayed != "")
|
|
{
|
|
if (context.verbose ("footnote"))
|
|
context.footnote (to_be_displayed);
|
|
else
|
|
context.debug (to_be_displayed);
|
|
}
|
|
}
|
|
|
|
// Some kind of low-level error:
|
|
// - Server down
|
|
// - Wrong address
|
|
// - Wrong port
|
|
// - Firewall
|
|
// - Network error
|
|
// - No signal/cable
|
|
else
|
|
{
|
|
context.error (STRING_CMD_SYNC_FAIL_CONNECT);
|
|
status = 1;
|
|
}
|
|
|
|
out << "\n";
|
|
output = out.str ();
|
|
|
|
/*
|
|
timer_sync.stop ();
|
|
std::stringstream s;
|
|
s << "Sync "
|
|
<< Date ().toISO ()
|
|
<< " sync:"
|
|
<< timer_sync.total ()
|
|
<< "\n";
|
|
debug (s.str ());
|
|
*/
|
|
|
|
return status;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
bool CmdSync::send (
|
|
const std::string& to,
|
|
const Msg& request,
|
|
Msg& response)
|
|
{
|
|
std::string::size_type colon = to.rfind (':');
|
|
if (colon == std::string::npos)
|
|
throw format (STRING_CMD_SYNC_BAD_SERVER, to);
|
|
|
|
std::string server = to.substr (0, colon);
|
|
std::string port = to.substr (colon + 1);
|
|
|
|
try
|
|
{
|
|
Socket s;
|
|
s.connect (server, port);
|
|
s.write (request.serialize () + "\n");
|
|
|
|
std::string incoming;
|
|
s.read (incoming);
|
|
s.close ();
|
|
|
|
response.parse (incoming);
|
|
return true;
|
|
}
|
|
|
|
catch (std::string& error)
|
|
{
|
|
context.debug (error);
|
|
}
|
|
|
|
// Indicate message failed.
|
|
return false;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|