]> git.lyx.org Git - lyx.git/blobdiff - src/EmbeddedFiles.cpp
Embedding: prepare to read/write manifest in .lyx file
[lyx.git] / src / EmbeddedFiles.cpp
index e67d1511413e968bc56f925ffa7cf419efa27766..155ed86ac787c1b57f58d669b59c9bc71f0bf482 100644 (file)
 #include "Buffer.h"
 #include "BufferParams.h"
 #include "Paragraph.h"
-#include "ParIterator.h"
+#include "InsetIterator.h"
 #include "debug.h"
 #include "gettext.h"
 #include "Format.h"
+#include "Lexer.h"
+#include "ErrorList.h"
 
 #include "frontends/alert.h"
 
@@ -31,6 +33,9 @@
 #include "support/lyxlib.h"
 #include "support/lstrings.h"
 
+#include "LyX.h"
+#include "Session.h"
+
 #include <sstream>
 #include <fstream>
 #include <utility>
@@ -67,10 +72,13 @@ using support::makedir;
 
 
 EmbeddedFile::EmbeddedFile(string const & file, string const & inzip_name,
-       bool embed, ParConstIterator const & pit)
+       bool embed, Inset const * inset)
        : DocFileName(file, true), inzip_name_(inzip_name), embedded_(embed),
-               valid_(true), par_it_(pit)
-{}
+               valid_(true), inset_list_()
+{
+       if (inset != NULL)
+               inset_list_.push_back(inset);
+}
 
 
 string EmbeddedFile::embeddedFile(Buffer const * buf) const
@@ -79,98 +87,141 @@ string EmbeddedFile::embeddedFile(Buffer const * buf) const
 }
 
 
-int const EmbeddedFile::parID() const
+void EmbeddedFile::addInset(Inset const * inset)
 {
+       inset_list_.push_back(inset);
+}
+
+
+Inset const * EmbeddedFile::inset(int idx) const
+{
+       BOOST_ASSERT(idx < refCount());
        // some embedded file do not have a valid par iterator
-       return par_it_ == ParConstIterator() ? 0 : par_it_->id();
+       return inset_list_[idx];
 }
 
 
-void EmbeddedFile::setParIter(ParConstIterator const & pit)
+void EmbeddedFile::saveBookmark(Buffer const * buf, int idx) const
 {
-       par_it_ = pit;
+       Inset const * ptr = inset(idx);
+       // This might not be the most efficient method ... 
+       for (InsetIterator it = inset_iterator_begin(buf->inset()); it; ++it)
+               if (&(*it) == ptr) {
+                       // this is basically BufferView::saveBookmark(0)
+                       LyX::ref().session().bookmarks().save(
+                               FileName(buf->fileName()),
+                               it.bottom().pit(),
+                               it.bottom().pos(),
+                               it.paragraph().id(),
+                               it.pos(),
+                               0
+                       );
+               }
+       // this inset can not be located. There is something wrong that needs
+       // to be fixed.
+       BOOST_ASSERT(true);
 }
 
 
 string EmbeddedFile::availableFile(Buffer const * buf) const
 {
-       if (embedded_)
+       if (embedded())
                return embeddedFile(buf);
        else
                return absFilename();
 }
 
 
