// -*- mode: js; js-indent-level: 4; indent-tabs-mode: nil -*- const Gio = imports.gi.Gio; const GLib = imports.gi.GLib; const Lang = imports.lang; const St = imports.gi.St; const FileUtils = imports.misc.fileUtils; const Search = imports.ui.search; const KEY_FILE_GROUP = 'Shell Search Provider'; const SearchProviderIface = ; const SearchProvider2Iface = ; var SearchProviderProxy = Gio.DBusProxy.makeProxyWrapper(SearchProviderIface); var SearchProvider2Proxy = Gio.DBusProxy.makeProxyWrapper(SearchProvider2Iface); function loadRemoteSearchProviders(addProviderCallback) { let data = { loadedProviders: [], objectPaths: {}, addProviderCallback: addProviderCallback }; FileUtils.collectFromDatadirsAsync('search-providers', { loadedCallback: remoteProvidersLoaded, processFile: loadRemoteSearchProvider, data: data }); } function loadRemoteSearchProvider(file, info, data) { let keyfile = new GLib.KeyFile(); let path = file.get_path(); try { keyfile.load_from_file(path, 0); } catch(e) { return; } if (!keyfile.has_group(KEY_FILE_GROUP)) return; let remoteProvider; try { let group = KEY_FILE_GROUP; let busName = keyfile.get_string(group, 'BusName'); let objectPath = keyfile.get_string(group, 'ObjectPath'); if (data.objectPaths[objectPath]) return; let appInfo = null; try { let desktopId = keyfile.get_string(group, 'DesktopId'); appInfo = Gio.DesktopAppInfo.new(desktopId); } catch (e) { log('Ignoring search provider ' + path + ': missing DesktopId'); return; } let version = '1'; try { version = keyfile.get_string(group, 'Version'); } catch (e) { // ignore error } if (version >= 2) remoteProvider = new RemoteSearchProvider2(appInfo, busName, objectPath); else remoteProvider = new RemoteSearchProvider(appInfo, busName, objectPath); data.objectPaths[objectPath] = remoteProvider; data.loadedProviders.push(remoteProvider); } catch(e) { log('Failed to add search provider %s: %s'.format(path, e.toString())); } } function remoteProvidersLoaded(loadState) { let searchSettings = new Gio.Settings({ schema: Search.SEARCH_PROVIDERS_SCHEMA }); let sortOrder = searchSettings.get_strv('sort-order'); let numSorted = sortOrder.length; loadState.loadedProviders.sort( function(providerA, providerB) { let idxA, idxB; let appIdA, appIdB; appIdA = providerA.appInfo.get_id(); appIdB = providerB.appInfo.get_id(); idxA = sortOrder.indexOf(appIdA); idxB = sortOrder.indexOf(appIdB); // if no provider is found in the order, use alphabetical order if ((idxA == -1) && (idxB == -1)) return GLib.utf8_collate(providerA.title, providerB.title); if (numSorted > 1) { // if providerA is the last, it goes after everything if ((idxA + 1) == numSorted) return 1; // if providerB is the last, it goes after everything else if ((idxB + 1) == numSorted) return -1; } // if providerA isn't found, it's sorted after providerB if (idxA == -1) return 1; // if providerB isn't found, it's sorted after providerA if (idxB == -1) return -1; // finally, if both providers are found, return their order in the list return (idxA - idxB); }); loadState.loadedProviders.forEach( function(provider) { loadState.addProviderCallback(provider); }); } const RemoteSearchProvider = new Lang.Class({ Name: 'RemoteSearchProvider', Extends: Search.SearchProvider, _init: function(appInfo, dbusName, dbusPath, proxyType) { if (!proxyType) proxyType = SearchProviderProxy; this.proxy = new proxyType(Gio.DBus.session, dbusName, dbusPath, Lang.bind(this, this._onProxyConstructed)); this.parent(appInfo, true); this._cancellable = new Gio.Cancellable(); }, _onProxyConstructed: function(proxy) { // Do nothing }, createIcon: function(size, meta) { if (meta['gicon']) { return new St.Icon({ gicon: Gio.icon_new_for_string(meta['gicon']), icon_size: size }); } else if (meta['icon-data']) { let [width, height, rowStride, hasAlpha, bitsPerSample, nChannels, data] = meta['icon-data']; let textureCache = St.TextureCache.get_default(); return textureCache.load_from_raw(data, hasAlpha, width, height, rowStride, size); } // Ugh, but we want to fall back to something ... return new St.Icon({ icon_name: 'text-x-generic', icon_size: size }); }, _getResultsFinished: function(results, error) { if (error) return; this.searchSystem.pushResults(this, results[0]); }, getInitialResultSet: function(terms) { this._cancellable.cancel(); this._cancellable.reset(); try { this.proxy.GetInitialResultSetRemote(terms, Lang.bind(this, this._getResultsFinished), this._cancellable); } catch(e) { log('Error calling GetInitialResultSet for provider %s: %s'.format( this.title, e.toString())); this.searchSystem.pushResults(this, []); } }, getSubsearchResultSet: function(previousResults, newTerms) { this._cancellable.cancel(); this._cancellable.reset(); try { this.proxy.GetSubsearchResultSetRemote(previousResults, newTerms, Lang.bind(this, this._getResultsFinished), this._cancellable); } catch(e) { log('Error calling GetSubsearchResultSet for provider %s: %s'.format(this.title, e.toString())); this.searchSystem.pushResults(this, []); } }, _getResultMetasFinished: function(results, error, callback) { if (error) { callback([]); return; } let metas = results[0]; let resultMetas = []; for (let i = 0; i < metas.length; i++) { for (let prop in metas[i]) metas[i][prop] = metas[i][prop].deep_unpack(); resultMetas.push({ id: metas[i]['id'], name: metas[i]['name'], createIcon: Lang.bind(this, this.createIcon, metas[i]) }); } callback(resultMetas); }, getResultMetas: function(ids, callback) { this._cancellable.cancel(); this._cancellable.reset(); try { this.proxy.GetResultMetasRemote(ids, Lang.bind(this, this._getResultMetasFinished, callback), this._cancellable); } catch(e) { log('Error calling GetResultMetas for provider %s: %s'.format(this.title, e.toString())); callback([]); } }, activateResult: function(id) { this.proxy.ActivateResultRemote(id); }, launchSearch: function(terms) { // the provider is not compatible with the new version of the interface, launch // the app itself but warn so we can catch the error in logs log('Search provider ' + this.appInfo.get_id() + ' does not implement LaunchSearch'); this.appInfo.launch([], global.create_app_launch_context()); } }); const RemoteSearchProvider2 = new Lang.Class({ Name: 'RemoteSearchProvider2', Extends: RemoteSearchProvider, _init: function(appInfo, dbusName, dbusPath) { this.parent(appInfo, dbusName, dbusPath, SearchProvider2Proxy); this.canLaunchSearch = true; }, activateResult: function(id, terms) { this.proxy.ActivateResultRemote(id, terms, global.get_current_time()); }, launchSearch: function(terms) { this.proxy.LaunchSearchRemote(terms); } });