Add a fboss2 config interface <name> description <desc> command.

Add some helper code to process interface-list arguments.

Author: benoit-nexthop

cmake/CliFboss2.cmake

@@ -577,6 +577,9 @@ add_library(fboss2_config_lib
   fboss/cli/fboss2/commands/config/CmdConfigAppliedInfo.cpp
   fboss/cli/fboss2/commands/config/CmdConfigReload.h
   fboss/cli/fboss2/commands/config/CmdConfigReload.cpp
+  fboss/cli/fboss2/commands/config/interface/CmdConfigInterface.h
+  fboss/cli/fboss2/commands/config/interface/CmdConfigInterfaceDescription.h
+  fboss/cli/fboss2/commands/config/interface/CmdConfigInterfaceDescription.cpp
   fboss/cli/fboss2/commands/config/history/CmdConfigHistory.h
   fboss/cli/fboss2/commands/config/history/CmdConfigHistory.cpp
   fboss/cli/fboss2/commands/config/rollback/CmdConfigRollback.h
@@ -587,6 +590,8 @@ add_library(fboss2_config_lib
   fboss/cli/fboss2/commands/config/session/CmdConfigSessionDiff.cpp
   fboss/cli/fboss2/session/ConfigSession.h
   fboss/cli/fboss2/session/ConfigSession.cpp
+  fboss/cli/fboss2/utils/InterfaceList.h
+  fboss/cli/fboss2/utils/InterfaceList.cpp
   fboss/cli/fboss2/CmdListConfig.cpp
   fboss/cli/fboss2/CmdHandlerImplConfig.cpp
 )

cmake/CliFboss2Test.cmake

@@ -35,6 +35,7 @@ add_executable(fboss2_cmd_test
   fboss/cli/fboss2/test/TestMain.cpp
   fboss/cli/fboss2/test/CmdConfigAppliedInfoTest.cpp
   fboss/cli/fboss2/test/CmdConfigHistoryTest.cpp
+  fboss/cli/fboss2/test/CmdConfigInterfaceDescriptionTest.cpp
   fboss/cli/fboss2/test/CmdConfigReloadTest.cpp
   fboss/cli/fboss2/test/CmdConfigSessionDiffTest.cpp
   fboss/cli/fboss2/test/CmdConfigSessionTest.cpp

fboss/cli/fboss2/BUCK

@@ -783,19 +783,24 @@ cpp_library(
         "commands/config/CmdConfigAppliedInfo.cpp",
         "commands/config/CmdConfigReload.cpp",
         "commands/config/history/CmdConfigHistory.cpp",
+        "commands/config/interface/CmdConfigInterfaceDescription.cpp",
         "commands/config/rollback/CmdConfigRollback.cpp",
         "commands/config/session/CmdConfigSessionCommit.cpp",
         "commands/config/session/CmdConfigSessionDiff.cpp",
         "session/ConfigSession.cpp",
+        "utils/InterfaceList.cpp",
     ],
     headers = [
         "commands/config/CmdConfigAppliedInfo.h",
         "commands/config/CmdConfigReload.h",
         "commands/config/history/CmdConfigHistory.h",
+        "commands/config/interface/CmdConfigInterface.h",
+        "commands/config/interface/CmdConfigInterfaceDescription.h",
         "commands/config/rollback/CmdConfigRollback.h",
         "commands/config/session/CmdConfigSessionCommit.h",
         "commands/config/session/CmdConfigSessionDiff.h",
         "session/ConfigSession.h",
+        "utils/InterfaceList.h",
     ],
     exported_deps = [
         "fbsource//third-party/fmt:fmt",

fboss/cli/fboss2/CmdHandler.h

@@ -169,6 +169,8 @@ class CmdHandler {
     RetType result;
     try {
       result = queryClientHelper(hostInfo);
+    } catch (std::invalid_argument const& err) {
+      errStr = folly::to<std::string>("Invalid argument: ", err.what());
     } catch (std::exception const& err) {
       errStr = folly::to<std::string>("Thrift call failed: '", err.what(), "'");
     }

fboss/cli/fboss2/CmdHandlerImplConfig.cpp

@@ -16,6 +16,8 @@
 #include "fboss/cli/fboss2/commands/config/CmdConfigAppliedInfo.h"
 #include "fboss/cli/fboss2/commands/config/CmdConfigReload.h"
 #include "fboss/cli/fboss2/commands/config/history/CmdConfigHistory.h"
+#include "fboss/cli/fboss2/commands/config/interface/CmdConfigInterface.h"
+#include "fboss/cli/fboss2/commands/config/interface/CmdConfigInterfaceDescription.h"
 #include "fboss/cli/fboss2/commands/config/rollback/CmdConfigRollback.h"
 #include "fboss/cli/fboss2/commands/config/session/CmdConfigSessionCommit.h"
 #include "fboss/cli/fboss2/commands/config/session/CmdConfigSessionDiff.h"
@@ -25,6 +27,10 @@ namespace facebook::fboss {
 template void
 CmdHandler<CmdConfigAppliedInfo, CmdConfigAppliedInfoTraits>::run();
 template void CmdHandler<CmdConfigReload, CmdConfigReloadTraits>::run();
+template void CmdHandler<CmdConfigInterface, CmdConfigInterfaceTraits>::run();
+template void CmdHandler<
+    CmdConfigInterfaceDescription,
+    CmdConfigInterfaceDescriptionTraits>::run();
 template void CmdHandler<CmdConfigHistory, CmdConfigHistoryTraits>::run();
 template void CmdHandler<CmdConfigRollback, CmdConfigRollbackTraits>::run();
 template void

fboss/cli/fboss2/CmdListConfig.cpp

@@ -14,6 +14,8 @@
 #include "fboss/cli/fboss2/commands/config/CmdConfigAppliedInfo.h"
 #include "fboss/cli/fboss2/commands/config/CmdConfigReload.h"
 #include "fboss/cli/fboss2/commands/config/history/CmdConfigHistory.h"
+#include "fboss/cli/fboss2/commands/config/interface/CmdConfigInterface.h"
+#include "fboss/cli/fboss2/commands/config/interface/CmdConfigInterfaceDescription.h"
 #include "fboss/cli/fboss2/commands/config/rollback/CmdConfigRollback.h"
 #include "fboss/cli/fboss2/commands/config/session/CmdConfigSessionCommit.h"
 #include "fboss/cli/fboss2/commands/config/session/CmdConfigSessionDiff.h"
@@ -34,6 +36,20 @@ const CommandTree& kConfigCommandTree() {
        commandHandler<CmdConfigHistory>,
        argTypeHandler<CmdConfigHistoryTraits>},
 
+      {
+          "config",
+          "interface",
+          "Configure interface settings",
+          commandHandler<CmdConfigInterface>,
+          argTypeHandler<CmdConfigInterfaceTraits>,
+          {{
+              "description",
+              "Set interface description",
+              commandHandler<CmdConfigInterfaceDescription>,
+              argTypeHandler<CmdConfigInterfaceDescriptionTraits>,
+          }},
+      },
+
       {
           "config",
           "session",

fboss/cli/fboss2/CmdSubcommands.cpp

@@ -219,6 +219,9 @@ CLI::App* CmdSubcommands::addCommand(
         case utils::ObjectArgTypeId::OBJECT_ARG_TYPE_FAN_PWM:
           subCmd->add_option("pwm", args, "Fan PWM (0..100) or 'disable'");
           break;
+        case utils::ObjectArgTypeId::OBJECT_ARG_TYPE_ID_INTERFACE_LIST:
+          subCmd->add_option("interfaces", args, "Interface(s)");
+          break;
         case utils::ObjectArgTypeId::OBJECT_ARG_TYPE_ID_REVISION_LIST:
           subCmd->add_option(
               "revisions", args, "Revision(s) in the form 'rN' or 'current'");

fboss/cli/fboss2/commands/config/interface/CmdConfigInterface.h

@@ -0,0 +1,38 @@
+/*
+ *  Copyright (c) 2004-present, Facebook, Inc.
+ *  All rights reserved.
+ *
+ *  This source code is licensed under the BSD-style license found in the
+ *  LICENSE file in the root directory of this source tree. An additional grant
+ *  of patent rights can be found in the PATENTS file in the same directory.
+ *
+ */
+
+#pragma once
+
+#include "fboss/cli/fboss2/CmdHandler.h"
+#include "fboss/cli/fboss2/utils/CmdUtils.h"
+
+namespace facebook::fboss {
+
+struct CmdConfigInterfaceTraits : public WriteCommandTraits {
+  static constexpr utils::ObjectArgTypeId ObjectArgTypeId =
+      utils::ObjectArgTypeId::OBJECT_ARG_TYPE_ID_PORT_LIST;
+  using ObjectArgType = std::vector<std::string>;
+  using RetType = std::string;
+};
+
+class CmdConfigInterface
+    : public CmdHandler<CmdConfigInterface, CmdConfigInterfaceTraits> {
+ public:
+  RetType queryClient(
+      const HostInfo& /* hostInfo */,
+      const ObjectArgType& /* interfaceNames */) {
+    throw std::runtime_error(
+        "Incomplete command, please use one of the subcommands");
+  }
+
+  void printOutput(const RetType& /* model */) {}
+};
+
+} // namespace facebook::fboss

fboss/cli/fboss2/commands/config/interface/CmdConfigInterfaceDescription.cpp

@@ -0,0 +1,48 @@
+/*
+ *  Copyright (c) 2004-present, Facebook, Inc.
+ *  All rights reserved.
+ *
+ *  This source code is licensed under the BSD-style license found in the
+ *  LICENSE file in the root directory of this source tree. An additional grant
+ *  of patent rights can be found in the PATENTS file in the same directory.
+ *
+ */
+
+#include "fboss/cli/fboss2/commands/config/interface/CmdConfigInterfaceDescription.h"
+
+#include <folly/Conv.h>
+#include "fboss/cli/fboss2/session/ConfigSession.h"
+
+namespace facebook::fboss {
+
+CmdConfigInterfaceDescriptionTraits::RetType
+CmdConfigInterfaceDescription::queryClient(
+    const HostInfo& hostInfo,
+    const utils::InterfaceList& interfaces,
+    const ObjectArgType& description) {
+  if (interfaces.empty()) {
+    throw std::invalid_argument("No interface name provided");
+  }
+
+  std::string descriptionStr = description.data()[0];
+
+  // Update description for all resolved ports
+  for (const utils::Intf& intf : interfaces) {
+    cfg::Port* port = intf.getPort();
+    if (port) {
+      port->description() = descriptionStr;
+    }
+  }
+
+  // Save the updated config
+  ConfigSession::getInstance().saveConfig();
+
+  std::string interfaceList = folly::join(", ", interfaces.getNames());
+  return "Successfully set description for interface(s) " + interfaceList;
+}
+
+void CmdConfigInterfaceDescription::printOutput(const RetType& logMsg) {
+  std::cout << logMsg << std::endl;
+}
+
+} // namespace facebook::fboss

fboss/cli/fboss2/commands/config/interface/CmdConfigInterfaceDescription.h

@@ -0,0 +1,43 @@
+/*
+ *  Copyright (c) 2004-present, Facebook, Inc.
+ *  All rights reserved.
+ *
+ *  This source code is licensed under the BSD-style license found in the
+ *  LICENSE file in the root directory of this source tree. An additional grant
+ *  of patent rights can be found in the PATENTS file in the same directory.
+ *
+ */
+
+#pragma once
+
+#include "fboss/cli/fboss2/CmdHandler.h"
+#include "fboss/cli/fboss2/commands/config/interface/CmdConfigInterface.h"
+#include "fboss/cli/fboss2/utils/CmdUtils.h"
+#include "fboss/cli/fboss2/utils/InterfaceList.h"
+
+namespace facebook::fboss {
+
+struct CmdConfigInterfaceDescriptionTraits : public WriteCommandTraits {
+  using ParentCmd = CmdConfigInterface;
+  static constexpr utils::ObjectArgTypeId ObjectArgTypeId =
+      utils::ObjectArgTypeId::OBJECT_ARG_TYPE_ID_MESSAGE;
+  using ObjectArgType = utils::Message;
+  using RetType = std::string;
+};
+
+class CmdConfigInterfaceDescription : public CmdHandler<
+                                          CmdConfigInterfaceDescription,
+                                          CmdConfigInterfaceDescriptionTraits> {
+ public:
+  using ObjectArgType = CmdConfigInterfaceDescriptionTraits::ObjectArgType;
+  using RetType = CmdConfigInterfaceDescriptionTraits::RetType;
+
+  RetType queryClient(
+      const HostInfo& hostInfo,
+      const utils::InterfaceList& interfaces,
+      const ObjectArgType& description);
+
+  void printOutput(const RetType& logMsg);
+};
+
+} // namespace facebook::fboss