/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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
 *
 *      http://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 org.apache.juneau.bean.mcp.v20260728;

import static org.apache.juneau.commons.utils.CollectionUtils.*;
import static org.apache.juneau.commons.utils.Shorts.*;

import java.util.*;

import org.apache.juneau.marshall.*;

/**
 * MCP server capability advertisement returned by {@link ServerDiscoverResult}.
 */
@Marshalled
public class ServerCapabilities {

   private ToolCapability tools;
   private PromptCapability prompts;
   private ResourceCapability resources;
   private LoggingCapability logging;
   private CompletionCapability completions;
   private Map<String,Object> experimental;
   private Map<String,Object> extensions;

   /**
    * Tools capability.
    *
    * @return The capability, or {@code null} if not set.
    */
   public ToolCapability getTools() {
      return tools;
   }

   /**
    * Sets tools capability.
    *
    * @param value The new value.  Can be <jk>null</jk> to unset the property.
    * @return This object (for method chaining).
    */
   public ServerCapabilities setTools(ToolCapability value) {
      tools = value;
      return this;
   }

   /**
    * Prompts capability.
    *
    * @return The capability, or {@code null} if not set.
    */
   public PromptCapability getPrompts() {
      return prompts;
   }

   /**
    * Sets prompts capability.
    *
    * @param value The new value.  Can be <jk>null</jk> to unset the property.
    * @return This object (for method chaining).
    */
   public ServerCapabilities setPrompts(PromptCapability value) {
      prompts = value;
      return this;
   }

   /**
    * Resources capability.
    *
    * @return The capability, or {@code null} if not set.
    */
   public ResourceCapability getResources() {
      return resources;
   }

   /**
    * Sets resources capability.
    *
    * @param value The new value.  Can be <jk>null</jk> to unset the property.
    * @return This object (for method chaining).
    */
   public ServerCapabilities setResources(ResourceCapability value) {
      resources = value;
      return this;
   }

   /**
    * Logging capability.
    *
    * @return The capability, or {@code null} if not set.
    */
   public LoggingCapability getLogging() {
      return logging;
   }

   /**
    * Sets logging capability.
    *
    * @param value The new value.  Can be <jk>null</jk> to unset the property.
    * @return This object (for method chaining).
    */
   public ServerCapabilities setLogging(LoggingCapability value) {
      logging = value;
      return this;
   }

   /**
    * Completions capability.
    *
    * @return The capability, or {@code null} if not set.
    */
   public CompletionCapability getCompletions() {
      return completions;
   }

   /**
    * Sets completions capability.
    *
    * @param value The new value.  Can be <jk>null</jk> to unset the property.
    * @return This object (for method chaining).
    */
   public ServerCapabilities setCompletions(CompletionCapability value) {
      completions = value;
      return this;
   }

   /**
    * Experimental capability extensions.
    *
    * @return The experimental map, or {@code null} if not set.
    */
   public Map<String,Object> getExperimental() {
      return u(experimental);
   }

   /**
    * Sets experimental extensions.
    *
    * @param value The new value.  Can be <jk>null</jk> to unset the property.
    * @return This object (for method chaining).
    */
   public ServerCapabilities setExperimental(Map<String,Object> value) {
      experimental = value;
      return this;
   }

   /**
    * Convenience method to add a single experimental extension entry.
    *
    * @param name The extension name.  Can be <jk>null</jk> ({@link LinkedHashMap} tolerates a <jk>null</jk> key).
    * @param value The extension value.  Can be <jk>null</jk> (stored as <jk>null</jk>).
    * @return This object (for method chaining).
    */
   public ServerCapabilities putExperimental(String name, Object value) {
      if (experimental == null)
         experimental = map();
      experimental.put(name, value);
      return this;
   }

   /**
    * Schema-defined capability extensions (distinct from the free-form {@code experimental} bag).
    *
    * @return The extensions map, or {@code null} if not set.
    */
   public Map<String,Object> getExtensions() {
      return u(extensions);
   }

   /**
    * Sets capability extensions.
    *
    * @param value The new value.  Can be <jk>null</jk> to unset the property.
    * @return This object (for method chaining).
    */
   public ServerCapabilities setExtensions(Map<String,Object> value) {
      extensions = value;
      return this;
   }

   /**
    * Convenience method to add a single capability extension entry.
    *
    * <p>
    * The extension map is lazily initialized on first use and preserves insertion order across repeated calls.
    *
    * @param name The extension name.  Can be <jk>null</jk> ({@link LinkedHashMap} tolerates a <jk>null</jk> key).
    * @param value The extension value.  Can be <jk>null</jk> (stored as <jk>null</jk>).
    * @return This object (for method chaining).
    */
   public ServerCapabilities putExtensions(String name, Object value) {
      if (extensions == null)
         extensions = map();
      extensions.put(name, value);
      return this;
   }
}