Add a helper function to ensure we have the correct datastore installed

app/bin/tools_common.dart

 // Copyright (c) 2015, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 import 'dart:io';
 import 'dart:async';
 
 import 'package:appengine/appengine.dart';
+import 'package:gcloud/db.dart';
 import 'package:gcloud/service_scope.dart';
 import 'package:pub_dartlang_org/upload_signer_service.dart';
 
 import 'server_common.dart';
 import 'configuration.dart';
 
 Future withProdServices(Future fun()) {
-  if (Platform.isMacOS) {
-    return _withProdServices(() {
-      return fork(() async {
-        await initializeApiaryDatastore();
-        return fun();
-      });
+  return withCorrectDatastore((_) {
+    if (!Platform.environment.containsKey('GCLOUD_PROJECT') ||
+        !Platform.environment.containsKey('GCLOUD_KEY')) {
+      throw 'Missing GCLOUD_* environments for package:appengine';
+    }
+    return withAppEngineServices(() {
+      registerUploadSigner(
+          new ServiceAccountBasedUploadSigner(activeConfiguration.credentials));
+      initBackend();
+      return fun();
     });
-  } else {
-    return _withProdServices(fun);
-  }
+  }, null);
 }
-
-Future _withProdServices(Future fun()) async {
-  if (!Platform.environment.containsKey('GCLOUD_PROJECT') ||
-      !Platform.environment.containsKey('GCLOUD_KEY')) {
-    throw 'Missing GCLOUD_* environments for package:appengine';
-  }
-  return withAppEngineServices(() {
-    registerUploadSigner(
-        new ServiceAccountBasedUploadSigner(activeConfiguration.credentials));
-    initBackend();
-    return fun();
-  });
-}