Internal change

src/java/com/google/devtools/mobileharness/fe/v6/service/config/util/BUILD

@@ -33,3 +33,17 @@ java_library(
         "@maven//:com_google_protobuf_protobuf_java",
     ],
 )
+
+java_library(
+    name = "config_service_capability",
+    srcs = [
+        "ConfigServiceCapability.java",
+        "ConfigServiceCapabilityFactory.java",
+    ],
+    deps = [
+        "//src/devtools/mobileharness/fe/v6/service/proto/config:config_resources_java_proto",
+        "//src/java/com/google/devtools/mobileharness/fe/v6/service/util",
+        "//src/java/com/google/devtools/mobileharness/shared/util/flags",
+        "@maven//:com_google_inject_extensions_guice_assistedinject",
+    ],
+)

src/java/com/google/devtools/mobileharness/fe/v6/service/config/util/ConfigServiceCapability.java

@@ -0,0 +1,148 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.config.util;
+
+import com.google.devtools.mobileharness.fe.v6.service.proto.config.DeviceConfigUiStatus;
+import com.google.devtools.mobileharness.fe.v6.service.proto.config.Editability;
+import com.google.devtools.mobileharness.fe.v6.service.proto.config.HostConfigUiStatus;
+import com.google.devtools.mobileharness.fe.v6.service.proto.config.HostDeviceConfigUiStatus;
+import com.google.devtools.mobileharness.fe.v6.service.proto.config.HostPropertiesUiStatus;
+import com.google.devtools.mobileharness.fe.v6.service.proto.config.PartStatus;
+import com.google.devtools.mobileharness.fe.v6.service.util.Environment;
+import com.google.devtools.mobileharness.shared.util.flags.Flags;
+import com.google.inject.assistedinject.Assisted;
+import com.google.inject.assistedinject.AssistedInject;
+
+/**
+ * Defines the capabilities and scope of the Config Service based on the current environment and
+ * target universe.
+ */
+public class ConfigServiceCapability {
+
+  private final Environment environment;
+  private final String universe;
+
+  @AssistedInject
+  ConfigServiceCapability(Environment environment, @Assisted String universe) {
+    this.environment = environment;
+    this.universe = universe;
+  }
+
+  /** Returns true if the configuration service is available for the current context. */
+  public boolean isConfigServiceAvailable() {
+    if (!Flags.instance().feConnectToConfigServer.get()) {
+      return false;
+    }
+
+    if (environment.isGoogleInternal()) {
+      return universe.isEmpty() || universe.equals("google_1p");
+    }
+
+    // OSS/ATS only supports the default context (empty universe).
+    return universe.isEmpty();
+  }
+
+  /**
+   * Checks if the configuration service is available and throws an {@link
+   * UnsupportedOperationException} with an environment-aware message if not.
+   */
+  public void checkConfigServiceAvailability() {
+    if (isConfigServiceAvailable()) {
+      return;
+    }
+
+    if (!Flags.instance().feConnectToConfigServer.getNonNull()) {
+      throw new UnsupportedOperationException("Configuration service is currently disabled.");
+    }
+
+    if (environment.isGoogleInternal()) {
+      throw new UnsupportedOperationException(
+          String.format(
+              "Configuration operations are not currently supported for universe '%s'.", universe));
+    }
+
+    throw new UnsupportedOperationException(
+        "Configuration operations are not currently supported.");
+  }
+
+  /** Calculates UI status for host configuration components. */
+  public HostConfigUiStatus calculateHostUiStatus() {
+    boolean isAts = environment.isAts();
+
+    return HostConfigUiStatus.newBuilder()
+        .setHostAdmins(
+            PartStatus.newBuilder()
+                .setVisible(!isAts)
+                .setEditability(Editability.newBuilder().setEditable(!isAts).build())
+                .build())
+        .setDeviceConfigMode(
+            PartStatus.newBuilder()
+                .setVisible(true)
+                .setEditability(Editability.newBuilder().setEditable(true).build())
+                .build())
+        .setDeviceConfig(
+            HostDeviceConfigUiStatus.newBuilder()
+                .setSectionStatus(
+                    PartStatus.newBuilder()
+                        .setVisible(true)
+                        .setEditability(Editability.newBuilder().setEditable(true).build())
+                        .build())
+                .setSubSections(calculateDeviceUiStatus())
+                .build())
+        .setHostProperties(
+            HostPropertiesUiStatus.newBuilder()
+                .setSectionStatus(
+                    PartStatus.newBuilder()
+                        .setVisible(!isAts)
+                        .setEditability(Editability.newBuilder().setEditable(!isAts).build())
+                        .build())
+                .build())
+        .setDeviceDiscovery(
+            PartStatus.newBuilder()
+                .setVisible(!isAts)
+                .setEditability(Editability.newBuilder().setEditable(!isAts).build())
+                .build())
+        .build();
+  }
+
+  /** Calculates UI status for device configuration components. */
+  public DeviceConfigUiStatus calculateDeviceUiStatus() {
+    boolean isAts = environment.isAts();
+    return DeviceConfigUiStatus.newBuilder()
+        .setPermissions(
+            PartStatus.newBuilder()
+                .setVisible(!isAts)
+                .setEditability(Editability.newBuilder().setEditable(!isAts).build())
+                .build())
+        .setWifi(
+            PartStatus.newBuilder()
+                .setVisible(true)
+                .setEditability(Editability.newBuilder().setEditable(true).build())
+                .build())
+        .setDimensions(
+            PartStatus.newBuilder()
+                .setVisible(true)
+                .setEditability(Editability.newBuilder().setEditable(true).build())
+                .build())
+        .setSettings(
+            PartStatus.newBuilder()
+                .setVisible(!isAts)
+                .setEditability(Editability.newBuilder().setEditable(!isAts).build())
+                .build())
+        .build();
+  }
+}

