All files / src/services ConnectionManager.ts

100% Statements 77/77
91.17% Branches 31/34
100% Functions 15/15
100% Lines 77/77

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 17710x 10x 10x 10x 10x               10x 10x 45x 45x 45x 45x 45x 45x 10x 10x         55x 55x       36x 2x     34x 34x 2x       32x   32x           31x 31x     31x 31x   31x 2x 2x 2x   29x     29x         26x 26x     26x           26x     28x   28x           135x 108x               6x 6x         5x 3x   2x       4x 3x   1x       4x 3x   1x       5x 3x   2x       4x 3x   1x       7x 1x   6x       2x 1x   1x       2x 1x   1x       2x 1x   1x       44x 44x 44x 44x 44x 44x      
import { GraphQLClient } from "../graphql/client";
import { GitLabVersionDetector, GitLabInstanceInfo } from "./GitLabVersionDetector";
import { SchemaIntrospector, SchemaInfo } from "./SchemaIntrospector";
import { GITLAB_BASE_URL, GITLAB_TOKEN } from "../config";
import { logger } from "../logger";
 
interface CacheEntry {
  schemaInfo: SchemaInfo;
  instanceInfo: GitLabInstanceInfo;
  timestamp: number;
}
 
export class ConnectionManager {
  private static instance: ConnectionManager | null = null;
  private client: GraphQLClient | null = null;
  private versionDetector: GitLabVersionDetector | null = null;
  private schemaIntrospector: SchemaIntrospector | null = null;
  private instanceInfo: GitLabInstanceInfo | null = null;
  private schemaInfo: SchemaInfo | null = null;
  private isInitialized: boolean = false;
  private static introspectionCache = new Map<string, CacheEntry>();
  private static readonly CACHE_TTL = 10 * 60 * 1000; // 10 minutes in milliseconds
 
  private constructor() {}
 
  public static getInstance(): ConnectionManager {
    ConnectionManager.instance ??= new ConnectionManager();
    return ConnectionManager.instance;
  }
 
  public async initialize(): Promise<void> {
    if (this.isInitialized) {
      return;
    }
 
    try {
      if (!GITLAB_BASE_URL || !GITLAB_TOKEN) {
        throw new Error("GitLab base URL and token are required");
      }
 
      // Construct GraphQL endpoint from base URL
      const endpoint = `${GITLAB_BASE_URL}/api/graphql`;
 
      this.client = new GraphQLClient(endpoint, {
        headers: {
          Authorization: `Bearer ${GITLAB_TOKEN}`,
        },
      });
 
      this.versionDetector = new GitLabVersionDetector(this.client);
      this.schemaIntrospector = new SchemaIntrospector(this.client);
 
      // Check cache first
      const cached = ConnectionManager.introspectionCache.get(endpoint);
      const now = Date.now();
 
      if (cached && now - cached.timestamp < ConnectionManager.CACHE_TTL) {
        logger.info("Using cached GraphQL introspection data");
        this.instanceInfo = cached.instanceInfo;
        this.schemaInfo = cached.schemaInfo;
      } else {
        logger.debug("Introspecting GitLab GraphQL schema...");
 
        // Detect instance info and introspect schema in parallel
        const [instanceInfo, schemaInfo] = await Promise.all([
          this.versionDetector.detectInstance(),
          this.schemaIntrospector.introspectSchema(),
        ]);
 
        this.instanceInfo = instanceInfo;
        this.schemaInfo = schemaInfo;
 
        // Cache the results
        ConnectionManager.introspectionCache.set(endpoint, {
          instanceInfo,
          schemaInfo,
          timestamp: now,
        });
 
        logger.info("GraphQL schema introspection completed");
      }
 
      this.isInitialized = true;
 
      logger.info(
        {
          version: this.instanceInfo?.version,
          tier: this.instanceInfo?.tier,
          features: this.instanceInfo
            ? Object.entries(this.instanceInfo.features)
                .filter(([, enabled]) => enabled)
                .map(([feature]) => feature)
            : [],
          widgetTypes: this.schemaInfo?.workItemWidgetTypes.length || 0,
          schemaTypes: this.schemaInfo?.typeDefinitions.size || 0,
        },
        "GitLab instance and schema detected"
      );
    } catch (error) {
      logger.error({ err: error as Error }, "Failed to initialize connection");
      throw error;
    }
  }
 
  public getClient(): GraphQLClient {
    if (!this.client) {
      throw new Error("Connection not initialized. Call initialize() first.");
    }
    return this.client;
  }
 
  public getVersionDetector(): GitLabVersionDetector {
    if (!this.versionDetector) {
      throw new Error("Connection not initialized. Call initialize() first.");
    }
    return this.versionDetector;
  }
 
  public getSchemaIntrospector(): SchemaIntrospector {
    if (!this.schemaIntrospector) {
      throw new Error("Connection not initialized. Call initialize() first.");
    }
    return this.schemaIntrospector;
  }
 
  public getInstanceInfo(): GitLabInstanceInfo {
    if (!this.instanceInfo) {
      throw new Error("Connection not initialized. Call initialize() first.");
    }
    return this.instanceInfo;
  }
 
  public getSchemaInfo(): SchemaInfo {
    if (!this.schemaInfo) {
      throw new Error("Connection not initialized. Call initialize() first.");
    }
    return this.schemaInfo;
  }
 
  public isFeatureAvailable(feature: keyof GitLabInstanceInfo["features"]): boolean {
    if (!this.instanceInfo) {
      return false;
    }
    return this.instanceInfo.features[feature];
  }
 
  public getTier(): string {
    if (!this.instanceInfo) {
      return "unknown";
    }
    return this.instanceInfo.tier;
  }
 
  public getVersion(): string {
    if (!this.instanceInfo) {
      return "unknown";
    }
    return this.instanceInfo.version;
  }
 
  public isWidgetAvailable(widgetType: string): boolean {
    if (!this.schemaIntrospector) {
      return false;
    }
    return this.schemaIntrospector.isWidgetTypeAvailable(widgetType);
  }
 
  public reset(): void {
    this.client = null;
    this.versionDetector = null;
    this.schemaIntrospector = null;
    this.instanceInfo = null;
    this.schemaInfo = null;
    this.isInitialized = false;
  }
}