emacs
/
mu
mirror of https://github.com/djcb/mu.git
1
0
Fork 0
Code Issues Releases Wiki Activity
mu/mu/mu-cmd-view.cc

345 lines
8.2 KiB
C++
Raw Blame History

/*
** Copyright (C) 2023 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
**
** 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; either version 3, or (at your option) any
** later version.
**
** 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 <config.h>
#include "mu-cmd.hh"
#include "message/mu-message.hh"
#include <iostream>
#include <iomanip>
using namespace Mu;
#define VIEW_TERMINATOR '\f' /* form-feed */
using namespace Mu;
static Mu::Result<void>
view_msg_sexp(const Message& message, const Options& opts)
{
::fputs(message.sexp().to_string().c_str(), stdout);
::fputs("\n", stdout);
return Ok();
}
static std::string /* return comma-sep'd list of attachments */
get_attach_str(const Message& message, const Options& opts)
{
std::string str;
seq_for_each(message.parts(), [&](auto&& part) {
if (auto fname = part.raw_filename(); fname) {
if (str.empty())
str = fname.value();
else
str += ", " + fname.value();
}
});
return str;
}
#define color_maybe(C) \
do { \
if (color) \
fputs((C), stdout); \
} while (0)
static void
print_field(const std::string& field, const std::string& val, bool color)
{
if (val.empty())
return;
color_maybe(MU_COLOR_MAGENTA);
fputs_encoded(field, stdout);
color_maybe(MU_COLOR_DEFAULT);
fputs(": ", stdout);
color_maybe(MU_COLOR_GREEN);
fputs_encoded(val, stdout);
color_maybe(MU_COLOR_DEFAULT);
fputs("\n", stdout);
}
/* a summary_len of 0 mean 'don't show summary, show body */
static void
body_or_summary(const Message& message, const Options& opts)
{
const auto color{!opts.nocolor};
using Format = Options::View::Format;
std::string body, btype;
switch (opts.view.format) {
case Format::Plain:
btype = "plain text";
body = message.body_text().value_or("");
break;
case Format::Html:
btype = "html";
body = message.body_html().value_or("");
break;
default:
throw std::range_error("unsupported format"); // bug
}
if (body.empty()) {
if (any_of(message.flags() & Flags::Encrypted)) {
color_maybe(MU_COLOR_CYAN);
mu_println("[No {} body found; message does have encrypted parts]",
btype);
} else {
color_maybe(MU_COLOR_MAGENTA);
mu_println("[No {} body found]", btype);
}
color_maybe(MU_COLOR_DEFAULT);
return;
}
if (opts.view.summary_len) {
const auto summ{summarize(body, *opts.view.summary_len)};
print_field("Summary", summ, color);
} else {
mu_print_encoded("{}", body);
if (!g_str_has_suffix(body.c_str(), "\n"))
mu_println("");
}
}
/* we ignore fields for now */
/* summary_len == 0 means "no summary */
static Mu::Result<void>
view_msg_plain(const Message& message, const Options& opts)
{
const auto color{!opts.nocolor};
print_field("From", to_string(message.from()), color);
print_field("To", to_string(message.to()), color);
print_field("Cc", to_string(message.cc()), color);
print_field("Bcc", to_string(message.bcc()), color);
print_field("Subject", message.subject(), color);
if (auto&& date = message.date(); date != 0)
print_field("Date", mu_format("{:%c}", mu_time(date)), color);
print_field("Tags", join(message.tags(), ", "), color);
print_field("Attachments",get_attach_str(message, opts), color);
mu_println("");
body_or_summary(message, opts);
return Ok();
}
static Mu::Result<void>
handle_msg(const Message& message, const Options& opts)
{
using Format = Options::View::Format;
switch (opts.view.format) {
case Format::Plain:
case Format::Html:
return view_msg_plain(message, opts);
case Format::Sexp:
return view_msg_sexp(message, opts);
default:
mu_critical("bug: should not be reached");
return Err(Error::Code::Internal, "error");
}
}
Mu::Result<void>
Mu::mu_cmd_view(const Options& opts)
{
for (auto&& file: opts.view.files) {
auto message{Message::make_from_path(
file, message_options(opts.view))};
if (!message)
return Err(message.error());
if (auto res = handle_msg(*message, opts); !res)
return res;
/* add a separator between two messages? */
if (opts.view.terminate)
mu_print("{}", VIEW_TERMINATOR);
}
// no files? read from stding
if (opts.view.files.empty()) {
const auto msgtxt = read_from_stdin();
if (!msgtxt)
return Err(msgtxt.error());
auto message = Message::make_from_text(*msgtxt,{},
message_options(opts.view));
if (!message)
return Err(message.error());
else
return handle_msg(*message, opts);
}
return Ok();
}
#ifdef BUILD_TESTS
/*
* Tests.
*
*/
#include <fcntl.h> /* Definition of AT_* constants */
#include <sys/stat.h>
#include <fstream>
#include <utils/mu-regex.hh>
#include "utils/mu-test-utils.hh"
static constexpr std::string_view test_msg =
R"(From: Test <test@example.com>
To: abc@example.com
Date: Mon, 23 May 2011 10:53:45 +0200
Subject: vla
MIME-Version: 1.0
Content-Type: multipart/alternative;
boundary="_=aspNetEmail=_5ed4592191214c7a99bd7f6a3a0f077d"
Message-ID: <10374608.109906.11909.20115aabbccdd.MSGID@mailinglijst.nl>
--_=aspNetEmail=_5ed4592191214c7a99bd7f6a3a0f077d
Content-Type: text/plain; charset="iso-8859-15"
Content-Transfer-Encoding: quoted-printable
text
--_=aspNetEmail=_5ed4592191214c7a99bd7f6a3a0f077d
Content-Type: text/html; charset="iso-8859-15"
Content-Transfer-Encoding: quoted-printable
html
--_=aspNetEmail=_5ed4592191214c7a99bd7f6a3a0f077d--
)";
static std::string msgpath;
static void
test_view_plain()
{
auto res = run_command({MU_PROGRAM, "view", msgpath});
assert_valid_command(res);
auto output{*res};
// silly hack to avoid locale diffs
auto rx = unwrap(Regex::make("^Date:.*", G_REGEX_MULTILINE));
output.standard_out = rx.replace(output.standard_out, "Date: xxx");
g_assert_true(output.standard_err.empty());
assert_equal(output.standard_out,
R"(From: Test <test@example.com>
To: abc@example.com
Subject: vla
Date: xxx
text
)");
}
static void
test_view_html()
{
auto res = run_command({MU_PROGRAM, "view", "--format=html", msgpath});
assert_valid_command(res);
auto output{*res};
auto rx = unwrap(Regex::make("^Date:.*", G_REGEX_MULTILINE));
output.standard_out = rx.replace(output.standard_out, "Date: xxx");
g_assert_true(output.standard_err.empty());
assert_equal(output.standard_out,
R"(From: Test <test@example.com>
To: abc@example.com
Subject: vla
Date: xxx
html
)");
}
static void
test_view_sexp()
{
TempTz tz("Europe/Amsterdam");
if (!tz.available()) {
g_test_skip("timezone not available");
return;
}
auto res = run_command({MU_PROGRAM, "view", "--format=sexp", msgpath});
assert_valid_command(res);
auto output{*res};
g_assert_true(output.standard_err.empty());
// Note: :path, :changed (file ctime) change per run.
struct stat statbuf{};
g_assert_true(::stat(msgpath.c_str(), &statbuf) == 0);
const auto expected = mu_format(
R"((:path "{}" :size 638 :changed ({} {} 0) :date (19930 8345 0) :flags (unread) :from ((:email "test@example.com" :name "Test")) :message-id "10374608.109906.11909.20115aabbccdd.MSGID@mailinglijst.nl" :priority normal :subject "vla" :to ((:email "abc@example.com")))
)",
msgpath,
statbuf.st_ctime >> 16,
statbuf.st_ctime & 0xffff);
assert_equal(output.standard_out, expected);
}
int
main(int argc, char* argv[]) try {
TempDir tmpdir{false};
msgpath = join_paths(tmpdir.path(), "test-message.txt");
std::ofstream strm{msgpath};
strm.write(test_msg.data(), test_msg.size());
strm.close();
g_assert_true(strm.good());
mu_test_init(&argc, &argv);
g_test_add_func("/cmd/view/plain", test_view_plain);
g_test_add_func("/cmd/view/html", test_view_html);
g_test_add_func("/cmd/view/sexp", test_view_sexp);
return g_test_run();
} catch (const Error& e) {
mu_printerrln("{}", e.what());
return 1;
} catch (...) {
mu_printerrln("caught exception");
return 1;
}
#endif /*BUILD_TESTS*/
Reference in New Issue View Git Blame Copy Permalink