All files / src/services WidgetAvailability.ts

94.54% Statements 52/55
92.85% Branches 26/28
100% Functions 6/6
96.15% Lines 50/52

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 234 235 23624x 24x   24x 24x 24x                                                   24x                                                             24x 24x                                                                                     206x   206x 206x 175x   175x   1x       174x 174x 7x       167x 90x     77x 77x   77x     31x         4x   120x         5x                                   14x         14x 14x 13x 13x     1x     13x 13x         13x   23x   19x 19x   16x 16x     16x 16x 3x                     13x 8x 8x   8x 3x                       7x             3x      
import { WorkItemWidgetType, WorkItemWidgetTypes } from '../graphql/workItems';
import { ConnectionManager } from './ConnectionManager';
import { GitLabTier } from './GitLabVersionDetector';
import { parseVersion } from '../utils/version';
import { effectiveMinVersion, TIER_ORDER } from './InstanceCapabilities';
import { logDebug } from '../logger';
 
interface WidgetRequirement {
  tier: GitLabTier;
  /** Declared only when newer than MIN_SUPPORTED_VERSION. */
  minVersion?: string;
}
 
/**
 * Result of a widget parameter availability check
 */
export interface WidgetValidationFailure {
  parameter: string;
  widget: WorkItemWidgetType;
  requiredVersion: string;
  detectedVersion: string;
  requiredTier: GitLabTier;
  currentTier: GitLabTier | null;
}
 
/**
 * 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,
  verificationStatus: WorkItemWidgetTypes.VERIFICATION_STATUS,
};
 
export class WidgetAvailability {
  private static widgetRequirements: Record<WorkItemWidgetType, WidgetRequirement> = {
    // Free tier widgets (available to all)
    [WorkItemWidgetTypes.ASSIGNEES]: { tier: 'free' },
    [WorkItemWidgetTypes.DESCRIPTION]: { tier: 'free' },
    [WorkItemWidgetTypes.HIERARCHY]: { tier: 'free' },
    [WorkItemWidgetTypes.LABELS]: { tier: 'free' },
    [WorkItemWidgetTypes.MILESTONE]: { tier: 'free' },
    [WorkItemWidgetTypes.NOTES]: { tier: 'free' },
    [WorkItemWidgetTypes.START_AND_DUE_DATE]: { tier: 'free' },
    [WorkItemWidgetTypes.STATUS]: { tier: 'free' },
    [WorkItemWidgetTypes.NOTIFICATIONS]: { tier: 'free' },
    [WorkItemWidgetTypes.CURRENT_USER_TODOS]: { tier: 'free' },
    [WorkItemWidgetTypes.AWARD_EMOJI]: { tier: 'free' },
    [WorkItemWidgetTypes.PARTICIPANTS]: { tier: 'free' },
    [WorkItemWidgetTypes.DESIGNS]: { tier: 'free' },
    [WorkItemWidgetTypes.DEVELOPMENT]: { tier: 'free' },
    [WorkItemWidgetTypes.TIME_TRACKING]: { tier: 'free' },
    [WorkItemWidgetTypes.ERROR_TRACKING]: { tier: 'free' },
 
    // Free tier widgets (linked items available on CE)
    [WorkItemWidgetTypes.LINKED_ITEMS]: { tier: 'free' },
 
    // Premium tier widgets
    [WorkItemWidgetTypes.WEIGHT]: { tier: 'premium' },
    [WorkItemWidgetTypes.ITERATION]: { tier: 'premium' },
    [WorkItemWidgetTypes.PROGRESS]: { tier: 'premium' },
    [WorkItemWidgetTypes.CRM_CONTACTS]: { tier: 'premium' },
    [WorkItemWidgetTypes.EMAIL_PARTICIPANTS]: { tier: 'premium' },
    [WorkItemWidgetTypes.LINKED_RESOURCES]: { tier: 'premium', minVersion: '16.5' },
 
    // Ultimate tier widgets
    [WorkItemWidgetTypes.HEALTH_STATUS]: { tier: 'ultimate' },
    [WorkItemWidgetTypes.COLOR]: { tier: 'ultimate' },
    [WorkItemWidgetTypes.CUSTOM_FIELDS]: { tier: 'ultimate', minVersion: '17.0' },
    [WorkItemWidgetTypes.VULNERABILITIES]: { tier: 'ultimate' },
 
    // Legacy widgets (may not be available)
    [WorkItemWidgetTypes.REQUIREMENT_LEGACY]: { tier: 'ultimate' },
    [WorkItemWidgetTypes.TEST_REPORTS]: { tier: 'ultimate' },
    [WorkItemWidgetTypes.VERIFICATION_STATUS]: { tier: 'ultimate' },
  };
 
  public static isWidgetAvailable(widget: WorkItemWidgetType, instanceUrl?: string): boolean {
    const connectionManager = ConnectionManager.getInstance();
 
    try {
      const instanceInfo = connectionManager.getInstanceInfo(instanceUrl);
      const requirement = this.widgetRequirements[widget];
 
      if (!requirement) {
        // Unknown widget, assume not available
        return false;
      }
 
      // Check version requirement
      const version = parseVersion(instanceInfo.version);
      if (version < parseVersion(effectiveMinVersion(requirement))) {
        return false;
      }
 
      // Check tier requirement
      if (requirement.tier === 'free' || instanceInfo.tier === null) {
        return true; // No confirmed subscription restriction.
      }
 
      const requiredTierLevel = TIER_ORDER[requirement.tier];
      const actualTierLevel = TIER_ORDER[instanceInfo.tier];
 
      return actualTierLevel >= requiredTierLevel;
    } catch {
      // If connection not initialized, assume widget not available
      return false;
    }
  }
 
  public static getAvailableWidgets(instanceUrl?: string): WorkItemWidgetType[] {
    return Object.values(WorkItemWidgetTypes).filter(
      (widget): widget is WorkItemWidgetType =>
        typeof widget === 'string' && this.isWidgetAvailable(widget, instanceUrl),
    );
  }
 
  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)
   * @param instanceUrl - Optional instance URL to validate against. When omitted,
   *   falls back to ConnectionManager.currentInstanceUrl. Callers in request context
   *   (e.g. workitems handler) should pass the per-request URL to avoid cross-instance
   *   leakage in concurrent OAuth traffic.
   * @returns WidgetValidationFailure for the first unavailable parameter, or null if all valid
   */
  public static validateWidgetParams(
    params: Record<string, unknown>,
    instanceUrl?: string,
  ): WidgetValidationFailure | null {
    const connectionManager = ConnectionManager.getInstance();
 
    let instanceVersion: string;
    let instanceTier: GitLabTier | null;
 
    try {
      const instanceInfo = connectionManager.getInstanceInfo(instanceUrl);
      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 requiredVersion = effectiveMinVersion(requirement);
      if (parsedVersion < parseVersion(requiredVersion)) {
        return {
          parameter: paramName,
          widget: widgetType,
          requiredVersion,
          detectedVersion: instanceVersion,
          requiredTier: requirement.tier,
          currentTier: instanceTier,
        };
      }
 
      // Check tier requirement
      if (requirement.tier !== 'free' && instanceTier !== null) {
        const requiredTierLevel = TIER_ORDER[requirement.tier];
        const actualTierLevel = TIER_ORDER[instanceTier];
 
        if (actualTierLevel < requiredTierLevel) {
          return {
            parameter: paramName,
            widget: widgetType,
            requiredVersion,
            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 };
  }
}