add seat analysis support for Github Enterprise, since newly released Github API support enterprise level seat information in 6/21, thanks for check.

Author: DevOps-zhuang

src/api/ExtractSeats.ts

@@ -8,21 +8,64 @@ import enterpriseMockedResponse_seats from '../assets/enterprise_response_sample
 export const getSeatsApi = async (): Promise<Seat[]> => {
   const perPage = 50;
   let page = 1;
+  let seatUrl = `https://api.github.com/`;
   let seatsData: Seat[] = [];
 
   let response;
-  if (process.env.VUE_APP_SCOPE !== "organization") {
-    // when the scope is not organization, return seatsData,by default it will return empty array
-    return seatsData;
-  }
-  else{
-    if (process.env.VUE_APP_MOCKED_DATA === "true") {
-      response = organizationMockedResponse_seats;
+  
+
+  if (process.env.VUE_APP_MOCKED_DATA === "true") {
+    console.log("Using mock data. Check VUE_APP_MOCKED_DATA variable.");
+      if (process.env.VUE_APP_SCOPE === "organization") {
+        response = organizationMockedResponse_seats;
+      } 
+      else if (process.env.VUE_APP_SCOPE === "enterprise") {
+        response = enterpriseMockedResponse_seats;
+      }
+      else {
+        throw new Error(`Invalid VUE_APP_SCOPE value: ${process.env.VUE_APP_SCOPE}. Expected "organization" or "enterprise".`);
+      }
       seatsData = seatsData.concat(response.seats.map((item: any) => new Seat(item)));
-    } 
-    else if (process.env.VUE_APP_MOCKED_DATA === "false") {
+      return seatsData;
+  }
+  else {
+    // if VUE_APP_GITHUB_TOKEN is not set, throw an error
+    if (!process.env.VUE_APP_GITHUB_TOKEN) {
+      throw new Error("VUE_APP_GITHUB_TOKEN environment variable is not set.");
+      return seatsData;
+    }
+    else if (process.env.VUE_APP_SCOPE === "organization") {
+      seatUrl=seatUrl+`orgs/${process.env.VUE_APP_GITHUB_ORG}/copilot/billing/seats`;
+    }
+    else if (process.env.VUE_APP_SCOPE === "enterprise") {
+      seatUrl=seatUrl+`enterprises/${process.env.VUE_APP_GITHUB_ENT}/copilot/billing/seats`;
+    }
+    else {
+      throw new Error(`Invalid VUE_APP_SCOPE value: ${process.env.VUE_APP_SCOPE}. Expected "organization" or "enterprise".`);
+      return seatsData;
+    }
+
     // Fetch the first page to get the total number of seats
-      response = await axios.get(`https://api.github.com/orgs/${process.env.VUE_APP_GITHUB_ORG}/copilot/billing/seats`, {
+    response = await axios.get(seatUrl, {
+      headers: {
+        Accept: "application/vnd.github+json",
+        Authorization: `Bearer ${process.env.VUE_APP_GITHUB_TOKEN}`,
+        "X-GitHub-Api-Version": "2022-11-28",
+      },
+      params: {
+        per_page: perPage,
+        page: page
+      }
+    });
+
+    seatsData = seatsData.concat(response.data.seats.map((item: any) => new Seat(item)));
+    // Calculate the total pages
+    const totalSeats = response.data.total_seats;
+    const totalPages = Math.ceil(totalSeats / perPage);
+
+    // Fetch the remaining pages
+    for (page = 2; page <= totalPages; page++) {
+      response = await axios.get(seatUrl, {
         headers: {
           Accept: "application/vnd.github+json",
           Authorization: `Bearer ${process.env.VUE_APP_GITHUB_TOKEN}`,
@@ -33,29 +76,8 @@ export const getSeatsApi = async (): Promise<Seat[]> => {
           page: page
         }
       });
-
       seatsData = seatsData.concat(response.data.seats.map((item: any) => new Seat(item)));
-      // Calculate the total pages
-      const totalSeats = response.data.total_seats;
-      const totalPages = Math.ceil(totalSeats / perPage);
-
-      // Fetch the remaining pages
-      for (page = 2; page <= totalPages; page++) {
-        response = await axios.get(`https://api.github.com/orgs/${process.env.VUE_APP_GITHUB_ORG}/copilot/billing/seats`, {
-          headers: {
-            Accept: "application/vnd.github+json",
-            Authorization: `Bearer ${process.env.VUE_APP_GITHUB_TOKEN}`,
-            "X-GitHub-Api-Version": "2022-11-28",
-          },
-          params: {
-            per_page: perPage,
-            page: page
-          }
-        });
-
-        seatsData = seatsData.concat(response.data.seats.map((item: any) => new Seat(item)));
-      } //end of else if (process.env.VUE_APP_MOCKED_DATA === "false")
-  }  //end of else if (process.env.VUE_APP_SCOPE !== "organization")
-  return seatsData;
-}
-}
+    }
+    return seatsData;
+    } 
+  }

src/assets/enterprise_response_sample_seats.json

@@ -8,7 +8,7 @@
         "last_activity_at": "2021-10-14T00:53:32-06:00",
         "last_activity_editor": "vscode/1.77.3/copilot/1.86.82",
         "assignee": {
-          "login": "octocat",
+          "login": "octocat_byEnterprise",
           "id": 1,
           "node_id": "MDQ6VXNlcjE=",
           "avatar_url": "https://github.com/images/error/octocat_happy.gif",

src/assets/organization_response_sample_seats.json

@@ -8,7 +8,7 @@
         "last_activity_at": "2021-10-14T00:53:32-06:00",
         "last_activity_editor": "vscode/1.77.3/copilot/1.86.82",
         "assignee": {
-          "login": "octocat",
+          "login": "octocat_org",
           "id": 1,
           "node_id": "MDQ6VXNlcjE=",
           "avatar_url": "https://github.com/images/error/octocat_happy.gif",
@@ -50,7 +50,7 @@
         "last_activity_at": "2021-10-13T00:53:32-06:00",
         "last_activity_editor": "vscode/1.77.3/copilot/1.86.82",
         "assignee": {
-          "login": "octokitten",
+          "login": "octokitten_org",
           "id": 1,
           "node_id": "MDQ76VNlcjE=",
           "avatar_url": "https://github.com/images/error/octokitten_happy.gif",

src/components/ApiResponse.vue

@@ -13,7 +13,8 @@
       </div>
 
       <br><br>
-    <div v-if="vueAppScope === 'organization'">
+    <!--<div v-if="vueAppScope === 'organization'">-->
+  
       <v-card max-height="575px" class="overflow-y-auto">
           <pre ref="jsonText">{{ JSON.stringify(seats, null, 2) }}</pre>
       </v-card>
@@ -24,7 +25,6 @@
           <div v-if="showSeatMessage" :class="{'copy-message': true, 'error': isError}">{{ message }}</div>
         </transition>
       </div>
-    </div>
   </v-container>
 </template>
 

src/components/MainComponent.vue

@@ -31,9 +31,9 @@
               <BreakdownComponent v-if="item === 'languages'" :metrics="metrics" :breakdownKey="'language'"/>
               <BreakdownComponent v-if="item === 'editors'" :metrics="metrics" :breakdownKey="'editor'"/>
               <CopilotChatViewer v-if="item === 'copilot chat'" :metrics="metrics" />
-              <div v-if="isScopeOrganization">
+              <!--<div v-if="isScopeOrganization">-->
                 <SeatsAnalysisViewer v-if="item === 'seat analysis'" :seats="seats" />
-              </div>
+              <!--</div>-->
               <ApiResponse v-if="item === 'api response'" :metrics="metrics" :seats="seats" />
             </v-card>
           </v-window-item>
@@ -94,23 +94,23 @@ export default defineComponent({
   },
   data () {
     return {
-      tabItems: ['languages', 'editors', 'copilot chat', 'api response'],
+      tabItems: ['languages', 'editors', 'copilot chat','seat analysis' , 'api response'],
       tab: null
     }
   },
   created() {
     if(this.itemName !== 'invalid'){
       this.tabItems.unshift(this.itemName);
     }
-    if (process.env.VUE_APP_SCOPE === 'organization') {
+    /* if (process.env.VUE_APP_SCOPE === 'organization') {
       // get the last item in the array,which is 'api response' 
       //and add 'seat analysis' before it
       let lastItem = this.tabItems.pop();
       this.tabItems.push('seat analysis');
       if (lastItem) {
         this.tabItems.push(lastItem);
       }
-    }
+    } */
   },
   setup() {
       const metricsReady = ref(false);