All files / src/services WidgetAvailability.ts

94.64% Statements 53/56
91.66% Branches 22/24
100% Functions 6/6
96.22% Lines 51/53

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 23415x 15x   15x 15x                                     15x                       15x                                                           15x 15x                                                                                   198x   198x 198x 168x   168x   1x       167x 167x 167x 6x       161x 89x     72x 72x   72x     30x         4x   116x         4x                         11x         11x 11x 10x 10x     1x     10x 10x         10x   20x   16x 16x   13x 13x     13x 13x 3x                     10x 6x 6x   6x 2x                       5x             3x      
import { WorkItemWidgetType, WorkItemWidgetTypes } from "../graphql/workItems";
import { ConnectionManager } from "./ConnectionManager";
import { GitLabTier } from "./GitLabVersionDetector";
import { parseVersion } from "../utils/version";
import { logDebug } from "../logger";
 
interface WidgetRequirement {
  tier: GitLabTier;
  minVersion: string;
}
 
/**
 * Result of a widget parameter availability check
 */
export interface WidgetValidationFailure {
  parameter: string;
  widget: WorkItemWidgetType;
  requiredVersion: string;
  detectedVersion: string;
  requiredTier: GitLabTier;
  currentTier: GitLabTier;
}
 
const TIER_HIERARCHY: Record<GitLabTier, number> = {
  free: 0,
  premium: 1,
  ultimate: 2,
};
 
/**
 * Maps manage_work_item input parameters to their corresponding widget types.
 * Includes both current schema parameters and those planned for #135.
 * Parameters not yet in ManageWorkItemSchema are harmless here — validation
 * only triggers when the parameter is actually present in the handler input.
 */
const PARAMETER_WIDGET_MAP: Record<string, WorkItemWidgetType> = {
  // Basic widget parameters
  assigneeIds: WorkItemWidgetTypes.ASSIGNEES,
  labelIds: WorkItemWidgetTypes.LABELS,
  addLabelIds: WorkItemWidgetTypes.LABELS,
  removeLabelIds: WorkItemWidgetTypes.LABELS,
  milestoneId: WorkItemWidgetTypes.MILESTONE,
  description: WorkItemWidgetTypes.DESCRIPTION,
  // Free tier: dates
  startDate: WorkItemWidgetTypes.START_AND_DUE_DATE,
  dueDate: WorkItemWidgetTypes.START_AND_DUE_DATE,
  isFixed: WorkItemWidgetTypes.START_AND_DUE_DATE,
  // Free tier: hierarchy
  parentId: WorkItemWidgetTypes.HIERARCHY,
  childrenIds: WorkItemWidgetTypes.HIERARCHY,
  // Free tier: time tracking
  timeEstimate: WorkItemWidgetTypes.TIME_TRACKING,
  timeSpent: WorkItemWidgetTypes.TIME_TRACKING,
  // Free tier: linked items
  linkType: WorkItemWidgetTypes.LINKED_ITEMS,
  targetId: WorkItemWidgetTypes.LINKED_ITEMS,
  // Premium tier
  weight: WorkItemWidgetTypes.WEIGHT,
  iterationId: WorkItemWidgetTypes.ITERATION,
  progressCurrentValue: WorkItemWidgetTypes.PROGRESS,
  // Ultimate tier
  healthStatus: WorkItemWidgetTypes.HEALTH_STATUS,
  color: WorkItemWidgetTypes.COLOR,
};
 