src/java/com/google/devtools/mobileharness/fe/v6/service/config/util/ConfigServiceCapabilityFactory.java

@@ -0,0 +1,23 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.config.util;
+
+/** Factory for creating {@link ConfigServiceCapability} instances. */
+public interface ConfigServiceCapabilityFactory {
+
+  ConfigServiceCapability create(String universe);
+}

src/java/com/google/devtools/mobileharness/shared/util/flags/Flags.java

@@ -1457,6 +1457,14 @@ public enum ConfigServiceStorageType {
       converter = Flag.StringConverter.class)
   public Flag<String> fastbootPathFromUser = fastbootPathFromUserDefault;
 
+  private static final Flag<Boolean> feConnectToConfigServerDefault = Flag.value(false);
+
+  @com.beust.jcommander.Parameter(
+      names = "--fe_connect_to_config_server",
+      description = "Whether to connect to the config server in the FE server.",
+      converter = Flag.BooleanConverter.class)
+  public Flag<Boolean> feConnectToConfigServer = feConnectToConfigServerDefault;
+
   private static final Flag<Boolean> feEnableDeviceFlashingDefault = Flag.value(false);
 
   @com.beust.jcommander.Parameter(

src/javatests/com/google/devtools/mobileharness/fe/v6/service/config/util/BUILD

@@ -25,6 +25,7 @@ java_library(
     testonly = 1,
     srcs = [
         "ConfigConverterTest.java",
+        "ConfigServiceCapabilityTest.java",
     ],
     deps = [
         "//src/devtools/mobileharness/api/deviceconfig/proto:basic_java_proto",
@@ -34,10 +35,14 @@ java_library(
         "//src/devtools/mobileharness/fe/v6/service/proto/common:common_resources_java_proto",
         "//src/devtools/mobileharness/fe/v6/service/proto/config:config_resources_java_proto",
         "//src/java/com/google/devtools/mobileharness/fe/v6/service/config/util:config_converter",
+        "//src/java/com/google/devtools/mobileharness/fe/v6/service/config/util:config_service_capability",
+        "//src/java/com/google/devtools/mobileharness/fe/v6/service/util",
         "//src/javatests/com/google/devtools/mobileharness/builddefs:truth",
+        "//src/javatests/com/google/devtools/mobileharness/shared/util/junit/rule:set_flags_oss",
         "@maven//:com_google_guava_guava",
         "@maven//:com_google_protobuf_protobuf_java",
         "@maven//:junit_junit",
+        "@maven//:org_mockito_mockito_core",
     ],
 )