[CB] Экспорт стилей Stylish | Форум Mozilla Россия

Создать кнопку или положить в инициализацию другой
скрытый текст
Выделить код
Код:

//Экспорт стилей Stylish в меню кнопки .................................................................................................................
function Exporter(toFileSet)
{
  Object.defineProperty(this, "toFileSet", {get: function(){ return toFileSet; }});
  var converter = Cc["@mozilla.org/intl/scriptableunicodeconverter"].createInstance(Ci.nsIScriptableUnicodeConverter);
  converter.charset = "UTF-8";
  Object.defineProperty(this, "converter", {get: function(){ return converter; }});
};

Exporter.prototype=
{
  styles: null,
  mDBConn: null,
  exportFile: null,
  
  toString: function() { return "[custombuttons.stylish-exporter]"; },
  
  handleEvent: function(event)
  {
    this.exportFile = this.toFileSet ? this.pickDirectory() : this.pickFile();
    if (!this.exportFile) return;
    this.getStyles();
  },
  
  pickFile: function()
  {
    var fp=Cc["@mozilla.org/filepicker;1"].createInstance(Ci.nsIFilePicker);
    fp.init(window, "Укажите файл экспорта", Ci.nsIFilePicker.modeSave);
    fp.appendFilter("Каскадные таблицы стилей (css)","*.css");
    var res = fp.show();
    if (res == Ci.nsIFilePicker.returnCancel)
        return null;
    var cssFile = fp.file;
    if (!/\.css$/i.test(cssFile.path)) cssFile.initWithPath(cssFile.path+".css");
    return cssFile;
  },
  
  pickDirectory: function()
  {
    var fp=Cc["@mozilla.org/filepicker;1"].createInstance(Ci.nsIFilePicker);
    fp.init(window, "Укажите папку для экспорта", Ci.nsIFilePicker.modeGetFolder);
    return fp.show()==Ci.nsIFilePicker.returnOK ? fp.file : null;
  },
  
  getStyles: function()
  {
    var dbFile = Services.dirsvc.get("ProfD", Ci.nsIFile);
    dbFile.append("stylish.sqlite");
    var ssvc=Cc["@mozilla.org/storage/service;1"].getService(Ci.mozIStorageService);
    this.mDBConn = ssvc.openDatabase(dbFile);
    var stmt = this.mDBConn.createStatement("SELECT name, code from styles");
    this.styles=[];
    stmt.executeAsync(this);
  },

  prepareStyles: function()
  {
    var data;
    if (!this.toFileSet)
    {
      var content = [];
      this.styles.forEach(function(elem)
      {
        content = content.concat(["/*", elem.name,"*/", elem.style, ""]);
      });
      data=[{file: this.exportFile, content: content.join("\n")}];
    }
    else
    {
      data=this.styles.map((function(elem)
      {
        var file = this.exportFile.clone();
        file.append(elem.name.replace(/[\\\/\:\*\?"<>\|]/g,"-").replace(/^\.|\.$/g,"_")+".css");
        return {file: file, content: ["/*", elem.name,"*/", elem.style].join("\n")};
      }).bind(this));
    }
    this.saveStyles(data);
  },
  
  saveStyles: function(data)
  {
    if (data.length==0)
    {
      custombuttons.alertSlide("Экспорт стилей Stylish", "Стили успешно экспортированы.");
      return;
    }
    Cu.import("resource://gre/modules/FileUtils.jsm");
    Cu.import("resource://gre/modules/NetUtil.jsm");
    var style = data.shift();
    var ostream = FileUtils.openSafeFileOutputStream(style.file)
    var istream = this.converter.convertToInputStream(style.content);
    NetUtil.asyncCopy(istream, ostream, (function(status)
    {
      if (!Components.isSuccessCode(status))
        Services.prompt.alert(null, "Ошибка","Ошибка сохранения стилей (код: "+status+")");
      else
        this.saveStyles(data);
    }).bind(this));
  },
  
  handleResult: function(aResultSet) 
  {
    for (let row = aResultSet.getNextRow(); row; row = aResultSet.getNextRow())
      this.styles.push({name: row.getResultByName("name"), style: row.getResultByName("code")});
  },
  
  handleError: function(aError) 
  {
    Services.prompt.alert(null, "Ошибка выполнения запроса", aError.message);
  },
  
  handleCompletion: function(aReason) 
  {
    this.mDBConn.asyncClose((function()
    {
      delete this.mDBConn;
    }).bind(this));
    this.prepareStyles();
  }
};

(function()
{
  if (document.getElementById("stylish-export-styles-to-file")) return;
  var miToSingleFile = document.createElement("menuitem");
  miToSingleFile.id = "stylish-export-styles-to-file";
  miToSingleFile.setAttribute("label", "В файл");
  var miToSetFiles = document.createElement("menuitem");
  miToSetFiles.id = "stylish-export-styles-to-set-files";
  miToSetFiles.setAttribute("label", "В набор файлов");
  var popup = document.createElement("menupopup");
  popup.appendChild(miToSingleFile);
  popup.appendChild(miToSetFiles);
  var menu = document.createElement("menu");
  menu.id="stylish-export-styles";
  menu.setAttribute("label","Экспортировать стили");
  menu.appendChild(popup);
  let sn = document.evaluate("//xul:popupset[@id='mainPopupSet']/descendant::xul:menuitem[@id='stylish-manage']/following-sibling::node()[position()=1]",
    document, function() { return "http://www.mozilla.org/keymaster/gatekeeper/there.is.only.xul"; },
    Ci.nsIDOMXPathResult.ANY_UNORDERED_NODE_TYPE, null);
  sn.singleNodeValue.parentNode.insertBefore(menu, sn.singleNodeValue);
  miToSingleFile.addEventListener("click", new Exporter(false), false);
  miToSetFiles.addEventListener("click", new Exporter(true), false);
})();