export class WidgetAvailability {
  private static widgetRequirements: Record<WorkItemWidgetType, WidgetRequirement> = {
    // Free tier widgets (available to all)
    [WorkItemWidgetTypes.ASSIGNEES]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.DESCRIPTION]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.HIERARCHY]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.LABELS]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.MILESTONE]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.NOTES]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.START_AND_DUE_DATE]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.STATUS]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.NOTIFICATIONS]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.CURRENT_USER_TODOS]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.AWARD_EMOJI]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.PARTICIPANTS]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.DESIGNS]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.DEVELOPMENT]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.TIME_TRACKING]: { tier: "free", minVersion: "15.0" },
    [WorkItemWidgetTypes.ERROR_TRACKING]: { tier: "free", minVersion: "15.0" },
 
    // Free tier widgets (linked items available on CE)
    [WorkItemWidgetTypes.LINKED_ITEMS]: { tier: "free", minVersion: "15.0" },
 
    // Premium tier widgets
    [WorkItemWidgetTypes.WEIGHT]: { tier: "premium", minVersion: "15.0" },
    [WorkItemWidgetTypes.ITERATION]: { tier: "premium", minVersion: "15.0" },
    [WorkItemWidgetTypes.PROGRESS]: { tier: "premium", minVersion: "15.0" },
    [WorkItemWidgetTypes.CRM_CONTACTS]: { tier: "premium", minVersion: "16.0" },
    [WorkItemWidgetTypes.EMAIL_PARTICIPANTS]: { tier: "premium", minVersion: "16.0" },
    [WorkItemWidgetTypes.LINKED_RESOURCES]: { tier: "premium", minVersion: "16.5" },
 
    // Ultimate tier widgets
    [WorkItemWidgetTypes.HEALTH_STATUS]: { tier: "ultimate", minVersion: "15.0" },
    [WorkItemWidgetTypes.COLOR]: { tier: "ultimate", minVersion: "15.0" },
    [WorkItemWidgetTypes.CUSTOM_FIELDS]: { tier: "ultimate", minVersion: "17.0" },
    [WorkItemWidgetTypes.VULNERABILITIES]: { tier: "ultimate", minVersion: "15.0" },
 
    // Legacy widgets (may not be available)
    [WorkItemWidgetTypes.REQUIREMENT_LEGACY]: { tier: "ultimate", minVersion: "13.1" },
    [WorkItemWidgetTypes.TEST_REPORTS]: { tier: "ultimate", minVersion: "13.6" },
  };
 
  public static isWidgetAvailable(widget: WorkItemWidgetType): boolean {
    const connectionManager = ConnectionManager.getInstance();
 
    try {
      const instanceInfo = connectionManager.getInstanceInfo();
      const requirement = this.widgetRequirements[widget];
 
      if (!requirement) {
        // Unknown widget, assume not available
        return false;
      }
 
      // Check version requirement
      const version = parseVersion(instanceInfo.version);
      const minVersion = parseVersion(requirement.minVersion);
      if (version < minVersion) {
        return false;
      }
 
      // Check tier requirement
      if (requirement.tier === "free") {
        return true; // Available to all tiers
      }
 
      const requiredTierLevel = TIER_HIERARCHY[requirement.tier];
      const actualTierLevel = TIER_HIERARCHY[instanceInfo.tier];
 
      return actualTierLevel >= requiredTierLevel;
    } catch {
      // If connection not initialized, assume widget not available
      return false;
    }
  }
 
  public static getAvailableWidgets(): WorkItemWidgetType[] {
    return Object.values(WorkItemWidgetTypes).filter(
      (widget): widget is WorkItemWidgetType =>
        typeof widget === "string" && this.isWidgetAvailable(widget as WorkItemWidgetType)
    );
  }
 
  public static getWidgetRequirement(widget: WorkItemWidgetType): WidgetRequirement | undefined {
    return this.widgetRequirements[widget];
  }
 
  /**
   * Validate widget parameters against the detected GitLab instance version and tier.
   * Returns the first unavailable widget parameter, or null if all are available.
   *
   * @param params - Object with parameter names as keys (only defined/present params checked)
   * @returns WidgetValidationFailure for the first unavailable parameter, or null if all valid
   */
  public static validateWidgetParams(
    params: Record<string, unknown>
  ): WidgetValidationFailure | null {
    const connectionManager = ConnectionManager.getInstance();
 
    let instanceVersion: string;
    let instanceTier: GitLabTier;
 
    try {
      const instanceInfo = connectionManager.getInstanceInfo();
      instanceVersion = instanceInfo.version;
      instanceTier = instanceInfo.tier;
    } catch {
      // Connection not initialized - skip validation (will fail at API call)
      return null;
    }
 
    const parsedVersion = parseVersion(instanceVersion);
    Iif (parsedVersion === 0) {
      logDebug("Widget param validation skipped: version could not be parsed", { instanceVersion });
      return null;
    }
 
    for (const [paramName, paramValue] of Object.entries(params)) {
      // Skip undefined/null parameters (not provided by user)
      if (paramValue === undefined || paramValue === null) continue;
 
      const widgetType = PARAMETER_WIDGET_MAP[paramName];
      if (!widgetType) continue; // Not a widget parameter
 
      const requirement = this.widgetRequirements[widgetType];
      Iif (!requirement) continue; // Unknown widget
 
      // Check version requirement
      const minVersion = parseVersion(requirement.minVersion);
      if (parsedVersion < minVersion) {
        return {
          parameter: paramName,
          widget: widgetType,
          requiredVersion: requirement.minVersion,
          detectedVersion: instanceVersion,
          requiredTier: requirement.tier,
          currentTier: instanceTier,
        };
      }
 
      // Check tier requirement
      if (requirement.tier !== "free") {
        const requiredTierLevel = TIER_HIERARCHY[requirement.tier];
        const actualTierLevel = TIER_HIERARCHY[instanceTier];
 
        if (actualTierLevel < requiredTierLevel) {
          return {
            parameter: paramName,
            widget: widgetType,
            requiredVersion: requirement.minVersion,
            detectedVersion: instanceVersion,
            requiredTier: requirement.tier,
            currentTier: instanceTier,
          };
        }
      }
    }
 
    return null;
  }
 
  /**
   * Get the parameter-to-widget mapping (for testing and external use)
   */
  public static getParameterWidgetMap(): Record<string, WorkItemWidgetType> {
    return { ...PARAMETER_WIDGET_MAP };
  }
}