зеркало из https://github.com/mozilla/gecko-dev.git
361 строка
12 KiB
JavaScript
361 строка
12 KiB
JavaScript
/* -*- indent-tabs-mode: nil; js-indent-level: 2 -*-
|
|
* vim: sw=2 ts=2 sts=2 et */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
"use strict";
|
|
|
|
const Cc = Components.classes;
|
|
const Ci = Components.interfaces;
|
|
const Cu = Components.utils;
|
|
const Cr = Components.results;
|
|
|
|
const FILE_INPUT_STREAM_CID = "@mozilla.org/network/file-input-stream;1";
|
|
|
|
const S100NS_FROM1601TO1970 = 0x19DB1DED53E8000;
|
|
const S100NS_PER_MS = 10;
|
|
|
|
Cu.import("resource://gre/modules/XPCOMUtils.jsm");
|
|
Cu.import("resource://gre/modules/Services.jsm");
|
|
Cu.import("resource://gre/modules/NetUtil.jsm");
|
|
Cu.import("resource://gre/modules/FileUtils.jsm");
|
|
Cu.import("resource://gre/modules/Task.jsm");
|
|
Cu.import("resource:///modules/MigrationUtils.jsm");
|
|
|
|
XPCOMUtils.defineLazyModuleGetter(this, "PlacesUtils",
|
|
"resource://gre/modules/PlacesUtils.jsm");
|
|
|
|
/**
|
|
* Convert Chrome time format to Date object
|
|
*
|
|
* @param aTime
|
|
* Chrome time
|
|
* @return converted Date object
|
|
* @note Google Chrome uses FILETIME / 10 as time.
|
|
* FILETIME is based on same structure of Windows.
|
|
*/
|
|
function chromeTimeToDate(aTime)
|
|
{
|
|
return new Date((aTime * S100NS_PER_MS - S100NS_FROM1601TO1970 ) / 10000);
|
|
}
|
|
|
|
/**
|
|
* Insert bookmark items into specific folder.
|
|
*
|
|
* @param parentGuid
|
|
* GUID of the folder where items will be inserted
|
|
* @param items
|
|
* bookmark items to be inserted
|
|
*/
|
|
function* insertBookmarkItems(parentGuid, items) {
|
|
for (let item of items) {
|
|
try {
|
|
if (item.type == "url") {
|
|
yield PlacesUtils.bookmarks.insert({
|
|
parentGuid, url: item.url, title: item.name
|
|
});
|
|
} else if (item.type == "folder") {
|
|
let newFolderGuid = (yield PlacesUtils.bookmarks.insert({
|
|
parentGuid, type: PlacesUtils.bookmarks.TYPE_FOLDER, title: item.name
|
|
})).guid;
|
|
|
|
yield insertBookmarkItems(newFolderGuid, item.children);
|
|
}
|
|
} catch (e) {
|
|
Cu.reportError(e);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
function ChromeProfileMigrator() {
|
|
let chromeUserDataFolder = FileUtils.getDir(
|
|
#ifdef XP_WIN
|
|
"LocalAppData", ["Google", "Chrome", "User Data"]
|
|
#elifdef XP_MACOSX
|
|
"ULibDir", ["Application Support", "Google", "Chrome"]
|
|
#else
|
|
"Home", [".config", "google-chrome"]
|
|
#endif
|
|
, false);
|
|
this._chromeUserDataFolder = chromeUserDataFolder.exists() ?
|
|
chromeUserDataFolder : null;
|
|
}
|
|
|
|
ChromeProfileMigrator.prototype = Object.create(MigratorPrototype);
|
|
|
|
ChromeProfileMigrator.prototype.getResources =
|
|
function Chrome_getResources(aProfile) {
|
|
if (this._chromeUserDataFolder) {
|
|
let profileFolder = this._chromeUserDataFolder.clone();
|
|
profileFolder.append(aProfile.id);
|
|
if (profileFolder.exists()) {
|
|
let possibleResources = [GetBookmarksResource(profileFolder),
|
|
GetHistoryResource(profileFolder),
|
|
GetCookiesResource(profileFolder)];
|
|
return [r for each (r in possibleResources) if (r != null)];
|
|
}
|
|
}
|
|
return [];
|
|
};
|
|
|
|
Object.defineProperty(ChromeProfileMigrator.prototype, "sourceProfiles", {
|
|
get: function Chrome_sourceProfiles() {
|
|
if ("__sourceProfiles" in this)
|
|
return this.__sourceProfiles;
|
|
|
|
if (!this._chromeUserDataFolder)
|
|
return [];
|
|
|
|
let profiles = [];
|
|
try {
|
|
// Local State is a JSON file that contains profile info.
|
|
let localState = this._chromeUserDataFolder.clone();
|
|
localState.append("Local State");
|
|
if (!localState.exists())
|
|
throw new Error("Chrome's 'Local State' file does not exist.");
|
|
if (!localState.isReadable())
|
|
throw new Error("Chrome's 'Local State' file could not be read.");
|
|
|
|
let fstream = Cc[FILE_INPUT_STREAM_CID].createInstance(Ci.nsIFileInputStream);
|
|
fstream.init(localState, -1, 0, 0);
|
|
let inputStream = NetUtil.readInputStreamToString(fstream, fstream.available(),
|
|
{ charset: "UTF-8" });
|
|
let info_cache = JSON.parse(inputStream).profile.info_cache;
|
|
for (let profileFolderName in info_cache) {
|
|
let profileFolder = this._chromeUserDataFolder.clone();
|
|
profileFolder.append(profileFolderName);
|
|
profiles.push({
|
|
id: profileFolderName,
|
|
name: info_cache[profileFolderName].name || profileFolderName,
|
|
});
|
|
}
|
|
} catch (e) {
|
|
Cu.reportError("Error detecting Chrome profiles: " + e);
|
|
// If we weren't able to detect any profiles above, fallback to the Default profile.
|
|
let defaultProfileFolder = this._chromeUserDataFolder.clone();
|
|
defaultProfileFolder.append("Default");
|
|
if (defaultProfileFolder.exists()) {
|
|
profiles = [{
|
|
id: "Default",
|
|
name: "Default",
|
|
}];
|
|
}
|
|
}
|
|
|
|
// Only list profiles from which any data can be imported
|
|
return this.__sourceProfiles = profiles.filter(function(profile) {
|
|
let resources = this.getResources(profile);
|
|
return resources && resources.length > 0;
|
|
}, this);
|
|
}
|
|
});
|
|
|
|
Object.defineProperty(ChromeProfileMigrator.prototype, "sourceHomePageURL", {
|
|
get: function Chrome_sourceHomePageURL() {
|
|
let prefsFile = this._chromeUserDataFolder.clone();
|
|
prefsFile.append("Preferences");
|
|
if (prefsFile.exists()) {
|
|
// XXX reading and parsing JSON is synchronous.
|
|
let fstream = Cc[FILE_INPUT_STREAM_CID].
|
|
createInstance(Ci.nsIFileInputStream);
|
|
fstream.init(file, -1, 0, 0);
|
|
try {
|
|
return JSON.parse(
|
|
NetUtil.readInputStreamToString(fstream, fstream.available(),
|
|
{ charset: "UTF-8" })
|
|
).homepage;
|
|
}
|
|
catch(e) {
|
|
Cu.reportError("Error parsing Chrome's preferences file: " + e);
|
|
}
|
|
}
|
|
return "";
|
|
}
|
|
});
|
|
|
|
function GetBookmarksResource(aProfileFolder) {
|
|
let bookmarksFile = aProfileFolder.clone();
|
|
bookmarksFile.append("Bookmarks");
|
|
if (!bookmarksFile.exists())
|
|
return null;
|
|
|
|
return {
|
|
type: MigrationUtils.resourceTypes.BOOKMARKS,
|
|
|
|
migrate: function(aCallback) {
|
|
return Task.spawn(function* () {
|
|
let jsonStream = yield new Promise(resolve =>
|
|
NetUtil.asyncFetch({ uri: NetUtil.newURI(bookmarksFile),
|
|
loadUsingSystemPrincipal: true
|
|
},
|
|
(inputStream, resultCode) => {
|
|
if (Components.isSuccessCode(resultCode)) {
|
|
resolve(inputStream);
|
|
} else {
|
|
reject(new Error("Could not read Bookmarks file"));
|
|
}
|
|
}
|
|
)
|
|
);
|
|
|
|
// Parse Chrome bookmark file that is JSON format
|
|
let bookmarkJSON = NetUtil.readInputStreamToString(
|
|
jsonStream, jsonStream.available(), { charset : "UTF-8" });
|
|
let roots = JSON.parse(bookmarkJSON).roots;
|
|
|
|
// Importing bookmark bar items
|
|
if (roots.bookmark_bar.children &&
|
|
roots.bookmark_bar.children.length > 0) {
|
|
// Toolbar
|
|
let parentGuid = PlacesUtils.bookmarks.toolbarGuid;
|
|
if (!MigrationUtils.isStartupMigration) {
|
|
parentGuid =
|
|
yield MigrationUtils.createImportedBookmarksFolder("Chrome", parentGuid);
|
|
}
|
|
yield insertBookmarkItems(parentGuid, roots.bookmark_bar.children);
|
|
}
|
|
|
|
// Importing bookmark menu items
|
|
if (roots.other.children &&
|
|
roots.other.children.length > 0) {
|
|
// Bookmark menu
|
|
let parentGuid = PlacesUtils.bookmarks.menuGuid;
|
|
if (!MigrationUtils.isStartupMigration) {
|
|
parentGuid =
|
|
yield MigrationUtils.createImportedBookmarksFolder("Chrome", parentGuid);
|
|
}
|
|
yield insertBookmarkItems(parentGuid, roots.other.children);
|
|
}
|
|
}.bind(this)).then(() => aCallback(true),
|
|
e => { Cu.reportError(e); aCallback(false) });
|
|
}
|
|
};
|
|
}
|
|
|
|
function GetHistoryResource(aProfileFolder) {
|
|
let historyFile = aProfileFolder.clone();
|
|
historyFile.append("History");
|
|
if (!historyFile.exists())
|
|
return null;
|
|
|
|
return {
|
|
type: MigrationUtils.resourceTypes.HISTORY,
|
|
|
|
migrate: function(aCallback) {
|
|
let dbConn = Services.storage.openUnsharedDatabase(historyFile);
|
|
let stmt = dbConn.createAsyncStatement(
|
|
"SELECT url, title, last_visit_time, typed_count FROM urls WHERE hidden = 0");
|
|
|
|
stmt.executeAsync({
|
|
handleResult : function(aResults) {
|
|
let places = [];
|
|
for (let row = aResults.getNextRow(); row; row = aResults.getNextRow()) {
|
|
try {
|
|
// if having typed_count, we changes transition type to typed.
|
|
let transType = PlacesUtils.history.TRANSITION_LINK;
|
|
if (row.getResultByName("typed_count") > 0)
|
|
transType = PlacesUtils.history.TRANSITION_TYPED;
|
|
|
|
places.push({
|
|
uri: NetUtil.newURI(row.getResultByName("url")),
|
|
title: row.getResultByName("title"),
|
|
visits: [{
|
|
transitionType: transType,
|
|
visitDate: chromeTimeToDate(
|
|
row.getResultByName(
|
|
"last_visit_time")) * 1000,
|
|
}],
|
|
});
|
|
} catch (e) {
|
|
Cu.reportError(e);
|
|
}
|
|
}
|
|
|
|
try {
|
|
PlacesUtils.asyncHistory.updatePlaces(places);
|
|
} catch (e) {
|
|
Cu.reportError(e);
|
|
}
|
|
},
|
|
|
|
handleError : function(aError) {
|
|
Cu.reportError("Async statement execution returned with '" +
|
|
aError.result + "', '" + aError.message + "'");
|
|
},
|
|
|
|
handleCompletion : function(aReason) {
|
|
dbConn.asyncClose();
|
|
aCallback(aReason == Ci.mozIStorageStatementCallback.REASON_FINISHED);
|
|
}
|
|
});
|
|
stmt.finalize();
|
|
}
|
|
};
|
|
}
|
|
|
|
function GetCookiesResource(aProfileFolder) {
|
|
let cookiesFile = aProfileFolder.clone();
|
|
cookiesFile.append("Cookies");
|
|
if (!cookiesFile.exists())
|
|
return null;
|
|
|
|
return {
|
|
type: MigrationUtils.resourceTypes.COOKIES,
|
|
|
|
migrate: function(aCallback) {
|
|
let dbConn = Services.storage.openUnsharedDatabase(cookiesFile);
|
|
// We don't support decrypting cookies yet so only import plaintext ones.
|
|
let stmt = dbConn.createAsyncStatement(`
|
|
SELECT host_key, name, value, path, expires_utc, secure, httponly, encrypted_value
|
|
FROM cookies
|
|
WHERE length(encrypted_value) = 0`);
|
|
|
|
stmt.executeAsync({
|
|
handleResult : function(aResults) {
|
|
for (let row = aResults.getNextRow(); row; row = aResults.getNextRow()) {
|
|
let host_key = row.getResultByName("host_key");
|
|
if (host_key.match(/^\./)) {
|
|
// 1st character of host_key may be ".", so we have to remove it
|
|
host_key = host_key.substr(1);
|
|
}
|
|
|
|
try {
|
|
let expiresUtc =
|
|
chromeTimeToDate(row.getResultByName("expires_utc")) / 1000;
|
|
Services.cookies.add(host_key,
|
|
row.getResultByName("path"),
|
|
row.getResultByName("name"),
|
|
row.getResultByName("value"),
|
|
row.getResultByName("secure"),
|
|
row.getResultByName("httponly"),
|
|
false,
|
|
parseInt(expiresUtc));
|
|
} catch (e) {
|
|
Cu.reportError(e);
|
|
}
|
|
}
|
|
},
|
|
|
|
handleError : function(aError) {
|
|
Cu.reportError("Async statement execution returned with '" +
|
|
aError.result + "', '" + aError.message + "'");
|
|
},
|
|
|
|
handleCompletion : function(aReason) {
|
|
dbConn.asyncClose();
|
|
aCallback(aReason == Ci.mozIStorageStatementCallback.REASON_FINISHED);
|
|
},
|
|
});
|
|
stmt.finalize();
|
|
}
|
|
}
|
|
}
|
|
|
|
ChromeProfileMigrator.prototype.classDescription = "Chrome Profile Migrator";
|
|
ChromeProfileMigrator.prototype.contractID = "@mozilla.org/profile/migrator;1?app=browser&type=chrome";
|
|
ChromeProfileMigrator.prototype.classID = Components.ID("{4cec1de4-1671-4fc3-a53e-6c539dc77a26}");
|
|
|
|
this.NSGetFactory = XPCOMUtils.generateNSGetFactory([ChromeProfileMigrator]);
|