duoops 0.2.2 → 0.2.4

package/mcp-server/src/lib/zone-mapper.ts

@@ -0,0 +1,63 @@
+import fs from 'node:fs'
+import path from 'node:path'
+
+import {ZoneMapping} from './types.js'
+
+type ZoneData = {
+  carbon_intensity_url?: string;
+  pue?: number;
+  zone: string;
+}
+
+type CloudZones = Record<string, ZoneData>
+
+export class ZoneMapper {
+  private awsZones: CloudZones | null = null
+  private gcpZones: CloudZones | null = null
+
+  constructor(private readonly dataDir: string = path.resolve(process.cwd(), 'data')) {}
+
+  getZoneMapping(provider: string, region: string): null | ZoneMapping {
+    const zones = this.loadZoneData(provider)
+    const data = zones[region]
+
+    if (!data) return null
+
+    return {
+      carbonIntensityUrl: data.carbon_intensity_url || '',
+      pue: data.pue || 1.2, // Default PUE fallback
+      zone: data.zone,
+    }
+  }
+
+  private loadJsonFile<T>(filename: string): T {
+    const filePath = path.join(this.dataDir, filename)
+    try {
+      const raw = fs.readFileSync(filePath, 'utf8')
+      return JSON.parse(raw) as T
+    } catch (error) {
+      console.error(`Failed to load data file ${filename}:`, error)
+      return {} as T
+    }
+  }
+
+  private loadZoneData(provider: string): CloudZones {
+    if (provider === 'gcp') {
+      if (!this.gcpZones) {
+        this.gcpZones = this.loadJsonFile<CloudZones>('gcp_zones.json')
+      }
+
+      return this.gcpZones
+    }
+
+    if (provider === 'aws') {
+      if (!this.awsZones) {
+        this.awsZones = this.loadJsonFile<CloudZones>('aws_zones.json')
+      }
+
+      return this.awsZones
+    }
+
+    return {}
+  }
+}

package/mcp-server/src/tools/emissions.ts

@@ -0,0 +1,107 @@
+/* eslint-disable camelcase */
+import {McpServer} from '@modelcontextprotocol/sdk/server/mcp.js'
+import {z} from 'zod'
+
+import {BigQueryService} from '../lib/bigquery.js'
+
+export function registerEmissionsTools(server: McpServer) {
+  const bq = new BigQueryService()
+
+  server.tool(
+    'get_job_emissions',
+    {
+      job_id: z.number().describe('GitLab CI Job ID'),
+      project_id: z.number().optional().describe('GitLab Project ID (optional filter)'),
+    },
+    async ({job_id, project_id}) => {
+      const emissions = await bq.getJobEmissions(job_id, project_id)
+      
+      if (!emissions) {
+        return {
+          content: [{text: `No emissions data found for job ${job_id}`, type: 'text'}],
+          isError: true,
+        }
+      }
+
+      return {
+        content: [{
+          text: JSON.stringify(emissions, null, 2),
+          type: 'text'
+        }]
+      }
+    }
+  )
+
+  server.tool(
+    'get_pipeline_emissions',
+    {
+      pipeline_id: z.number().describe('GitLab Pipeline ID'),
+      project_id: z.number().optional().describe('GitLab Project ID (optional filter)'),
+    },
+    async ({pipeline_id, project_id}) => {
+      const result = await bq.getPipelineEmissions(pipeline_id, project_id)
+      
+      if (result.jobs.length === 0) {
+        return {
+          content: [{text: `No emissions data found for pipeline ${pipeline_id}`, type: 'text'}],
+          isError: true,
+        }
+      }
+
+      const summary = `
+Pipeline ${pipeline_id} Carbon Report:
+Total Emissions: ${result.total_emissions_g.toFixed(3)} gCO2e
+Total Energy: ${result.total_energy_kwh.toFixed(4)} kWh
+Jobs Recorded: ${result.jobs.length}
+
+Top Emitters:
+${result.jobs
+  .sort((a, b) => (b.total_emissions || 0) - (a.total_emissions || 0))
+  .slice(0, 3)
+  .map(j => `- Job ${j.job_id} (${j.machine_type}): ${j.total_emissions?.toFixed(3)} gCO2e`)
+  .join('\n')}
+`
+
+      return {
+        content: [
+          {text: summary, type: 'text'},
+          {text: JSON.stringify(result, null, 2), type: 'text'} // Full JSON also provided
+        ]
+      }
+    }
+  )
+
+  server.tool(
+    'get_project_emissions_summary',
+    {
+      limit: z.number().optional().default(10).describe('Number of recent jobs to return'),
+      project_id: z.number().describe('GitLab Project ID'),
+    },
+    async ({limit, project_id}) => {
+      const records = await bq.getProjectEmissionsSummary(project_id, limit)
+      
+      if (records.length === 0) {
+        return {
+          content: [{text: `No emissions history found for project ${project_id}`, type: 'text'}],
+          isError: true,
+        }
+      }
+
+      const totalEmissions = records.reduce((sum, r) => sum + r.total_emissions, 0)
+      const avgEmissions = totalEmissions / records.length
+
+      const summary = `
+Project ${project_id} Emissions Summary (Last ${records.length} jobs):
+Average per job: ${avgEmissions.toFixed(3)} gCO2e
+Total recent: ${totalEmissions.toFixed(3)} gCO2e
+
+Recent Jobs:
+${records.map(r => `- ${r.created_at}: ${r.total_emissions.toFixed(3)} gCO2e (${r.machine_type})`).join('\n')}
+`
+
+      return {
+        content: [{text: summary, type: 'text'}]
+      }
+    }
+  )
+}

