• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2013 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.printspooler.ui;
18 
19 import android.annotation.NonNull;
20 import android.annotation.Nullable;
21 import android.app.Activity;
22 import android.app.LoaderManager;
23 import android.content.ComponentName;
24 import android.content.Context;
25 import android.content.Loader;
26 import android.content.pm.ServiceInfo;
27 import android.location.Criteria;
28 import android.location.Location;
29 import android.location.LocationListener;
30 import android.location.LocationManager;
31 import android.location.LocationRequest;
32 import android.os.AsyncTask;
33 import android.os.Bundle;
34 import android.os.Handler;
35 import android.os.Looper;
36 import android.os.SystemClock;
37 import android.print.PrintManager;
38 import android.print.PrintServicesLoader;
39 import android.print.PrinterDiscoverySession;
40 import android.print.PrinterDiscoverySession.OnPrintersChangeListener;
41 import android.print.PrinterId;
42 import android.print.PrinterInfo;
43 import android.printservice.PrintServiceInfo;
44 import android.util.ArrayMap;
45 import android.util.ArraySet;
46 import android.util.AtomicFile;
47 import android.util.Log;
48 import android.util.Pair;
49 import android.util.Slog;
50 import android.util.Xml;
51 
52 import com.android.internal.util.FastXmlSerializer;
53 
54 import libcore.io.IoUtils;
55 
56 import org.xmlpull.v1.XmlPullParser;
57 import org.xmlpull.v1.XmlPullParserException;
58 import org.xmlpull.v1.XmlSerializer;
59 
60 import java.io.File;
61 import java.io.FileInputStream;
62 import java.io.FileNotFoundException;
63 import java.io.FileOutputStream;
64 import java.io.IOException;
65 import java.nio.charset.StandardCharsets;
66 import java.util.ArrayList;
67 import java.util.Collections;
68 import java.util.HashSet;
69 import java.util.LinkedHashMap;
70 import java.util.List;
71 import java.util.Map;
72 import java.util.Objects;
73 import java.util.Set;
74 
75 /**
76  * This class is responsible for loading printers by doing discovery
77  * and merging the discovered printers with the previously used ones.
78  */
79 public final class FusedPrintersProvider extends Loader<List<PrinterInfo>>
80         implements LocationListener {
81     private static final String LOG_TAG = "FusedPrintersProvider";
82 
83     private static final boolean DEBUG = false;
84 
85     private static final double WEIGHT_DECAY_COEFFICIENT = 0.95f;
86     private static final int MAX_HISTORY_LENGTH = 50;
87 
88     private static final int MAX_FAVORITE_PRINTER_COUNT = 4;
89 
90     /** Interval of location updated in ms */
91     private static final int LOCATION_UPDATE_MS = 30 * 1000;
92 
93     /** Maximum acceptable age of the location in ms */
94     private static final int MAX_LOCATION_AGE_MS = 10 * 60 * 1000;
95 
96     /** The worst accuracy that is considered usable in m */
97     private static final int MIN_LOCATION_ACCURACY = 50;
98 
99     /** Maximum distance where a printer is still considered "near" */
100     private static final int MAX_PRINTER_DISTANCE = MIN_LOCATION_ACCURACY * 2;
101 
102     private final List<PrinterInfo> mPrinters =
103             new ArrayList<>();
104 
105     private final List<Pair<PrinterInfo, Location>> mFavoritePrinters =
106             new ArrayList<>();
107 
108     private final PersistenceManager mPersistenceManager;
109 
110     private PrinterDiscoverySession mDiscoverySession;
111 
112     private PrinterId mTrackedPrinter;
113 
114     private boolean mPrintersUpdatedBefore;
115 
116     /** Last known location, can be null or out of date */
117     private final Object mLocationLock;
118     private Location mLocation;
119 
120     /** Location used when the printers were updated the last time */
121     private Location mLocationOfLastPrinterUpdate;
122 
123     /** Reference to the system's location manager */
124     private final LocationManager mLocationManager;
125 
126     /**
127      * Get a reference to the current location.
128      */
getCurrentLocation()129     private Location getCurrentLocation() {
130         synchronized (mLocationLock) {
131             return mLocation;
132         }
133     }
134 
FusedPrintersProvider(Activity activity, int internalLoaderId)135     public FusedPrintersProvider(Activity activity, int internalLoaderId) {
136         super(activity);
137         mLocationLock = new Object();
138         mPersistenceManager = new PersistenceManager(activity, internalLoaderId);
139         mLocationManager = (LocationManager) activity.getSystemService(Context.LOCATION_SERVICE);
140     }
141 
addHistoricalPrinter(PrinterInfo printer)142     public void addHistoricalPrinter(PrinterInfo printer) {
143         mPersistenceManager.addPrinterAndWritePrinterHistory(printer);
144     }
145 
146     /**
147      * Add printer to dest, or if updatedPrinters add the updated printer. If the updated printer
148      * was added, remove it from updatedPrinters.
149      *
150      * @param dest The list the printers should be added to
151      * @param printer The printer to add
152      * @param updatedPrinters The printer to add
153      */
updateAndAddPrinter(List<PrinterInfo> dest, PrinterInfo printer, Map<PrinterId, PrinterInfo> updatedPrinters)154     private void updateAndAddPrinter(List<PrinterInfo> dest, PrinterInfo printer,
155             Map<PrinterId, PrinterInfo> updatedPrinters) {
156         PrinterInfo updatedPrinter = updatedPrinters.remove(printer.getId());
157         if (updatedPrinter != null) {
158             dest.add(updatedPrinter);
159         } else {
160             dest.add(printer);
161         }
162     }
163 
164     /**
165      * Compute the printers, order them appropriately and deliver the printers to the clients. We
166      * prefer printers that have been previously used (favorites) and printers that have been used
167      * previously close to the current location (near printers).
168      *
169      * @param discoveredPrinters All printers currently discovered by the print discovery session.
170      * @param favoritePrinters The ordered list of printers. The earlier in the list, the more
171      *            preferred.
172      */
computeAndDeliverResult(Map<PrinterId, PrinterInfo> discoveredPrinters, List<Pair<PrinterInfo, Location>> favoritePrinters)173     private void computeAndDeliverResult(Map<PrinterId, PrinterInfo> discoveredPrinters,
174             List<Pair<PrinterInfo, Location>> favoritePrinters) {
175         List<PrinterInfo> printers = new ArrayList<>();
176 
177         // Store the printerIds that have already been added. We cannot compare the printerInfos in
178         // "printers" as they might have been taken from discoveredPrinters and the printerInfo does
179         // not equals() anymore
180         HashSet<PrinterId> alreadyAddedPrinter = new HashSet<>(MAX_FAVORITE_PRINTER_COUNT);
181 
182         Location location = getCurrentLocation();
183 
184         // Add the favorite printers that have last been used close to the current location
185         final int favoritePrinterCount = favoritePrinters.size();
186         if (location != null) {
187             for (int i = 0; i < favoritePrinterCount; i++) {
188                 // Only add a certain amount of favorite printers
189                 if (printers.size() == MAX_FAVORITE_PRINTER_COUNT) {
190                     break;
191                 }
192 
193                 PrinterInfo favoritePrinter = favoritePrinters.get(i).first;
194                 Location printerLocation = favoritePrinters.get(i).second;
195 
196                 if (printerLocation != null
197                         && !alreadyAddedPrinter.contains(favoritePrinter.getId())) {
198                     if (printerLocation.distanceTo(location) <= MAX_PRINTER_DISTANCE) {
199                         updateAndAddPrinter(printers, favoritePrinter, discoveredPrinters);
200                         alreadyAddedPrinter.add(favoritePrinter.getId());
201                     }
202                 }
203             }
204         }
205 
206         // Add the other favorite printers
207         for (int i = 0; i < favoritePrinterCount; i++) {
208             // Only add a certain amount of favorite printers
209             if (printers.size() == MAX_FAVORITE_PRINTER_COUNT) {
210                 break;
211             }
212 
213             PrinterInfo favoritePrinter = favoritePrinters.get(i).first;
214             if (!alreadyAddedPrinter.contains(favoritePrinter.getId())) {
215                 updateAndAddPrinter(printers, favoritePrinter, discoveredPrinters);
216                 alreadyAddedPrinter.add(favoritePrinter.getId());
217             }
218         }
219 
220         // Add other updated printers. Printers that have already been added have been removed from
221         // discoveredPrinters in the calls to updateAndAddPrinter
222         final int printerCount = mPrinters.size();
223         for (int i = 0; i < printerCount; i++) {
224             PrinterInfo printer = mPrinters.get(i);
225             PrinterInfo updatedPrinter = discoveredPrinters.remove(
226                     printer.getId());
227             if (updatedPrinter != null) {
228                 printers.add(updatedPrinter);
229             }
230         }
231 
232         // Add the new printers, i.e. what is left.
233         printers.addAll(discoveredPrinters.values());
234 
235         // Update the list of printers.
236         mPrinters.clear();
237         mPrinters.addAll(printers);
238 
239         if (isStarted()) {
240             // If stated deliver the new printers.
241             deliverResult(printers);
242         } else {
243             // Otherwise, take a note for the change.
244             onContentChanged();
245         }
246     }
247 
248     @Override
onStartLoading()249     protected void onStartLoading() {
250         if (DEBUG) {
251             Log.i(LOG_TAG, "onStartLoading() " + FusedPrintersProvider.this.hashCode());
252         }
253 
254         mLocationManager.requestLocationUpdates(LocationRequest.create()
255                 .setQuality(LocationRequest.POWER_LOW).setInterval(LOCATION_UPDATE_MS), this,
256                 Looper.getMainLooper());
257 
258         Location lastLocation = mLocationManager.getLastLocation();
259         if (lastLocation != null) {
260             onLocationChanged(lastLocation);
261         }
262 
263         // Jumpstart location with a single forced update
264         Criteria oneTimeCriteria = new Criteria();
265         oneTimeCriteria.setAccuracy(Criteria.ACCURACY_FINE);
266         mLocationManager.requestSingleUpdate(oneTimeCriteria, this, Looper.getMainLooper());
267 
268         // The contract is that if we already have a valid,
269         // result the we have to deliver it immediately.
270         (new Handler(Looper.getMainLooper())).post(new Runnable() {
271             @Override public void run() {
272                 deliverResult(new ArrayList<>(mPrinters));
273             }
274         });
275 
276         // Always load the data to ensure discovery period is
277         // started and to make sure obsolete printers are updated.
278         onForceLoad();
279     }
280 
281     @Override
onStopLoading()282     protected void onStopLoading() {
283         if (DEBUG) {
284             Log.i(LOG_TAG, "onStopLoading() " + FusedPrintersProvider.this.hashCode());
285         }
286         onCancelLoad();
287 
288         mLocationManager.removeUpdates(this);
289     }
290 
291     @Override
onForceLoad()292     protected void onForceLoad() {
293         if (DEBUG) {
294             Log.i(LOG_TAG, "onForceLoad() " + FusedPrintersProvider.this.hashCode());
295         }
296         loadInternal();
297     }
298 
loadInternal()299     private void loadInternal() {
300         if (mDiscoverySession == null) {
301             PrintManager printManager = (PrintManager) getContext()
302                     .getSystemService(Context.PRINT_SERVICE);
303             mDiscoverySession = printManager.createPrinterDiscoverySession();
304             mPersistenceManager.readPrinterHistory();
305         } else if (mPersistenceManager.isHistoryChanged()) {
306             mPersistenceManager.readPrinterHistory();
307         }
308         if (mPersistenceManager.isReadHistoryCompleted()
309                 && !mDiscoverySession.isPrinterDiscoveryStarted()) {
310             mDiscoverySession.setOnPrintersChangeListener(new OnPrintersChangeListener() {
311                 @Override
312                 public void onPrintersChanged() {
313                     if (DEBUG) {
314                         Log.i(LOG_TAG, "onPrintersChanged() count:"
315                                 + mDiscoverySession.getPrinters().size()
316                                 + " " + FusedPrintersProvider.this.hashCode());
317                     }
318 
319                     updatePrinters(mDiscoverySession.getPrinters(), mFavoritePrinters,
320                             getCurrentLocation());
321                 }
322             });
323             final int favoriteCount = mFavoritePrinters.size();
324             List<PrinterId> printerIds = new ArrayList<>(favoriteCount);
325             for (int i = 0; i < favoriteCount; i++) {
326                 printerIds.add(mFavoritePrinters.get(i).first.getId());
327             }
328             mDiscoverySession.startPrinterDiscovery(printerIds);
329             List<PrinterInfo> printers = mDiscoverySession.getPrinters();
330 
331             updatePrinters(printers, mFavoritePrinters, getCurrentLocation());
332         }
333     }
334 
updatePrinters(List<PrinterInfo> printers, List<Pair<PrinterInfo, Location>> favoritePrinters, Location location)335     private void updatePrinters(List<PrinterInfo> printers,
336             List<Pair<PrinterInfo, Location>> favoritePrinters,
337             Location location) {
338         if (mPrintersUpdatedBefore && mPrinters.equals(printers)
339                 && mFavoritePrinters.equals(favoritePrinters)
340                 && Objects.equals(mLocationOfLastPrinterUpdate, location)) {
341             return;
342         }
343 
344         mLocationOfLastPrinterUpdate = location;
345         mPrintersUpdatedBefore = true;
346 
347         // Some of the found printers may have be a printer that is in the
348         // history but with its properties changed. Hence, we try to update the
349         // printer to use its current properties instead of the historical one.
350         mPersistenceManager.updateHistoricalPrintersIfNeeded(printers);
351 
352         Map<PrinterId, PrinterInfo> printersMap = new LinkedHashMap<>();
353         final int printerCount = printers.size();
354         for (int i = 0; i < printerCount; i++) {
355             PrinterInfo printer = printers.get(i);
356             printersMap.put(printer.getId(), printer);
357         }
358 
359         computeAndDeliverResult(printersMap, favoritePrinters);
360     }
361 
362     @Override
onCancelLoad()363     protected boolean onCancelLoad() {
364         if (DEBUG) {
365             Log.i(LOG_TAG, "onCancelLoad() " + FusedPrintersProvider.this.hashCode());
366         }
367         return cancelInternal();
368     }
369 
cancelInternal()370     private boolean cancelInternal() {
371         if (mDiscoverySession != null
372                 && mDiscoverySession.isPrinterDiscoveryStarted()) {
373             if (mTrackedPrinter != null) {
374                 mDiscoverySession.stopPrinterStateTracking(mTrackedPrinter);
375                 mTrackedPrinter = null;
376             }
377             mDiscoverySession.stopPrinterDiscovery();
378             return true;
379         } else if (mPersistenceManager.isReadHistoryInProgress()) {
380             return mPersistenceManager.stopReadPrinterHistory();
381         }
382         return false;
383     }
384 
385     @Override
onReset()386     protected void onReset() {
387         if (DEBUG) {
388             Log.i(LOG_TAG, "onReset() " + FusedPrintersProvider.this.hashCode());
389         }
390         onStopLoading();
391         mPrinters.clear();
392         if (mDiscoverySession != null) {
393             mDiscoverySession.destroy();
394         }
395     }
396 
397     @Override
onAbandon()398     protected void onAbandon() {
399         if (DEBUG) {
400             Log.i(LOG_TAG, "onAbandon() " + FusedPrintersProvider.this.hashCode());
401         }
402         onStopLoading();
403     }
404 
405     /**
406      * Check if the location is acceptable. This is to filter out excessively old or inaccurate
407      * location updates.
408      *
409      * @param location the location to check
410      * @return true iff the location is usable.
411      */
isLocationAcceptable(Location location)412     private boolean isLocationAcceptable(Location location) {
413         return location != null
414                 && location.getElapsedRealtimeNanos() > SystemClock.elapsedRealtimeNanos()
415                         - MAX_LOCATION_AGE_MS * 1000_000L
416                 && location.hasAccuracy()
417                 && location.getAccuracy() < MIN_LOCATION_ACCURACY;
418     }
419 
420     @Override
onLocationChanged(Location location)421     public void onLocationChanged(Location location) {
422         synchronized(mLocationLock) {
423             // We expect the user to not move too fast while printing. Hence prefer more accurate
424             // updates over more recent ones for LOCATION_UPDATE_MS. We add a 10% fudge factor here
425             // as the location provider might send an update slightly too early.
426             if (isLocationAcceptable(location)
427                     && !location.equals(mLocation)
428                     && (mLocation == null
429                             || location
430                                     .getElapsedRealtimeNanos() > mLocation.getElapsedRealtimeNanos()
431                                             + LOCATION_UPDATE_MS * 0.9 * 1000_000L
432                             || (!mLocation.hasAccuracy()
433                                     || location.getAccuracy() < mLocation.getAccuracy()))) {
434                 // Other callers of updatePrinters might want to know the location, hence cache it
435                 mLocation = location;
436 
437                 if (areHistoricalPrintersLoaded()) {
438                     updatePrinters(mDiscoverySession.getPrinters(), mFavoritePrinters, mLocation);
439                 }
440             }
441         }
442     }
443 
444     @Override
onStatusChanged(String provider, int status, Bundle extras)445     public void onStatusChanged(String provider, int status, Bundle extras) {
446         // nothing to do
447     }
448 
449     @Override
onProviderEnabled(String provider)450     public void onProviderEnabled(String provider) {
451         // nothing to do
452     }
453 
454     @Override
onProviderDisabled(String provider)455     public void onProviderDisabled(String provider) {
456         // nothing to do
457     }
458 
areHistoricalPrintersLoaded()459     public boolean areHistoricalPrintersLoaded() {
460         return mPersistenceManager.mReadHistoryCompleted;
461     }
462 
setTrackedPrinter(@ullable PrinterId printerId)463     public void setTrackedPrinter(@Nullable PrinterId printerId) {
464         if (isStarted() && mDiscoverySession != null
465                 && mDiscoverySession.isPrinterDiscoveryStarted()) {
466             if (mTrackedPrinter != null) {
467                 if (mTrackedPrinter.equals(printerId)) {
468                     return;
469                 }
470                 mDiscoverySession.stopPrinterStateTracking(mTrackedPrinter);
471             }
472             mTrackedPrinter = printerId;
473             if (printerId != null) {
474                 mDiscoverySession.startPrinterStateTracking(printerId);
475             }
476         }
477     }
478 
isFavoritePrinter(PrinterId printerId)479     public boolean isFavoritePrinter(PrinterId printerId) {
480         final int printerCount = mFavoritePrinters.size();
481         for (int i = 0; i < printerCount; i++) {
482             PrinterInfo favoritePritner = mFavoritePrinters.get(i).first;
483             if (favoritePritner.getId().equals(printerId)) {
484                 return true;
485             }
486         }
487         return false;
488     }
489 
forgetFavoritePrinter(PrinterId printerId)490     public void forgetFavoritePrinter(PrinterId printerId) {
491         final int favoritePrinterCount = mFavoritePrinters.size();
492         List<Pair<PrinterInfo, Location>> newFavoritePrinters = new ArrayList<>(
493                 favoritePrinterCount - 1);
494 
495         // Remove the printer from the favorites.
496         for (int i = 0; i < favoritePrinterCount; i++) {
497             if (!mFavoritePrinters.get(i).first.getId().equals(printerId)) {
498                 newFavoritePrinters.add(mFavoritePrinters.get(i));
499             }
500         }
501 
502         // Remove the printer from history and persist the latter.
503         mPersistenceManager.removeHistoricalPrinterAndWritePrinterHistory(printerId);
504 
505         // Recompute and deliver the printers.
506         updatePrinters(mDiscoverySession.getPrinters(), newFavoritePrinters, getCurrentLocation());
507     }
508 
509     private final class PersistenceManager implements
510             LoaderManager.LoaderCallbacks<List<PrintServiceInfo>> {
511         private static final String PERSIST_FILE_NAME = "printer_history.xml";
512 
513         private static final String TAG_PRINTERS = "printers";
514 
515         private static final String TAG_PRINTER = "printer";
516         private static final String TAG_LOCATION = "location";
517         private static final String TAG_PRINTER_ID = "printerId";
518 
519         private static final String ATTR_LOCAL_ID = "localId";
520         private static final String ATTR_SERVICE_NAME = "serviceName";
521 
522         private static final String ATTR_LONGITUDE = "longitude";
523         private static final String ATTR_LATITUDE = "latitude";
524         private static final String ATTR_ACCURACY = "accuracy";
525 
526         private static final String ATTR_NAME = "name";
527         private static final String ATTR_DESCRIPTION = "description";
528 
529         private final AtomicFile mStatePersistFile;
530 
531         /**
532          * Whether the enabled print services have been updated since last time the history was
533          * read.
534          */
535         private boolean mAreEnabledServicesUpdated;
536 
537         /** The enabled services read when they were last updated */
538         private @NonNull List<PrintServiceInfo> mEnabledServices;
539 
540         private List<Pair<PrinterInfo, Location>> mHistoricalPrinters = new ArrayList<>();
541 
542         private boolean mReadHistoryCompleted;
543 
544         private ReadTask mReadTask;
545 
546         private volatile long mLastReadHistoryTimestamp;
547 
PersistenceManager(final Activity activity, final int internalLoaderId)548         private PersistenceManager(final Activity activity, final int internalLoaderId) {
549             mStatePersistFile = new AtomicFile(new File(activity.getFilesDir(),
550                     PERSIST_FILE_NAME));
551 
552             // Initialize enabled services to make sure they are set are the read task might be done
553             // before the loader updated the services the first time.
554             mEnabledServices = ((PrintManager) activity
555                     .getSystemService(Context.PRINT_SERVICE))
556                     .getPrintServices(PrintManager.ENABLED_SERVICES);
557 
558             mAreEnabledServicesUpdated = true;
559 
560             // Cannot start a loader while starting another, hence delay this loader
561             (new Handler(activity.getMainLooper())).post(new Runnable() {
562                 @Override
563                 public void run() {
564                     activity.getLoaderManager().initLoader(internalLoaderId, null,
565                             PersistenceManager.this);
566                 }
567             });
568         }
569 
570 
571         @Override
onCreateLoader(int id, Bundle args)572         public Loader<List<PrintServiceInfo>> onCreateLoader(int id, Bundle args) {
573             return new PrintServicesLoader(
574                     (PrintManager) getContext().getSystemService(Context.PRINT_SERVICE),
575                     getContext(), PrintManager.ENABLED_SERVICES);
576         }
577 
578         @Override
onLoadFinished(Loader<List<PrintServiceInfo>> loader, List<PrintServiceInfo> services)579         public void onLoadFinished(Loader<List<PrintServiceInfo>> loader,
580                 List<PrintServiceInfo> services) {
581             mAreEnabledServicesUpdated = true;
582             mEnabledServices = services;
583 
584             // Ask the fused printer provider to reload which will cause the persistence manager to
585             // reload the history and reconsider the enabled services.
586             if (isStarted()) {
587                 forceLoad();
588             }
589         }
590 
591         @Override
onLoaderReset(Loader<List<PrintServiceInfo>> loader)592         public void onLoaderReset(Loader<List<PrintServiceInfo>> loader) {
593             // no data is cached
594         }
595 
isReadHistoryInProgress()596         public boolean isReadHistoryInProgress() {
597             return mReadTask != null;
598         }
599 
isReadHistoryCompleted()600         public boolean isReadHistoryCompleted() {
601             return mReadHistoryCompleted;
602         }
603 
stopReadPrinterHistory()604         public boolean stopReadPrinterHistory() {
605             return mReadTask.cancel(true);
606         }
607 
readPrinterHistory()608         public void readPrinterHistory() {
609             if (DEBUG) {
610                 Log.i(LOG_TAG, "read history started "
611                         + FusedPrintersProvider.this.hashCode());
612             }
613             mReadTask = new ReadTask();
614             mReadTask.executeOnExecutor(AsyncTask.SERIAL_EXECUTOR, (Void[]) null);
615         }
616 
updateHistoricalPrintersIfNeeded(List<PrinterInfo> printers)617         public void updateHistoricalPrintersIfNeeded(List<PrinterInfo> printers) {
618             boolean writeHistory = false;
619 
620             final int printerCount = printers.size();
621             for (int i = 0; i < printerCount; i++) {
622                 PrinterInfo printer = printers.get(i);
623                 writeHistory |= updateHistoricalPrinterIfNeeded(printer);
624             }
625 
626             if (writeHistory) {
627                 writePrinterHistory();
628             }
629         }
630 
631         /**
632          * Updates the historical printer state with the given printer.
633          *
634          * @param printer the printer to update
635          *
636          * @return true iff the historical printer list needs to be updated
637          */
updateHistoricalPrinterIfNeeded(PrinterInfo printer)638         public boolean updateHistoricalPrinterIfNeeded(PrinterInfo printer) {
639             boolean writeHistory = false;
640             final int printerCount = mHistoricalPrinters.size();
641             for (int i = 0; i < printerCount; i++) {
642                 PrinterInfo historicalPrinter = mHistoricalPrinters.get(i).first;
643 
644                 if (!historicalPrinter.getId().equals(printer.getId())) {
645                     continue;
646                 }
647 
648                 // Overwrite the historical printer with the updated printer as some properties
649                 // changed. We ignore the status as this is a volatile state.
650                 if (historicalPrinter.equalsIgnoringStatus(printer)) {
651                     continue;
652                 }
653 
654                 mHistoricalPrinters.set(i, new Pair<PrinterInfo, Location>(printer,
655                                 mHistoricalPrinters.get(i).second));
656 
657                 // We only persist limited information in the printer history, hence check if
658                 // we need to persist the update.
659                 // @see PersistenceManager.WriteTask#doWritePrinterHistory
660                 if (!historicalPrinter.getName().equals(printer.getName())) {
661                     if (Objects.equals(historicalPrinter.getDescription(),
662                             printer.getDescription())) {
663                         writeHistory = true;
664                     }
665                 }
666             }
667             return writeHistory;
668         }
669 
addPrinterAndWritePrinterHistory(PrinterInfo printer)670         public void addPrinterAndWritePrinterHistory(PrinterInfo printer) {
671             if (mHistoricalPrinters.size() >= MAX_HISTORY_LENGTH) {
672                 mHistoricalPrinters.remove(0);
673             }
674 
675             Location location = getCurrentLocation();
676             if (!isLocationAcceptable(location)) {
677                 location = null;
678             }
679 
680             mHistoricalPrinters.add(new Pair<PrinterInfo, Location>(printer, location));
681 
682             writePrinterHistory();
683         }
684 
removeHistoricalPrinterAndWritePrinterHistory(PrinterId printerId)685         public void removeHistoricalPrinterAndWritePrinterHistory(PrinterId printerId) {
686             boolean writeHistory = false;
687             final int printerCount = mHistoricalPrinters.size();
688             for (int i = printerCount - 1; i >= 0; i--) {
689                 PrinterInfo historicalPrinter = mHistoricalPrinters.get(i).first;
690                 if (historicalPrinter.getId().equals(printerId)) {
691                     mHistoricalPrinters.remove(i);
692                     writeHistory = true;
693                 }
694             }
695             if (writeHistory) {
696                 writePrinterHistory();
697             }
698         }
699 
700         @SuppressWarnings("unchecked")
writePrinterHistory()701         private void writePrinterHistory() {
702             new WriteTask().executeOnExecutor(AsyncTask.SERIAL_EXECUTOR,
703                     new ArrayList<>(mHistoricalPrinters));
704         }
705 
isHistoryChanged()706         public boolean isHistoryChanged() {
707             return mAreEnabledServicesUpdated ||
708                     mLastReadHistoryTimestamp != mStatePersistFile.getBaseFile().lastModified();
709         }
710 
711         /**
712          * Sort the favorite printers by weight. If a printer is in the list multiple times for
713          * different locations, all instances are considered to have the accumulative weight. The
714          * actual favorite printers to display are computed in {@link #computeAndDeliverResult} as
715          * only at this time we know the location to use to determine if a printer is close enough
716          * to be preferred.
717          *
718          * @param printers The printers to sort.
719          * @return The sorted printers.
720          */
sortFavoritePrinters( List<Pair<PrinterInfo, Location>> printers)721         private List<Pair<PrinterInfo, Location>> sortFavoritePrinters(
722                 List<Pair<PrinterInfo, Location>> printers) {
723             Map<PrinterId, PrinterRecord> recordMap = new ArrayMap<>();
724 
725             // Compute the weights.
726             float currentWeight = 1.0f;
727             final int printerCount = printers.size();
728             for (int i = printerCount - 1; i >= 0; i--) {
729                 PrinterId printerId = printers.get(i).first.getId();
730                 PrinterRecord record = recordMap.get(printerId);
731                 if (record == null) {
732                     record = new PrinterRecord();
733                     recordMap.put(printerId, record);
734                 }
735 
736                 record.printers.add(printers.get(i));
737 
738                 // Aggregate weight for the same printer
739                 record.weight += currentWeight;
740                 currentWeight *= WEIGHT_DECAY_COEFFICIENT;
741             }
742 
743             // Sort the favorite printers.
744             List<PrinterRecord> favoriteRecords = new ArrayList<>(
745                     recordMap.values());
746             Collections.sort(favoriteRecords);
747 
748             // Write the favorites to the output.
749             final int recordCount = favoriteRecords.size();
750             List<Pair<PrinterInfo, Location>> favoritePrinters = new ArrayList<>(printerCount);
751             for (int i = 0; i < recordCount; i++) {
752                 favoritePrinters.addAll(favoriteRecords.get(i).printers);
753             }
754 
755             return favoritePrinters;
756         }
757 
758         /**
759          * A set of printers with the same ID and the weight associated with them during
760          * {@link #sortFavoritePrinters}.
761          */
762         private final class PrinterRecord implements Comparable<PrinterRecord> {
763             /**
764              * The printers, all with the same ID, but potentially different properties or locations
765              */
766             public final List<Pair<PrinterInfo, Location>> printers;
767 
768             /** The weight associated with the printers */
769             public float weight;
770 
771             /**
772              * Create a new record.
773              */
PrinterRecord()774             public PrinterRecord() {
775                 printers = new ArrayList<>();
776             }
777 
778             /**
779              * Compare two records by weight.
780              */
781             @Override
compareTo(PrinterRecord another)782             public int compareTo(PrinterRecord another) {
783                 return Float.floatToIntBits(another.weight) - Float.floatToIntBits(weight);
784             }
785         }
786 
787         private final class ReadTask
788                 extends AsyncTask<Void, Void, List<Pair<PrinterInfo, Location>>> {
789             @Override
doInBackground(Void... args)790             protected List<Pair<PrinterInfo, Location>> doInBackground(Void... args) {
791                return doReadPrinterHistory();
792             }
793 
794             @Override
onPostExecute(List<Pair<PrinterInfo, Location>> printers)795             protected void onPostExecute(List<Pair<PrinterInfo, Location>> printers) {
796                 if (DEBUG) {
797                     Log.i(LOG_TAG, "read history completed "
798                             + FusedPrintersProvider.this.hashCode());
799                 }
800 
801                 // Ignore printer records whose target services are not enabled.
802                 Set<ComponentName> enabledComponents = new ArraySet<>();
803                 final int installedServiceCount = mEnabledServices.size();
804                 for (int i = 0; i < installedServiceCount; i++) {
805                     ServiceInfo serviceInfo = mEnabledServices.get(i).getResolveInfo().serviceInfo;
806                     ComponentName componentName = new ComponentName(
807                             serviceInfo.packageName, serviceInfo.name);
808                     enabledComponents.add(componentName);
809                 }
810                 mAreEnabledServicesUpdated = false;
811 
812                 final int printerCount = printers.size();
813                 for (int i = printerCount - 1; i >= 0; i--) {
814                     ComponentName printerServiceName = printers.get(i).first.getId()
815                             .getServiceName();
816                     if (!enabledComponents.contains(printerServiceName)) {
817                         printers.remove(i);
818                     }
819                 }
820 
821                 // Store the filtered list.
822                 mHistoricalPrinters = printers;
823 
824                 // Compute the favorite printers.
825                 mFavoritePrinters.clear();
826                 mFavoritePrinters.addAll(sortFavoritePrinters(mHistoricalPrinters));
827 
828                 mReadHistoryCompleted = true;
829 
830                 // Deliver the printers.
831                 updatePrinters(mDiscoverySession.getPrinters(), mFavoritePrinters,
832                         getCurrentLocation());
833 
834                 // We are done.
835                 mReadTask = null;
836 
837                 // Loading the available printers if needed.
838                 loadInternal();
839             }
840 
841             @Override
onCancelled(List<Pair<PrinterInfo, Location>> printerInfos)842             protected void onCancelled(List<Pair<PrinterInfo, Location>> printerInfos) {
843                 // We are done.
844                 mReadTask = null;
845             }
846 
doReadPrinterHistory()847             private List<Pair<PrinterInfo, Location>> doReadPrinterHistory() {
848                 final FileInputStream in;
849                 try {
850                     in = mStatePersistFile.openRead();
851                 } catch (FileNotFoundException fnfe) {
852                     if (DEBUG) {
853                         Log.i(LOG_TAG, "No existing printer history "
854                                 + FusedPrintersProvider.this.hashCode());
855                     }
856                     return new ArrayList<>();
857                 }
858                 try {
859                     List<Pair<PrinterInfo, Location>> printers = new ArrayList<>();
860                     XmlPullParser parser = Xml.newPullParser();
861                     parser.setInput(in, StandardCharsets.UTF_8.name());
862                     parseState(parser, printers);
863                     // Take a note which version of the history was read.
864                     mLastReadHistoryTimestamp = mStatePersistFile.getBaseFile().lastModified();
865                     return printers;
866                 } catch (IllegalStateException
867                         | NullPointerException
868                         | NumberFormatException
869                         | XmlPullParserException
870                         | IOException
871                         | IndexOutOfBoundsException e) {
872                     Slog.w(LOG_TAG, "Failed parsing ", e);
873                 } finally {
874                     IoUtils.closeQuietly(in);
875                 }
876 
877                 return Collections.emptyList();
878             }
879 
parseState(XmlPullParser parser, List<Pair<PrinterInfo, Location>> outPrinters)880             private void parseState(XmlPullParser parser,
881                     List<Pair<PrinterInfo, Location>> outPrinters)
882                             throws IOException, XmlPullParserException {
883                 parser.next();
884                 skipEmptyTextTags(parser);
885                 expect(parser, XmlPullParser.START_TAG, TAG_PRINTERS);
886                 parser.next();
887 
888                 while (parsePrinter(parser, outPrinters)) {
889                     // Be nice and respond to cancellation
890                     if (isCancelled()) {
891                         return;
892                     }
893                     parser.next();
894                 }
895 
896                 skipEmptyTextTags(parser);
897                 expect(parser, XmlPullParser.END_TAG, TAG_PRINTERS);
898             }
899 
parsePrinter(XmlPullParser parser, List<Pair<PrinterInfo, Location>> outPrinters)900             private boolean parsePrinter(XmlPullParser parser,
901                     List<Pair<PrinterInfo, Location>> outPrinters)
902                             throws IOException, XmlPullParserException {
903                 skipEmptyTextTags(parser);
904                 if (!accept(parser, XmlPullParser.START_TAG, TAG_PRINTER)) {
905                     return false;
906                 }
907 
908                 String name = parser.getAttributeValue(null, ATTR_NAME);
909                 String description = parser.getAttributeValue(null, ATTR_DESCRIPTION);
910 
911                 parser.next();
912 
913                 skipEmptyTextTags(parser);
914                 expect(parser, XmlPullParser.START_TAG, TAG_PRINTER_ID);
915                 String localId = parser.getAttributeValue(null, ATTR_LOCAL_ID);
916                 ComponentName service = ComponentName.unflattenFromString(parser.getAttributeValue(
917                         null, ATTR_SERVICE_NAME));
918                 PrinterId printerId =  new PrinterId(service, localId);
919                 parser.next();
920                 skipEmptyTextTags(parser);
921                 expect(parser, XmlPullParser.END_TAG, TAG_PRINTER_ID);
922                 parser.next();
923 
924                 skipEmptyTextTags(parser);
925                 Location location;
926                 if (accept(parser, XmlPullParser.START_TAG, TAG_LOCATION)) {
927                     location = new Location("");
928                     location.setLongitude(
929                             Double.parseDouble(parser.getAttributeValue(null, ATTR_LONGITUDE)));
930                     location.setLatitude(
931                             Double.parseDouble(parser.getAttributeValue(null, ATTR_LATITUDE)));
932                     location.setAccuracy(
933                             Float.parseFloat(parser.getAttributeValue(null, ATTR_ACCURACY)));
934                     parser.next();
935 
936                     skipEmptyTextTags(parser);
937                     expect(parser, XmlPullParser.END_TAG, TAG_LOCATION);
938                     parser.next();
939                 } else {
940                     location = null;
941                 }
942 
943                 // If the printer is available the printer will be replaced by the one read from the
944                 // discovery session, hence the only time when this object is used is when the
945                 // printer is unavailable.
946                 PrinterInfo.Builder builder = new PrinterInfo.Builder(printerId, name,
947                         PrinterInfo.STATUS_UNAVAILABLE);
948                 builder.setDescription(description);
949                 PrinterInfo printer = builder.build();
950 
951                 outPrinters.add(new Pair<PrinterInfo, Location>(printer, location));
952 
953                 if (DEBUG) {
954                     Log.i(LOG_TAG, "[RESTORED] " + printer);
955                 }
956 
957                 skipEmptyTextTags(parser);
958                 expect(parser, XmlPullParser.END_TAG, TAG_PRINTER);
959 
960                 return true;
961             }
962 
expect(XmlPullParser parser, int type, String tag)963             private void expect(XmlPullParser parser, int type, String tag)
964                     throws XmlPullParserException {
965                 if (!accept(parser, type, tag)) {
966                     throw new XmlPullParserException("Exepected event: " + type
967                             + " and tag: " + tag + " but got event: " + parser.getEventType()
968                             + " and tag:" + parser.getName());
969                 }
970             }
971 
skipEmptyTextTags(XmlPullParser parser)972             private void skipEmptyTextTags(XmlPullParser parser)
973                     throws IOException, XmlPullParserException {
974                 while (accept(parser, XmlPullParser.TEXT, null)
975                         && "\n".equals(parser.getText())) {
976                     parser.next();
977                 }
978             }
979 
accept(XmlPullParser parser, int type, String tag)980             private boolean accept(XmlPullParser parser, int type, String tag)
981                     throws XmlPullParserException {
982                 if (parser.getEventType() != type) {
983                     return false;
984                 }
985                 if (tag != null) {
986                     if (!tag.equals(parser.getName())) {
987                         return false;
988                     }
989                 } else if (parser.getName() != null) {
990                     return false;
991                 }
992                 return true;
993             }
994         }
995 
996         private final class WriteTask
997                 extends AsyncTask<List<Pair<PrinterInfo, Location>>, Void, Void> {
998             @Override
doInBackground( @uppressWarnings"unchecked") List<Pair<PrinterInfo, Location>>.... printers)999             protected Void doInBackground(
1000                     @SuppressWarnings("unchecked") List<Pair<PrinterInfo, Location>>... printers) {
1001                 doWritePrinterHistory(printers[0]);
1002                 return null;
1003             }
1004 
doWritePrinterHistory(List<Pair<PrinterInfo, Location>> printers)1005             private void doWritePrinterHistory(List<Pair<PrinterInfo, Location>> printers) {
1006                 FileOutputStream out = null;
1007                 try {
1008                     out = mStatePersistFile.startWrite();
1009 
1010                     XmlSerializer serializer = new FastXmlSerializer();
1011                     serializer.setOutput(out, StandardCharsets.UTF_8.name());
1012                     serializer.startDocument(null, true);
1013                     serializer.startTag(null, TAG_PRINTERS);
1014 
1015                     final int printerCount = printers.size();
1016                     for (int i = 0; i < printerCount; i++) {
1017                         PrinterInfo printer = printers.get(i).first;
1018 
1019                         serializer.startTag(null, TAG_PRINTER);
1020 
1021                         serializer.attribute(null, ATTR_NAME, printer.getName());
1022                         String description = printer.getDescription();
1023                         if (description != null) {
1024                             serializer.attribute(null, ATTR_DESCRIPTION, description);
1025                         }
1026 
1027                         PrinterId printerId = printer.getId();
1028                         serializer.startTag(null, TAG_PRINTER_ID);
1029                         serializer.attribute(null, ATTR_LOCAL_ID, printerId.getLocalId());
1030                         serializer.attribute(null, ATTR_SERVICE_NAME, printerId.getServiceName()
1031                                 .flattenToString());
1032                         serializer.endTag(null, TAG_PRINTER_ID);
1033 
1034                         Location location = printers.get(i).second;
1035                         if (location != null) {
1036                             serializer.startTag(null, TAG_LOCATION);
1037                             serializer.attribute(null, ATTR_LONGITUDE,
1038                                     String.valueOf(location.getLongitude()));
1039                             serializer.attribute(null, ATTR_LATITUDE,
1040                                     String.valueOf(location.getLatitude()));
1041                             serializer.attribute(null, ATTR_ACCURACY,
1042                                     String.valueOf(location.getAccuracy()));
1043                             serializer.endTag(null, TAG_LOCATION);
1044                         }
1045 
1046                         serializer.endTag(null, TAG_PRINTER);
1047 
1048                         if (DEBUG) {
1049                             Log.i(LOG_TAG, "[PERSISTED] " + printer);
1050                         }
1051                     }
1052 
1053                     serializer.endTag(null, TAG_PRINTERS);
1054                     serializer.endDocument();
1055                     mStatePersistFile.finishWrite(out);
1056 
1057                     if (DEBUG) {
1058                         Log.i(LOG_TAG, "[PERSIST END]");
1059                     }
1060                 } catch (IOException ioe) {
1061                     Slog.w(LOG_TAG, "Failed to write printer history, restoring backup.", ioe);
1062                     mStatePersistFile.failWrite(out);
1063                 } finally {
1064                     IoUtils.closeQuietly(out);
1065                 }
1066             }
1067         }
1068     }
1069 }
1070