+void EmbeddedFile::invalidate()
+{
+       // Clear inset_list_ because they will be registered again.
+       inset_list_.clear();
+       valid_ = false;
+}
+
+
 bool EmbeddedFile::extract(Buffer const * buf) const
 {
-       if (!embedded_)
-               return true;
 
        string ext_file = absFilename();
        string emb_file = embeddedFile(buf);
-       bool copyFile = false;
-       // both files exist, are different
-       if (fs::exists(ext_file) && fs::exists(emb_file)
-               && sum(*this) != sum(FileName(emb_file))) {
-               int const ret = Alert::prompt(
+
+       if (!fs::exists(emb_file))
+               return false;
+
+       // if external file already exists ...
+       if (fs::exists(ext_file)) {
+               // no need to copy if the files are the same
+               if (sum(*this) == sum(FileName(emb_file)))
+                       return true;
+               // otherwise, ask if overwrite
+               int ret = Alert::prompt(
                        _("Overwrite external file?"),
                        bformat(_("External file %1$s already exists, do you want to overwrite it"),
                                from_utf8(ext_file)), 1, 1, _("&Overwrite"), _("&Cancel"));
-               copyFile = ret == 0;
-       }
-       // copy file in the previous case, and a new case
-       if (copyFile || (!fs::exists(ext_file) && fs::exists(emb_file))) {
-               try {
-                       // need to make directory?
-                       string path = onlyPath(ext_file);
-                       if (!fs::is_directory(path))
-                               makedir(const_cast<char*>(path.c_str()), 0755);
-                       fs::copy_file(emb_file, ext_file, false);
+               if (ret != 0)
+                       // if the user does not want to overwrite, we still consider it
+                       // a successful operation.
                        return true;
-               } catch (fs::filesystem_error const & fe) {
-                       Alert::error(_("Copy file failure"),
-                                bformat(_("Cannot copy file %1$s to %2$s.\n"
-                                          "Please check whether the directory exists and is writeable."),
-                                               from_utf8(emb_file), from_utf8(ext_file)));
-                       LYXERR(Debug::DEBUG) << "Fs error: " << fe.what() << endl;
-               }
+       }
+       // copy file
+       try {
+               // need to make directory?
+               string path = onlyPath(ext_file);
+               if (!fs::is_directory(path))
+                       makedir(const_cast<char*>(path.c_str()), 0755);
+               fs::copy_file(emb_file, ext_file, false);
+               return true;
+       } catch (fs::filesystem_error const & fe) {
+               Alert::error(_("Copy file failure"),
+                        bformat(_("Cannot copy file %1$s to %2$s.\n"
+                                  "Please check whether the directory exists and is writeable."),
+                                       from_utf8(emb_file), from_utf8(ext_file)));
+               LYXERR(Debug::DEBUG) << "Fs error: " << fe.what() << endl;
        }
        return false;
 }
 
 
-bool EmbeddedFile::update(Buffer const * buf) const
+bool EmbeddedFile::updateFromExternalFile(Buffer const * buf) const
 {
        string ext_file = absFilename();
        string emb_file = embeddedFile(buf);
-       bool copyFile = false;
-       // both files exist, are different
-       if (fs::exists(ext_file) && fs::exists(emb_file)
-               && sum(*this) != sum(FileName(emb_file))) {
+
+       if (!fs::exists(ext_file))
+               return false;
+       
+       // if embedded file already exists ...
+       if (fs::exists(emb_file)) {
+               // no need to copy if the files are the same
+               if (sum(*this) == sum(FileName(emb_file)))
+                       return true;
+               // other wise, ask if overwrite
                int const ret = Alert::prompt(
                        _("Update embedded file?"),
                        bformat(_("Embeddedl file %1$s already exists, do you want to overwrite it"),
                                from_utf8(ext_file)), 1, 1, _("&Overwrite"), _("&Cancel"));
-               copyFile = ret == 0;
+               if (ret != 0)
+                       // if the user does not want to overwrite, we still consider it
+                       // a successful operation.
+                       return true;
        }
-       // copy file in the previous case, and a new case
-       if (copyFile || (fs::exists(ext_file) && !fs::exists(emb_file))) {
-               try {
-                       // need to make directory?
-                       string path = onlyPath(emb_file);
-                       if (!fs::is_directory(path))
-                               makedir(const_cast<char*>(path.c_str()), 0755);
-                       fs::copy_file(ext_file, emb_file, false);
-               } catch (fs::filesystem_error const & fe) {
-                       Alert::error(_("Copy file failure"),
-                                bformat(_("Cannot copy file %1$s to %2$s.\n"
-                                          "Please check whether the directory exists and is writeable."),
-                                               from_utf8(ext_file), from_utf8(emb_file)));
-                       LYXERR(Debug::DEBUG) << "Fs error: " << fe.what() << endl;
-                       return false;
-               }
+       // copy file
+       try {
+               // need to make directory?
+               string path = onlyPath(emb_file);
+               if (!fs::is_directory(path))
+                       makedir(const_cast<char*>(path.c_str()), 0755);
+               fs::copy_file(ext_file, emb_file, false);
+               return true;
+       } catch (fs::filesystem_error const & fe) {
+               Alert::error(_("Copy file failure"),
+                        bformat(_("Cannot copy file %1$s to %2$s.\n"
+                                  "Please check whether the directory exists and is writeable."),
+                                       from_utf8(ext_file), from_utf8(emb_file)));
+               LYXERR(Debug::DEBUG) << "Fs error: " << fe.what() << endl;
        }
-       return true;
+       return false;
 }
 
 
@@ -180,22 +231,24 @@ bool EmbeddedFiles::enabled() const
 }
 
 
-void EmbeddedFiles::enable(bool flag)
+bool EmbeddedFiles::enable(bool flag)
 {
        if (enabled() != flag) {
-               // if disable embedding, first extract all embedded files
-               if (flag || (!flag && extractAll())) {
-                       // file will be changed
-                       buffer_->markDirty();
-                       buffer_->params().embedded = flag;
-               }
+               // if enable, copy all files to temppath()
+               // if disable, extract all files
+               if ((flag && !updateFromExternalFile()) || (!flag && !extract()))
+                       return false;
+               // if operation is successful
+               buffer_->markDirty();
+               buffer_->params().embedded = flag;
        }
 }
 
 
 void EmbeddedFiles::registerFile(string const & filename,
-       bool embed, ParConstIterator const & pit)
+       bool embed, Inset const * inset, string const & inzipName)
 {
+       // filename can be relative or absolute, translate to absolute filename
        string abs_filename = makeAbsPath(filename, buffer_->filePath()).absFilename();
        // try to find this file from the list
        EmbeddedFileList::iterator it = file_list_.begin();
@@ -203,16 +256,15 @@ void EmbeddedFiles::registerFile(string const & filename,
        for (; it != it_end; ++it)
                if (it->absFilename() == abs_filename || it->embeddedFile(buffer_) == abs_filename)
                        break;
-       // find this filename
+       // find this filename, keep the original embedding status
        if (it != file_list_.end()) {
-               it->setParIter(pit);
-               it->update(buffer_);
-               it->setEmbed(true);
+               it->addInset(inset);
                it->validate();
                return;
        }
+       // try to be more careful
        file_list_.push_back(EmbeddedFile(abs_filename, 
-               getInzipName(abs_filename), embed, pit));
+               getInzipName(abs_filename, inzipName), embed, inset));
 }
 
 
@@ -225,22 +277,12 @@ void EmbeddedFiles::update()
        EmbeddedFileList::iterator it = file_list_.begin();
        EmbeddedFileList::iterator it_end = file_list_.end();
        for (; it != it_end; ++it)
-               it->invalidate();
-
-       ParIterator pit = buffer_->par_iterator_begin();
-       ParIterator pit_end = buffer_->par_iterator_end();
-       for (; pit != pit_end; ++pit) {
-               // For each paragraph, traverse its insets and register embedded files
-               InsetList::const_iterator iit = pit->insetlist.begin();
-               InsetList::const_iterator iit_end = pit->insetlist.end();
-               for (; iit != iit_end; ++iit) {
-                       Inset & inset = *iit->inset;
-                       inset.registerEmbeddedFiles(*buffer_, *this, pit);
-               }
-       }
-       LYXERR(Debug::FILES) << "Manifest updated: " << endl
-               << *this
-               << "End Manifest" << endl;
+               // we do not update items that are manually inserted
+               if (it->refCount() > 0)
+                       it->invalidate();
+
+       for (InsetIterator it = inset_iterator_begin(buffer_->inset()); it; ++it)
+               it->registerEmbeddedFiles(*buffer_, *this);
 }
 
 