package/mcp-server/src/tools/profiles.ts

@@ -0,0 +1,72 @@
+/* eslint-disable camelcase */
+import {McpServer} from '@modelcontextprotocol/sdk/server/mcp.js'
+import {z} from 'zod'
+
+import {IntensityProvider} from '../lib/intensity.js'
+import {PowerProfileRepository} from '../lib/power-profiles.js'
+import {ZoneMapper} from '../lib/zone-mapper.js'
+
+export function registerProfileTools(server: McpServer) {
+  const powerRepo = new PowerProfileRepository()
+  const zoneMapper = new ZoneMapper()
+  const intensityProvider = new IntensityProvider()
+
+  server.tool(
+    'get_machine_profile',
+    {
+      machine_type: z.string().describe('Instance type name (e.g. e2-standard-4)'),
+      provider: z.enum(['aws', 'gcp']).describe('Cloud provider (aws or gcp)'),
+    },
+    async ({machine_type, provider}) => {
+      const profile = await powerRepo.getMachineProfile(provider, machine_type)
+      
+      if (!profile) {
+        return {
+          content: [{text: `Unknown machine type: ${machine_type} for provider ${provider}`, type: 'text'}],
+          isError: true,
+        }
+      }
+
+      return {
+        content: [{
+          text: JSON.stringify(profile, null, 2),
+          type: 'text'
+        }]
+      }
+    }
+  )
+
+  server.tool(
+    'get_region_carbon_intensity',
+    {
+      provider: z.enum(['aws', 'gcp']).describe('Cloud provider (aws or gcp)'),
+      region: z.string().describe('Cloud region (e.g. us-central1)'),
+    },
+    async ({provider, region}) => {
+      const mapping = zoneMapper.getZoneMapping(provider, region)
+      
+      if (!mapping) {
+        return {
+          content: [{text: `Unknown region: ${region} for provider ${provider}`, type: 'text'}],
+          isError: true,
+        }
+      }
+
+      const intensity = await intensityProvider.getCarbonIntensity(mapping.zone)
+
+      const result = {
+        carbon_intensity_g_kwh: intensity,
+        grid_zone: mapping.zone,
+        pue: mapping.pue,
+        region,
+      }
+
+      return {
+        content: [{
+          text: JSON.stringify(result, null, 2),
+          type: 'text'
+        }]
+      }
+    }
+  )
+}

package/mcp-server/tsconfig.json

@@ -0,0 +1,18 @@
+{
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "target": "ES2022",
+    "sourceMap": true,
+    "declaration": true,
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist"]
+}

package/oclif.manifest.json

@@ -669,5 +669,5 @@
       ]
     }
   },
-  "version": "0.2.2"
+  "version": "0.2.4"
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "duoops",
   "description": "Toolset for Explainable and Sustainable CI on Gitlab.",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "author": "Younes Laaroussi",
   "bin": {
     "duoops": "./bin/run.js"
@@ -62,6 +62,7 @@
     "./bin",
     "./dist",
     "./data",
+    "./mcp-server",
     "./oclif.manifest.json",
     "./templates"
   ],