src/insets/InsetCitation.cpp

   1 /**
   2  * \file InsetCitation.cpp
   3  * This file is part of LyX, the document processor.
   4  * Licence details can be found in the file COPYING.
   5  *
   6  * \author Angus Leeming
   7  * \author Herbert Voß
   8  *
   9  * Full author contact details are available in file CREDITS.
  10  */
  11
  12 #include <config.h>
  13
  14 #include "InsetCitation.h"
  15
  16 #include "BiblioInfo.h"
  17 #include "Buffer.h"
  18 #include "buffer_funcs.h"
  19 #include "BufferParams.h"
  20 #include "BufferView.h"
  21 #include "DispatchResult.h"
  22 #include "FuncCode.h"
  23 #include "FuncRequest.h"
  24 #include "FuncStatus.h"
  25 #include "LaTeXFeatures.h"
  26 #include "output_xhtml.h"
  27 #include "ParIterator.h"
  28 #include "texstream.h"
  29 #include "TocBackend.h"
  30
  31 #include "support/debug.h"
  32 #include "support/docstream.h"
  33 #include "support/FileNameList.h"
  34 #include "support/gettext.h"
  35 #include "support/lstrings.h"
  36
  37 #include <algorithm>
  38 #include <climits>
  39
  40 using namespace std;
  41 using namespace lyx::support;
  42
  43 namespace lyx {
  44
  45 InsetCitation::InsetCitation(Buffer * buf, InsetCommandParams const & p)
  46         : InsetCommand(buf, p)
  47 {
  48         buffer().removeBiblioTempFiles();
  49 }
  50
  51
  52 InsetCitation::~InsetCitation()
  53 {
  54         if (isBufferLoaded())
  55                 /* We do not use buffer() because Coverity believes that this
  56                  * may throw an exception. Actually this code path is not
  57                  * taken when buffer_ == 0 */
  58                 buffer_->removeBiblioTempFiles();
  59 }
  60
  61
  62 // May well be over-ridden when session settings are loaded
  63 // in GuiCitation. Unfortunately, that will not happen until
  64 // such a dialog is created.
  65 bool InsetCitation::last_literal = true;
  66
  67
  68 ParamInfo const & InsetCitation::findInfo(string const & /* cmdName */)
  69 {
  70         static ParamInfo param_info_;
  71
  72         // standard cite does only take one argument, but biblatex, jurabib
  73         // and natbib extend this to two arguments, so
  74         // we have to allow both here. InsetCitation takes care that
  75         // LaTeX output is nevertheless correct.
  76         if (param_info_.empty()) {
  77                 param_info_.add("after", ParamInfo::LATEX_OPTIONAL,
  78                                 ParamInfo::HANDLING_LATEXIFY);
  79                 param_info_.add("before", ParamInfo::LATEX_OPTIONAL,
  80                                 ParamInfo::HANDLING_LATEXIFY);
  81                 param_info_.add("key", ParamInfo::LATEX_REQUIRED);
  82                 param_info_.add("pretextlist", ParamInfo::LATEX_OPTIONAL,
  83                                 ParamInfo::HANDLING_LATEXIFY);
  84                 param_info_.add("posttextlist", ParamInfo::LATEX_OPTIONAL,
  85                                 ParamInfo::HANDLING_LATEXIFY);
  86                 param_info_.add("literal", ParamInfo::LYX_INTERNAL);
  87         }
  88         return param_info_;
  89 }
  90
  91
  92 // We allow any command here, since we fall back to cite
  93 // anyway if a command is not allowed by a style
  94 bool InsetCitation::isCompatibleCommand(string const &)
  95 {
  96         return true;
  97 }
  98
  99
 100 CitationStyle InsetCitation::getCitationStyle(BufferParams const & bp, string const & input,
 101                                   vector<CitationStyle> const & valid_styles) const
 102 {
 103         CitationStyle cs = valid_styles[0];
 104         cs.forceUpperCase = false;
 105         cs.hasStarredVersion = false;
 106
 107         string normalized_input = input;
 108         string::size_type const n = input.size() - 1;
 109         if (isUpperCase(input[0]))
 110                 normalized_input[0] = lowercase(input[0]);
 111         if (input[n] == '*')
 112                 normalized_input = normalized_input.substr(0, n);
 113
 114         string const alias = bp.getCiteAlias(normalized_input);
 115         if (!alias.empty())
 116                 normalized_input = alias;
 117
 118         vector<CitationStyle>::const_iterator it  = valid_styles.begin();
 119         vector<CitationStyle>::const_iterator end = valid_styles.end();
 120         for (; it != end; ++it) {
 121                 CitationStyle this_cs = *it;
 122                 if (this_cs.name == normalized_input) {
 123                         cs = *it;
 124                         break;
 125                 }
 126         }
 127
 128         return cs;
 129 }
 130
 131
 132 void InsetCitation::doDispatch(Cursor & cur, FuncRequest & cmd)
 133 {
 134         switch (cmd.action()) {
 135         case LFUN_INSET_MODIFY: {
 136                 buffer().removeBiblioTempFiles();
 137                 cache.recalculate = true;
 138                 if (cmd.getArg(0) == "toggleparam") {
 139                         string cmdname = getCmdName();
 140                         string const alias =
 141                                 buffer().masterParams().getCiteAlias(cmdname);
 142                         if (!alias.empty())
 143                                 cmdname = alias;
 144                         string const par = cmd.getArg(1);
 145                         string newcmdname = cmdname;
 146                         if (par == "star") {
 147                                 if (suffixIs(cmdname, "*"))
 148                                         newcmdname = rtrim(cmdname, "*");
 149                                 else
 150                                         newcmdname = cmdname + "*";
 151                         } else if (par == "casing") {
 152                                 if (isUpperCase(cmdname[0]))
 153                                         newcmdname[0] = lowercase(cmdname[0]);
 154                                 else
 155                                         newcmdname[0] = uppercase(newcmdname[0]);
 156                         }
 157                         cmd = FuncRequest(LFUN_INSET_MODIFY, "changetype " + newcmdname);
 158                 }
 159         }
 160         // fall through
 161         default:
 162                 InsetCommand::doDispatch(cur, cmd);
 163         }
 164 }
 165
 166
 167 bool InsetCitation::getStatus(Cursor & cur, FuncRequest const & cmd,
 168         FuncStatus & status) const
 169 {
 170         switch (cmd.action()) {
 171         // Handle the alias case
 172         case LFUN_INSET_MODIFY:
 173                 if (cmd.getArg(0) == "changetype") {
 174                         string cmdname = getCmdName();
 175                         string const alias =
 176                                 buffer().masterParams().getCiteAlias(cmdname);
 177                         if (!alias.empty())
 178                                 cmdname = alias;
 179                         if (suffixIs(cmdname, "*"))
 180                                 cmdname = rtrim(cmdname, "*");
 181                         string const newtype = cmd.getArg(1);
 182                         status.setEnabled(isCompatibleCommand(newtype));
 183                         status.setOnOff(newtype == cmdname);
 184                 }
 185                 if (cmd.getArg(0) == "toggleparam") {
 186                         string cmdname = getCmdName();
 187                         string const alias =
 188                                 buffer().masterParams().getCiteAlias(cmdname);
 189                         if (!alias.empty())
 190                                 cmdname = alias;
 191                         vector<CitationStyle> citation_styles =
 192                                 buffer().masterParams().citeStyles();
 193                         CitationStyle cs = getCitationStyle(buffer().masterParams(),
 194                                                             cmdname, citation_styles);
 195                         if (cmd.getArg(1) == "star") {
 196                                 status.setEnabled(cs.hasStarredVersion);
 197                                 status.setOnOff(suffixIs(cmdname, "*"));
 198                         }
 199                         else if (cmd.getArg(1) == "casing") {
 200                                 status.setEnabled(cs.forceUpperCase);
 201                                 status.setOnOff(isUpperCase(cmdname[0]));
 202                         }
 203                 }
 204                 return true;
 205         default:
 206                 return InsetCommand::getStatus(cur, cmd, status);
 207         }
 208 }
 209
 210
 211 bool InsetCitation::addKey(string const & key)
 212 {
 213         docstring const ukey = from_utf8(key);
 214         docstring const & curkeys = getParam("key");
 215         if (curkeys.empty()) {
 216                 setParam("key", ukey);
 217                 cache.recalculate = true;
 218                 return true;
 219         }
 220
 221         vector<docstring> keys = getVectorFromString(curkeys);
 222         vector<docstring>::const_iterator it = keys.begin();
 223         vector<docstring>::const_iterator en = keys.end();
 224         for (; it != en; ++it) {
 225                 if (*it == ukey) {
 226                         LYXERR0("Key " << key << " already present.");
 227                         return false;
 228                 }
 229         }
 230         keys.push_back(ukey);
 231         setParam("key", getStringFromVector(keys));
 232         cache.recalculate = true;
 233         return true;
 234 }
 235
 236
 237 docstring InsetCitation::toolTip(BufferView const & bv, int, int) const
 238 {
 239         Buffer const & buf = bv.buffer();
 240         // Only after the buffer is loaded from file...
 241         if (!buf.isFullyLoaded())
 242                 return docstring();
 243
 244         BiblioInfo const & bi = buf.masterBibInfo();
 245         if (bi.empty())
 246                 return _("No bibliography defined!");
 247
 248         docstring const & key = getParam("key");
 249         if (key.empty())
 250                 return _("No citations selected!");
 251
 252         CiteItem ci;
 253         ci.richtext = true;
 254         vector<docstring> keys = getVectorFromString(key);
 255         if (keys.size() == 1)
 256                 return bi.getInfo(keys[0], buffer(), ci);
 257
 258         docstring tip;
 259         tip += "<ol>";
 260         int count = 0;
 261         for (docstring const & kvar : keys) {
 262                 docstring const key_info = bi.getInfo(kvar, buffer(), ci);
 263                 // limit to reasonable size.
 264                 if (count > 9 && keys.size() > 11) {
 265                         tip.push_back(0x2026);// HORIZONTAL ELLIPSIS
 266                         tip += "<p>"
 267                                 + bformat(_("+ %1$d more entries."), int(keys.size() - count))
 268                                 + "</p>";
 269                         break;
 270                 }
 271                 if (key_info.empty())
 272                         continue;
 273                 tip += "<li>" + key_info + "</li>";
 274                 ++count;
 275         }
 276         tip += "</ol>";
 277         return tip;
 278 }
 279
 280
 281 namespace {
 282
 283 CitationStyle asValidLatexCommand(BufferParams const & bp, string const & input,
 284                                   vector<CitationStyle> const & valid_styles)
 285 {
 286         CitationStyle cs = valid_styles[0];
 287         cs.forceUpperCase = false;
 288         cs.hasStarredVersion = false;
 289
 290         string normalized_input = input;
 291         string::size_type const n = input.size() - 1;
 292         if (isUpperCase(input[0]))
 293                 normalized_input[0] = lowercase(input[0]);
 294         if (input[n] == '*')
 295                 normalized_input = normalized_input.substr(0, n);
 296
 297         string const alias = bp.getCiteAlias(normalized_input);
 298         if (!alias.empty())
 299                 normalized_input = alias;
 300
 301         vector<CitationStyle>::const_iterator it  = valid_styles.begin();
 302         vector<CitationStyle>::const_iterator end = valid_styles.end();
 303         for (; it != end; ++it) {
 304                 CitationStyle this_cs = *it;
 305                 if (this_cs.name == normalized_input) {
 306                         cs = *it;
 307                         break;
 308                 }
 309         }
 310
 311         cs.forceUpperCase &= input[0] == uppercase(input[0]);
 312         cs.hasStarredVersion &= input[n] == '*';
 313
 314         return cs;
 315 }
 316
 317
 318 inline docstring wrapCitation(docstring const & key,
 319                 docstring const & content, bool for_xhtml)
 320 {
 321         if (!for_xhtml)
 322                 return content;
 323         // we have to do the escaping here, because we will ultimately
 324         // write this as a raw string, so as not to escape the tags.
 325         return "<a href='#LyXCite-" + html::cleanAttr(key) + "'>" +
 326                         html::htmlize(content, XHTMLStream::ESCAPE_ALL) + "</a>";
 327 }
 328
 329 } // anonymous namespace
 330
 331
 332 vector<pair<docstring, docstring>> InsetCitation::getQualifiedLists(docstring const p) const
 333 {
 334         vector<docstring> ps =
 335                 getVectorFromString(p, from_ascii("\t"));
 336         QualifiedList res;
 337         for (docstring const & s: ps) {
 338                 docstring key = s;
 339                 docstring val;
 340                 if (contains(s, ' '))
 341                         val = split(s, key, ' ');
 342                 res.push_back(make_pair(key, val));
 343         }
 344         return res;
 345 }
 346
 347 docstring InsetCitation::generateLabel(bool for_xhtml) const
 348 {
 349         docstring label;
 350         label = complexLabel(for_xhtml);
 351
 352         // Fallback to fail-safe
 353         if (label.empty())
 354                 label = basicLabel(for_xhtml);
 355
 356         return label;
 357 }
 358
 359
 360 docstring InsetCitation::complexLabel(bool for_xhtml) const
 361 {
 362         Buffer const & buf = buffer();
 363         // Only start the process off after the buffer is loaded from file.
 364         if (!buf.isFullyLoaded())
 365                 return docstring();
 366
 367         BiblioInfo const & biblist = buf.masterBibInfo();
 368         if (biblist.empty())
 369                 return docstring();
 370
 371         docstring const & key = getParam("key");
 372         if (key.empty())
 373                 return _("No citations selected!");
 374
 375         string cite_type = getCmdName();
 376         bool const uppercase = isUpperCase(cite_type[0]);
 377         if (uppercase)
 378                 cite_type[0] = lowercase(cite_type[0]);
 379         bool const starred = (cite_type[cite_type.size() - 1] == '*');
 380         if (starred)
 381                 cite_type = cite_type.substr(0, cite_type.size() - 1);
 382
 383         // handle alias
 384         string const alias = buf.masterParams().getCiteAlias(cite_type);
 385         if (!alias.empty())
 386                 cite_type = alias;
 387
 388         // FIXME: allow to add cite macros
 389         /*
 390         buffer().params().documentClass().addCiteMacro("!textbefore", to_utf8(before));
 391         buffer().params().documentClass().addCiteMacro("!textafter", to_utf8(after));
 392         */
 393         docstring label;
 394         // we only really want the last 'false', to suppress trimming, but
 395         // we need to give the other defaults, too, to set it.
 396         vector<docstring> keys =
 397                 getVectorFromString(key, from_ascii(","), false, false);
 398         CitationStyle cs = getCitationStyle(buffer().masterParams(),
 399                         cite_type, buffer().masterParams().citeStyles());
 400         bool const qualified = cs.hasQualifiedList
 401                 && (keys.size() > 1
 402                     || !getParam("pretextlist").empty()
 403                     || !getParam("posttextlist").empty());
 404         QualifiedList pres = getQualifiedLists(getParam("pretextlist"));
 405         QualifiedList posts = getQualifiedLists(getParam("posttextlist"));
 406
 407         CiteItem ci;
 408         ci.textBefore = getParam("before");
 409         ci.textAfter = getParam("after");
 410         ci.forceUpperCase = uppercase;
 411         ci.Starred = starred;
 412         ci.max_size = UINT_MAX;
 413         ci.isQualified = qualified;
 414         ci.pretexts = pres;
 415         ci.posttexts = posts;
 416         if (for_xhtml) {
 417                 ci.max_key_size = UINT_MAX;
 418                 ci.context = CiteItem::Export;
 419         }
 420         ci.richtext = for_xhtml;
 421         label = biblist.getLabel(keys, buffer(), cite_type, ci);
 422         return label;
 423 }
 424
 425
 426 docstring InsetCitation::basicLabel(bool for_xhtml) const
 427 {
 428         docstring keys = getParam("key");
 429         docstring label;
 430
 431         docstring key;
 432         do {
 433                 // if there is no comma, then everything goes into key
 434                 // and keys will be empty.
 435                 keys = split(keys, key, ',');
 436                 if (!label.empty())
 437                         label += ", ";
 438                 label += wrapCitation(key, key, for_xhtml);
 439         } while (!keys.empty());
 440
 441         docstring const & after = getParam("after");
 442         if (!after.empty())
 443                 label += ", " + after;
 444
 445         return '[' + label + ']';
 446 }
 447
 448 docstring InsetCitation::screenLabel() const
 449 {
 450         return cache.screen_label;
 451 }
 452
 453
 454 void InsetCitation::updateBuffer(ParIterator const &, UpdateType)
 455 {
 456         if (!cache.recalculate && buffer().citeLabelsValid())
 457                 return;
 458         // The label may have changed, so we have to re-create it.
 459         docstring const glabel = generateLabel();
 460         cache.recalculate = false;
 461         cache.generated_label = glabel;
 462         unsigned int const maxLabelChars = 45;
 463         cache.screen_label = glabel;
 464         support::truncateWithEllipsis(cache.screen_label, maxLabelChars, true);
 465 }
 466
 467
 468 void InsetCitation::addToToc(DocIterator const & cpit, bool output_active,
 469                                                          UpdateType, TocBackend & backend) const
 470 {
 471         // NOTE
 472         // BiblioInfo::collectCitedEntries() uses the TOC to collect the citations
 473         // from the document. It is used indirectly, via BiblioInfo::makeCitationLables,
 474         // by both XHTML and plaintext output. So, if we change what goes into the TOC,
 475         // then we will also need to change that routine.
 476         docstring const tocitem = getParam("key");
 477         TocBuilder & b = backend.builder("citation");
 478         b.pushItem(cpit, tocitem, output_active);
 479         b.pop();
 480 }
 481
 482
 483 int InsetCitation::plaintext(odocstringstream & os,
 484        OutputParams const &, size_t) const
 485 {
 486         string const & cmd = getCmdName();
 487         if (cmd == "nocite")
 488                 return 0;
 489
 490         docstring const label = generateLabel(false);
 491         os << label;
 492         return label.size();
 493 }
 494
 495
 496 static docstring const cleanupWhitespace(docstring const & citelist)
 497 {
 498         docstring::const_iterator it  = citelist.begin();
 499         docstring::const_iterator end = citelist.end();
 500         // Paranoia check: make sure that there is no whitespace in here
 501         // -- at least not behind commas or at the beginning
 502         docstring result;
 503         char_type last = ',';
 504         for (; it != end; ++it) {
 505                 if (*it != ' ')
 506                         last = *it;
 507                 if (*it != ' ' || last != ',')
 508                         result += *it;
 509         }
 510         return result;
 511 }
 512
 513
 514 int InsetCitation::docbook(odocstream & os, OutputParams const &) const
 515 {
 516         os << from_ascii("<citation>")
 517            << cleanupWhitespace(getParam("key"))
 518            << from_ascii("</citation>");
 519         return 0;
 520 }
 521
 522
 523 docstring InsetCitation::xhtml(XHTMLStream & xs, OutputParams const &) const
 524 {
 525         string const & cmd = getCmdName();
 526         if (cmd == "nocite")
 527                 return docstring();
 528
 529         // have to output this raw, because generateLabel() will include tags
 530         xs << XHTMLStream::ESCAPE_NONE << generateLabel(true);
 531
 532         return docstring();
 533 }
 534
 535
 536 void InsetCitation::toString(odocstream & os) const
 537 {
 538         odocstringstream ods;
 539         plaintext(ods, OutputParams(0));
 540         os << ods.str();
 541 }
 542
 543
 544 void InsetCitation::forOutliner(docstring & os, size_t const, bool const) const
 545 {
 546         os += screenLabel();
 547 }
 548
 549
 550 // Have to overwrite the default InsetCommand method in order to check that
 551 // the \cite command is valid. Eg, the user has natbib enabled, inputs some
 552 // citations and then changes his mind, turning natbib support off. The output
 553 // should revert to the default citation command as provided by the citation
 554 // engine, e.g. \cite[]{} for the basic engine.
 555 void InsetCitation::latex(otexstream & os, OutputParams const & runparams) const
 556 {
 557         // When this is a child compiled on its own, we use the childs
 558         // own bibinfo, else the master's
 559         BiblioInfo const & bi = runparams.is_child
 560                         ? buffer().masterBibInfo() : buffer().bibInfo();
 561         docstring const key = getParam("key");
 562         // "keyonly" command: output the plain key and stop.
 563         if (getCmdName() == "keyonly") {
 564                 // Special command to only return the key
 565                 if (!bi.isBibtex(getParam("key")))
 566                         // escape chars with bibitems
 567                         os << escape(cleanupWhitespace(key));
 568                 else
 569                         os << cleanupWhitespace(key);
 570                 return;
 571         }
 572         vector<CitationStyle> citation_styles = buffer().masterParams().citeStyles();
 573         CitationStyle cs = asValidLatexCommand(buffer().masterParams(),
 574                                                getCmdName(), citation_styles);
 575         // FIXME UNICODE
 576         docstring const cite_str = from_utf8(citationStyleToString(cs, true));
 577
 578         // check if we have to do a qualified list
 579         vector<docstring> keys = getVectorFromString(cleanupWhitespace(key));
 580         bool const qualified = cs.hasQualifiedList
 581                 && (!getParam("pretextlist").empty()
 582                     || !getParam("posttextlist").empty());
 583
 584         if (runparams.inulemcmd > 0)
 585                 os << "\\mbox{";
 586
 587         os << "\\" << cite_str;
 588
 589         if (qualified)
 590                 os << "s";
 591
 592         ParamInfo const & pinfo = findInfo(string());
 593         docstring before = params().prepareCommand(runparams, getParam("before"),
 594                                                    pinfo["before"].handling());
 595         docstring after = params().prepareCommand(runparams, getParam("after"),
 596                                                    pinfo["after"].handling());
 597         if (!before.empty() && cs.textBefore) {
 598                 if (qualified)
 599                         os << '(' << protectArgument(before, '(', ')')
 600                            << ")(" << protectArgument(after, '(', ')') << ')';
 601                 else
 602                         os << '[' << protectArgument(before) << "]["
 603                            << protectArgument(after) << ']';
 604         } else if (!after.empty() && cs.textAfter) {
 605                 if (qualified)
 606                         os << '(' << protectArgument(after, '(', ')') << ')';
 607                 else
 608                         os << '[' << protectArgument(after) << ']';
 609         }
 610
 611         if (!bi.isBibtex(key))
 612                 // escape chars with bibitems
 613                 os << '{' << escape(cleanupWhitespace(key)) << '}';
 614         else {
 615                 if (qualified) {
 616                         QualifiedList pres = getQualifiedLists(getParam("pretextlist"));
 617                         QualifiedList posts = getQualifiedLists(getParam("posttextlist"));
 618                         for (docstring const & k : keys) {
 619                                 docstring prenote;
 620                                 QualifiedList::iterator it = pres.begin();
 621                                 for (; it != pres.end() ; ++it) {
 622                                         if ((*it).first == k) {
 623                                                 prenote = (*it).second;
 624                                                 pres.erase(it);
 625                                                 break;
 626                                         }
 627                                 }
 628                                 docstring bef = params().prepareCommand(runparams, prenote,
 629                                                    pinfo["pretextlist"].handling());
 630                                 docstring postnote;
 631                                 QualifiedList::iterator pit = posts.begin();
 632                                 for (; pit != posts.end() ; ++pit) {
 633                                         if ((*pit).first == k) {
 634                                                 postnote = (*pit).second;
 635                                                 posts.erase(pit);
 636                                                 break;
 637                                         }
 638                                 }
 639                                 docstring aft = params().prepareCommand(runparams, postnote,
 640                                                    pinfo["posttextlist"].handling());
 641                                 if (!bef.empty())
 642                                         os << '[' << protectArgument(bef)
 643                                            << "][" << protectArgument(aft) << ']';
 644                                 else if (!aft.empty())
 645                                         os << '[' << protectArgument(aft) << ']';
 646                                 os << '{' << k << '}';
 647                         }
 648                 } else
 649                         os << '{' << cleanupWhitespace(key) << '}';
 650         }
 651
 652         if (runparams.inulemcmd)
 653                 os << "}";
 654 }
 655
 656
 657 string InsetCitation::contextMenuName() const
 658 {
 659         return "context-citation";
 660 }
 661
 662
 663 } // namespace lyx