@@ -248,17 +290,11 @@ bool EmbeddedFiles::write(DocFileName const & filename)
 {
        // file in the temporary path has the content
        string const content = FileName(addName(buffer_->temppath(),
-               onlyFilename(filename.toFilesystemEncoding()))).toFilesystemEncoding();
+               "content.lyx")).toFilesystemEncoding();
 
-       // get a file list and write a manifest file
        vector<pair<string, string> > filenames;
-       string const manifest = FileName(
-               addName(buffer_->temppath(), "manifest.txt")).toFilesystemEncoding();
-
-       // write a manifest file
-       ofstream os(manifest.c_str());
-       os << *this;
-       os.close();
+       // add content.lyx to filenames
+       filenames.push_back(make_pair(content, "content.lyx"));
        // prepare list of embedded file
        EmbeddedFileList::iterator it = file_list_.begin();
        EmbeddedFileList::iterator it_end = file_list_.end();
@@ -271,9 +307,6 @@ bool EmbeddedFiles::write(DocFileName const & filename)
                                filenames.push_back(make_pair(file, it->inzipName()));
                }
        }
-       // add filename (.lyx) and manifest to filenames
-       filenames.push_back(make_pair(content, onlyFilename(filename.toFilesystemEncoding())));
-       filenames.push_back(make_pair(manifest, "manifest.txt"));
        // write a zip file with all these files. Write to a temp file first, to
        // avoid messing up the original file in case something goes terribly wrong.
        DocFileName zipfile(addName(buffer_->temppath(),
@@ -306,25 +339,37 @@ EmbeddedFiles::EmbeddedFileList::const_iterator EmbeddedFiles::find(std::string
 }
 
 
-bool EmbeddedFiles::extractAll() const
+bool EmbeddedFiles::extract() const
+{
+       EmbeddedFileList::const_iterator it = file_list_.begin();
+       EmbeddedFileList::const_iterator it_end = file_list_.end();
+       for (; it != it_end; ++it)
+               if (it->valid() && it->embedded())
+                       if(!it->extract(buffer_))
+                               return false;
+       return true;
+}
+
+
+bool EmbeddedFiles::updateFromExternalFile() const
 {
        EmbeddedFileList::const_iterator it = file_list_.begin();
        EmbeddedFileList::const_iterator it_end = file_list_.end();
-       // FIXME: the logic here is hard to decide, we should allow cancel for
-       // 'do not overwrite' this file, and cancel for 'cancel extract all files'.
-       // I am not sure how to do this now.
        for (; it != it_end; ++it)
                if (it->valid() && it->embedded())
-                       it->extract(buffer_);
+                       if (!it->updateFromExternalFile(buffer_))
+                               return false;
        return true;
 }
 
 
-string const EmbeddedFiles::getInzipName(string const & abs_filename)
+string const EmbeddedFiles::getInzipName(string const & abs_filename, string const & name)
 {
        // register a new one, using relative file path as inzip_name
-       string inzip_name = to_utf8(makeRelPath(from_utf8(abs_filename),
-               from_utf8(buffer_->fileName())));
+       string inzip_name = name;
+       if (name.empty())
+               inzip_name = to_utf8(makeRelPath(from_utf8(abs_filename),
+                       from_utf8(buffer_->filePath())));
        // if inzip_name is an absolute path, use filename only to avoid
        // leaking of filesystem information in inzip_name
        // The second case covers cases '../path/file' and '.'
@@ -353,81 +398,74 @@ string const EmbeddedFiles::getInzipName(string const & abs_filename)
 }
 
 
-istream & operator>> (istream & is, EmbeddedFiles & files)
+bool EmbeddedFiles::readManifest(Lexer & lex, ErrorList & errorList)
 {
-       files.clear();
-       string tmp;
-       getline(is, tmp);
-       // get version
-       istringstream itmp(tmp);
-       int version;
-       itmp.ignore(string("# LyX manifest version ").size());
-       itmp >> version;
-
-       if (version != 1) {
-               lyxerr << "This version of LyX can only read LyX manifest version 1" << endl;
-               return is;
-       }
+       int line = -1;
+       int begin_manifest_line = -1;
 
-       getline(is, tmp);
-       if (tmp != "<manifest>") {
-               lyxerr << "Invalid manifest file, lacking <manifest>" << endl;
-               return is;
-       }
-       // manifest file may be messed up, be carefully
-       while (is.good()) {
-               getline(is, tmp);
-               if (tmp != "<file>")
-                       break;
+       file_list_.clear();
+       string filename = "";
+       string inzipName = "";
+       bool status = "";
+
+       while (lex.isOK()) {
+               lex.next();
+               string const token = lex.getString();
+
+               if (token.empty())
+                       continue;
 
-               string fname;
-               getline(is, fname);
-               string inzip_name;
-               getline(is, inzip_name);
-               getline(is, tmp);
-               istringstream itmp(tmp);
-               int embed;
-               itmp >> embed;
-
-               getline(is, tmp);
-               if (tmp != "</file>") {
-                       lyxerr << "Invalid manifest file, lacking </file>" << endl;
+               if (token == "\\end_manifest")
                        break;
-               }
 
-               files.registerFile(fname, embed);
-       };
-       // the last line must be </manifest>
-       if (tmp != "</manifest>") {
-               lyxerr << "Invalid manifest file, lacking </manifest>" << endl;
-               return is;
+               ++line;
+               if (token == "\\begin_manifest") {
+                       begin_manifest_line = line;
+                       continue;
+               }
+               
+               LYXERR(Debug::PARSER) << "Handling document manifest token: `"
+                                     << token << '\'' << endl;
+
+               if (token == "\\filename")
+                       lex >> filename;
+               else if (token == "\\inzipName")
+                       lex >> inzipName;
+               else if (token == "\\status") {
+                       lex >> status;
+                       registerFile(filename, status, NULL, inzipName);
+                       filename = "";
+                       inzipName = "";
+               } else {
+                       docstring const s = _("\\begin_file is missing");
+                       errorList.push_back(ErrorItem(_("Manifest error"),
+                               s, -1, 0, 0));
+               }
+       }
+       if (begin_manifest_line) {
+               docstring const s = _("\\begin_manifest is missing");
+               errorList.push_back(ErrorItem(_("Manifest error"),
+                       s, -1, 0, 0));
        }
-       return is;
+       return true;
 }
 
 
-ostream & operator<< (ostream & os, EmbeddedFiles const & files)
+void EmbeddedFiles::writeManifest(ostream & os) const
 {
-       // store a version so that operator >> can read later versions
-       // using version information.
-       os << "# lyx manifest version 1\n";
-       os << "<manifest>\n";
-       EmbeddedFiles::EmbeddedFileList::const_iterator it = files.begin();
-       EmbeddedFiles::EmbeddedFileList::const_iterator it_end = files.end();
+       EmbeddedFiles::EmbeddedFileList::const_iterator it = begin();
+       EmbeddedFiles::EmbeddedFileList::const_iterator it_end = end();
        for (; it != it_end; ++it) {
                if (!it->valid())
                        continue;
-               // use differnt lines to make reading easier.
-               os << "<file>\n"
-                       // save the relative path
+               // save the relative path
+               os << "\\filename "
                        << to_utf8(makeRelPath(from_utf8(it->absFilename()),
-                               from_utf8(files.buffer_->filePath()))) << '\n'
-                       << it->inzipName() << '\n'
-                       << it->embedded() << '\n'
-                       << "</file>\n";
+                               from_utf8(buffer_->filePath()))) << '\n'
+                       << "\\inzipName " << it->inzipName() << '\n'
+                       << "\\status " << (it->embedded() ? "true" : "false") << '\n';
        }
-       os << "</manifest>\n";
-       return os;
 }
 
+
 }