timewarrior/src/commands/CmdReport.cpp
Thomas Lauf c546add03f Use new filtering in getTracked
Replace getTracked in commands CmdSummary, CmdChart, CmdExport, CmdFill,
CmdReport, CmdTags, and CmdContinue. Also in  functions domGet and autoFill

Relates to #468

Signed-off-by: Thomas Lauf <thomas.lauf@tngtech.com>
2021-12-02 22:31:15 +01:00

133 lines
4.2 KiB
C++

////////////////////////////////////////////////////////////////////////////////
//
// Copyright 2016 - 2021, Thomas Lauf, 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.
//
// https://www.opensource.org/licenses/mit-license.php
//
////////////////////////////////////////////////////////////////////////////////
#include <cmake.h>
#include <commands.h>
#include <shared.h>
#include <format.h>
#include <timew.h>
#include <iostream>
#include <sstream>
#include <FS.h>
#include <IntervalFilterAllInRange.h>
#include <IntervalFilterAllWithTags.h>
#include <IntervalFilterAndGroup.h>
////////////////////////////////////////////////////////////////////////////////
// Given a partial match for an extension script name, find the full patch of
// the extension it may match.
static std::string findExtension (
const Extensions& extensions,
const std::string& partial)
{
auto scripts = extensions.all ();
std::vector <std::string> options;
for (auto& script : scripts)
{
options.push_back (Path (script).name ());
}
std::vector <std::string> matches;
autoComplete (partial, options, matches);
if (matches.empty ())
{
throw format ("The report '{1}' is not recognized.", partial);
}
if (matches.size () > 1)
{
throw format ("The report '{1}' is ambiguous, and could mean one of: {2}",
partial,
join (", ", matches));
}
// Now map matches[0] back to it's corresponding option.
for (auto& script : scripts)
{
if (Path (script).name () == matches[0])
{
return script;
}
}
return "";
}
////////////////////////////////////////////////////////////////////////////////
int CmdReport (
const CLI& cli,
Rules& rules,
Database& database,
const Extensions& extensions)
{
std::string script;
for (auto& arg : cli._args)
if (arg.hasTag ("EXT"))
script = findExtension (extensions, arg.attribute ("canonical"));
if (script.empty ())
throw std::string ("Specify which report to run.");
// Compose Header info.
auto filter = cli.getFilter ();
auto filtering = IntervalFilterAndGroup ({
new IntervalFilterAllInRange ({ filter.start, filter.end }),
new IntervalFilterAllWithTags (filter.tags ())
});
auto tracked = getTracked (database, rules, filtering);
rules.set ("temp.report.start", filter.is_started () ? filter.start.toISO () : "");
rules.set ("temp.report.end", filter.is_ended () ? filter.end.toISO () : "");
rules.set ("temp.report.tags", joinQuotedIfNeeded (",", filter.tags ()));
rules.set ("temp.version", VERSION);
std::stringstream header;
for (auto& name : rules.all ())
header << name << ": " << rules.get (name) << '\n';
// Get the data.
auto input = header.str ()
+ '\n'
+ jsonFromIntervals (tracked);
// Run the extensions.
std::vector <std::string> output;
int rc = extensions.callExtension (script, split (input, '\n'), output);
if (rc != 0 && output.size () == 0)
{
throw format ("'{1}' returned {2} without producing output.", script, rc);
}
// Display the output.
for (auto& line : output)
std::cout << line << '\n';
return 0;
}
////////////////////////////////////////////////////////////////////////////////