Pass user session type to extension feature checks

extensions/common/features/feature_session_type.cc

Index: extensions/common/features/feature_session_type.cc
diff --git a/extensions/common/features/feature_session_type.cc b/extensions/common/features/feature_session_type.cc
new file mode 100644
index 0000000000000000000000000000000000000000..10f2af7a6123912f474978cc57d4fce3879a314c
--- /dev/null
+++ b/extensions/common/features/feature_session_type.cc
@@ -0,0 +1,39 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "extensions/common/features/feature_session_type.h"
+
+#include "base/logging.h"
+#include "base/memory/ptr_util.h"
+
+namespace extensions {
+
+namespace {
+
+const FeatureSessionType kDefaultSessionType = FeatureSessionType::INITIAL;
+FeatureSessionType g_current_session_type = kDefaultSessionType;
+
+}  // namespace
+
+FeatureSessionType GetCurrentFeatureSessionType() {
+  return g_current_session_type;
+}
+
+void SetCurrentFeatureSessionType(FeatureSessionType session_type) {
+  // Make sure that session type stays constant after it's been initialized.
+  // Note that this requirement can be bypassed in tests by using
+  // |ScopedCurrentFeatureSessionType|.
+  CHECK(g_current_session_type == kDefaultSessionType ||
+        session_type == g_current_session_type);
+  g_current_session_type = session_type;
+}
+
+std::unique_ptr<base::AutoReset<FeatureSessionType>>
+ScopedCurrentFeatureSessionType(FeatureSessionType type) {
+  CHECK_EQ(g_current_session_type, kDefaultSessionType);
+  return base::MakeUnique<base::AutoReset<FeatureSessionType>>(
+      &g_current_session_type, type);
+}
+
+}  // namespace extensions