transmission/utils/edit.cc

409 lines
12 KiB
C++
Raw Normal View History

// This file Copyright © 2012-2022 Mnemosyne LLC.
// It may be used under GPLv2 (SPDX: GPL-2.0-only), GPLv3 (SPDX: GPL-3.0-only),
// or any future license endorsed by Mnemosyne LLC.
// License text can be found in the licenses/ folder.
2021-12-17 20:48:02 +00:00
#include <array>
#include <stdio.h> /* fprintf() */
#include <stdlib.h> /* EXIT_FAILURE */
#include <string>
#include <string_view>
2021-12-17 20:48:02 +00:00
#include <vector>
#include <libtransmission/transmission.h>
#include <libtransmission/error.h>
#include <libtransmission/log.h>
#include <libtransmission/tr-getopt.h>
#include <libtransmission/utils.h>
#include <libtransmission/variant.h>
#include <libtransmission/version.h>
2021-12-17 20:48:02 +00:00
static char constexpr MyName[] = "transmission-edit";
static char constexpr Usage[] = "Usage: transmission-edit [options] torrent-file(s)";
2021-12-17 20:48:02 +00:00
struct app_options
{
std::vector<std::string_view> files;
2021-12-17 20:48:02 +00:00
char const* add = nullptr;
char const* deleteme = nullptr;
std::array<char const*, 2> replace;
char const* source = nullptr;
2021-12-17 20:48:02 +00:00
bool show_version = false;
};
static auto constexpr Options = std::array<tr_option, 6>{
2021-12-17 20:48:02 +00:00
{ { 'a', "add", "Add a tracker's announce URL", "a", true, "<url>" },
{ 'd', "delete", "Delete a tracker's announce URL", "d", true, "<url>" },
{ 'r', "replace", "Search and replace a substring in the announce URLs", "r", true, "<old> <new>" },
{ 's', "source", "Set the source", "s", true, "<source>" },
2021-12-17 20:48:02 +00:00
{ 'V', "version", "Show version number and exit", "V", false, nullptr },
{ 0, nullptr, nullptr, nullptr, false, nullptr } }
};
2021-12-17 20:48:02 +00:00
static int parseCommandLine(app_options& opts, int argc, char const* const* argv)
{
int c;
char const* optarg;
2021-12-17 20:48:02 +00:00
while ((c = tr_getopt(Usage, argc, argv, std::data(Options), &optarg)) != TR_OPT_DONE)
{
switch (c)
{
case 'a':
2021-12-17 20:48:02 +00:00
opts.add = optarg;
break;
case 'd':
2021-12-17 20:48:02 +00:00
opts.deleteme = optarg;
break;
case 'r':
2021-12-17 20:48:02 +00:00
opts.replace[0] = optarg;
c = tr_getopt(Usage, argc, argv, std::data(Options), &optarg);
if (c != TR_OPT_UNK)
{
return 1;
}
2021-12-17 20:48:02 +00:00
opts.replace[1] = optarg;
break;
case 's':
opts.source = optarg;
break;
case 'V':
2021-12-17 20:48:02 +00:00
opts.show_version = true;
break;
case TR_OPT_UNK:
2021-12-17 20:48:02 +00:00
opts.files.push_back(optarg);
break;
default:
return 1;
}
}
return 0;
}
static bool removeURL(tr_variant* metainfo, std::string_view url)
{
auto sv = std::string_view{};
tr_variant* announce_list;
bool changed = false;
if (tr_variantDictFindStrView(metainfo, TR_KEY_announce, &sv) && url == sv)
{
printf("\tRemoved \"%" TR_PRIsv "\" from \"announce\"\n", TR_PRIsv_ARG(sv));
tr_variantDictRemove(metainfo, TR_KEY_announce);
changed = true;
}
if (tr_variantDictFindList(metainfo, TR_KEY_announce_list, &announce_list))
{
tr_variant* tier;
int tierIndex = 0;
while ((tier = tr_variantListChild(announce_list, tierIndex)) != nullptr)
{
int nodeIndex = 0;
tr_variant const* node;
while ((node = tr_variantListChild(tier, nodeIndex)) != nullptr)
{
if (tr_variantGetStrView(node, &sv) && url == sv)
{
printf("\tRemoved \"%" TR_PRIsv "\" from \"announce-list\" tier #%d\n", TR_PRIsv_ARG(sv), tierIndex + 1);
tr_variantListRemove(tier, nodeIndex);
changed = true;
}
else
{
++nodeIndex;
}
}
if (tr_variantListSize(tier) == 0)
{
printf("\tNo URLs left in tier #%d... removing tier\n", tierIndex + 1);
tr_variantListRemove(announce_list, tierIndex);
}
else
{
++tierIndex;
}
}
if (tr_variantListSize(announce_list) == 0)
{
printf("\tNo tiers left... removing announce-list\n");
tr_variantDictRemove(metainfo, TR_KEY_announce_list);
}
}
/* if we removed the "announce" field and there's still another track left,
* use it as the "announce" field */
if (changed && !tr_variantDictFindStrView(metainfo, TR_KEY_announce, &sv))
{
tr_variant* const tier = tr_variantListChild(announce_list, 0);
if (tier != nullptr)
{
tr_variant const* const node = tr_variantListChild(tier, 0);
if ((node != nullptr) && tr_variantGetStrView(node, &sv))
{
tr_variantDictAddStr(metainfo, TR_KEY_announce, sv);
printf("\tAdded \"%" TR_PRIsv "\" to announce\n", TR_PRIsv_ARG(sv));
}
}
}
return changed;
}
static std::string replaceSubstr(std::string_view str, std::string_view oldval, std::string_view newval)
{
auto ret = std::string{};
while (!std::empty(str))
{
auto const pos = str.find(oldval);
ret += str.substr(0, pos);
if (pos == str.npos)
{
break;
}
2021-12-17 20:48:02 +00:00
ret += newval;
str.remove_prefix(pos + std::size(oldval));
}
return ret;
}
static bool replaceURL(tr_variant* metainfo, std::string_view oldval, std::string_view newval)
{
auto sv = std::string_view{};
tr_variant* announce_list;
bool changed = false;
if (tr_variantDictFindStrView(metainfo, TR_KEY_announce, &sv) && tr_strvContains(sv, oldval))
{
auto const newstr = replaceSubstr(sv, oldval, newval);
printf("\tReplaced in \"announce\": \"%" TR_PRIsv "\" --> \"%s\"\n", TR_PRIsv_ARG(sv), newstr.c_str());
tr_variantDictAddStr(metainfo, TR_KEY_announce, newstr);
changed = true;
}
if (tr_variantDictFindList(metainfo, TR_KEY_announce_list, &announce_list))
{
tr_variant* tier;
int tierCount = 0;
while ((tier = tr_variantListChild(announce_list, tierCount)) != nullptr)
{
tr_variant* node;
int nodeCount = 0;
while ((node = tr_variantListChild(tier, nodeCount)) != nullptr)
{
if (tr_variantGetStrView(node, &sv) && tr_strvContains(sv, oldval))
{
auto const newstr = replaceSubstr(sv, oldval, newval);
printf(
"\tReplaced in \"announce-list\" tier %d: \"%" TR_PRIsv "\" --> \"%s\"\n",
tierCount + 1,
TR_PRIsv_ARG(sv),
newstr.c_str());
tr_variantClear(node);
tr_variantInitStr(node, newstr);
changed = true;
}
++nodeCount;
}
++tierCount;
}
}
return changed;
}
static bool announce_list_has_url(tr_variant* announce_list, char const* url)
{
int tierCount = 0;
tr_variant* tier;
while ((tier = tr_variantListChild(announce_list, tierCount)) != nullptr)
{
int nodeCount = 0;
tr_variant const* node;
while ((node = tr_variantListChild(tier, nodeCount)) != nullptr)
{
auto sv = std::string_view{};
if (tr_variantGetStrView(node, &sv) && sv == url)
{
return true;
}
++nodeCount;
}
++tierCount;
}
return false;
}
static bool addURL(tr_variant* metainfo, char const* url)
{
auto announce = std::string_view{};
tr_variant* announce_list = nullptr;
bool changed = false;
bool const had_announce = tr_variantDictFindStrView(metainfo, TR_KEY_announce, &announce);
bool const had_announce_list = tr_variantDictFindList(metainfo, TR_KEY_announce_list, &announce_list);
if (!had_announce && !had_announce_list)
{
/* this new tracker is the only one, so add it to "announce"... */
printf("\tAdded \"%s\" in \"announce\"\n", url);
tr_variantDictAddStr(metainfo, TR_KEY_announce, url);
changed = true;
}
else
{
if (!had_announce_list)
{
announce_list = tr_variantDictAddList(metainfo, TR_KEY_announce_list, 2);
if (had_announce)
{
/* we're moving from an 'announce' to an 'announce-list',
* so copy the old announce URL to the list */
tr_variant* tier = tr_variantListAddList(announce_list, 1);
tr_variantListAddStr(tier, announce);
changed = true;
}
}
/* If the user-specified URL isn't in the announce list yet, add it */
if (!announce_list_has_url(announce_list, url))
{
tr_variant* tier = tr_variantListAddList(announce_list, 1);
tr_variantListAddStr(tier, url);
printf("\tAdded \"%s\" to \"announce-list\" tier %zu\n", url, tr_variantListSize(announce_list));
changed = true;
}
}
return changed;
}
static bool setSource(tr_variant* metainfo, char const* source_value)
{
auto current_source = std::string_view{};
bool const had_source = tr_variantDictFindStrView(metainfo, TR_KEY_source, &current_source);
bool changed = false;
if (!had_source)
{
printf("\tAdded \"%s\" as source\n", source_value);
tr_variantDictAddStr(metainfo, TR_KEY_source, source_value);
changed = true;
}
else if (current_source.compare(source_value) != 0)
{
printf("\tUpdated source: \"%s\" -> \"%s\"\n", current_source.data(), source_value);
tr_variantDictAddStr(metainfo, TR_KEY_source, source_value);
changed = true;
}
return changed;
}
int tr_main(int argc, char* argv[])
{
int changedCount = 0;
tr_logSetLevel(TR_LOG_ERROR);
2021-12-17 20:48:02 +00:00
auto options = app_options{};
if (parseCommandLine(options, argc, (char const* const*)argv) != 0)
{
return EXIT_FAILURE;
}
2021-12-17 20:48:02 +00:00
if (options.show_version)
{
2021-12-17 20:48:02 +00:00
fprintf(stderr, "%s %s\n", MyName, LONG_VERSION_STRING);
return EXIT_SUCCESS;
}
2021-12-17 20:48:02 +00:00
if (std::empty(options.files))
{
fprintf(stderr, "ERROR: No torrent files specified.\n");
2021-12-17 20:48:02 +00:00
tr_getopt_usage(MyName, Usage, std::data(Options));
fprintf(stderr, "\n");
return EXIT_FAILURE;
}
if (options.add == nullptr && options.deleteme == nullptr && options.replace[0] == nullptr && options.source == nullptr)
{
fprintf(stderr, "ERROR: Must specify -a, -d, -r or -s\n");
2021-12-17 20:48:02 +00:00
tr_getopt_usage(MyName, Usage, std::data(Options));
fprintf(stderr, "\n");
return EXIT_FAILURE;
}
2021-12-17 20:48:02 +00:00
for (auto const& filename : options.files)
{
tr_variant top;
bool changed = false;
tr_error* error = nullptr;
printf("%" TR_PRIsv "\n", TR_PRIsv_ARG(filename));
if (!tr_variantFromFile(&top, TR_VARIANT_PARSE_BENC, filename, &error))
{
printf("\tError reading file: %s\n", error->message);
tr_error_free(error);
continue;
}
2021-12-17 20:48:02 +00:00
if (options.deleteme != nullptr)
{
2021-12-17 20:48:02 +00:00
changed |= removeURL(&top, options.deleteme);
}
2021-12-17 20:48:02 +00:00
if (options.add != nullptr)
{
2021-12-17 20:48:02 +00:00
changed = addURL(&top, options.add);
}
2021-12-17 20:48:02 +00:00
if (options.replace[0] != nullptr && options.replace[1] != nullptr)
{
2021-12-17 20:48:02 +00:00
changed |= replaceURL(&top, options.replace[0], options.replace[1]);
}
if (options.source != nullptr)
{
changed = setSource(&top, options.source);
}
if (changed)
{
++changedCount;
tr_variantToFile(&top, TR_VARIANT_FMT_BENC, filename);
}
tr_variantClear(&top);
}
printf("Changed %d files\n", changedCount);
return EXIT_SUCCESS;
}