src/chrome/common/extensions/docs/examples/api/tabs/screenshot/background.js

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 // To make sure we can uniquely identify each screenshot tab, add an id as a
   6 // query param to the url that displays the screenshot.
   7 // Note: It's OK that this is a global variable (and not in localStorage),
   8 // because the event page will stay open as long as any screenshot tabs are
   9 // open.
  10 var id = 100;
  11
  12 function takeScreenshot() {
  13   chrome.tabs.captureVisibleTab(null, function(img) {
  14     var screenshotUrl = img;
  15     var viewTabUrl = chrome.extension.getURL('screenshot.html?id=' + id++)
  16
  17     chrome.tabs.create({url: viewTabUrl}, function(tab) {
  18       var targetId = tab.id;
  19
  20       var addSnapshotImageToTab = function(tabId, changedProps) {
  21         // We are waiting for the tab we opened to finish loading.
  22         // Check that the the tab's id matches the tab we opened,
  23         // and that the tab is done loading.
  24         if (tabId != targetId || changedProps.status != "complete")
  25           return;
  26
  27         // Passing the above test means this is the event we were waiting for.
  28         // There is nothing we need to do for future onUpdated events, so we
  29         // use removeListner to stop geting called when onUpdated events fire.
  30         chrome.tabs.onUpdated.removeListener(addSnapshotImageToTab);
  31
  32         // Look through all views to find the window which will display
  33         // the screenshot.  The url of the tab which will display the
  34         // screenshot includes a query parameter with a unique id, which
  35         // ensures that exactly one view will have the matching URL.
  36         var views = chrome.extension.getViews();
  37         for (var i = 0; i < views.length; i++) {
  38           var view = views[i];
  39           if (view.location.href == viewTabUrl) {
  40             view.setScreenshotUrl(screenshotUrl);
  41             break;
  42           }
  43         }
  44       };
  45       chrome.tabs.onUpdated.addListener(addSnapshotImageToTab);
  46     });
  47   });
  48 }
  49
  50 // Listen for a click on the camera icon.  On that click, take a screenshot.
  51 chrome.browserAction.onClicked.addListener(function(tab) {
  52   if (tab.url.match("^https?://code.google.com")) {
  53     takeScreenshot();
  54   } else {
  55     alert('This sample can only take screenshots of code.google.com pages');
  56   }
  57 });