feat: add CompanyType

src/Tmdb.js

@@ -23,6 +23,7 @@ import type {
   MovieType,
   MovieVideoType,
   PersonType,
+  CompanyType,
 } from './types';
 
 type QueryType = {
@@ -158,6 +159,14 @@ class Tmdb {
     return person;
   }
 
+  async getCompany (companyId: number): Promise<CompanyType> {
+    const company = await this.get('company/' + companyId, {
+      language: this.language,
+    });
+
+    return company;
+  }
+
   async findId (resourceType: 'movie' | 'person', externalSource: 'imdb', externalId: string): Promise<number> {
     if (resourceType !== 'movie' && resourceType !== 'person') {
       throw new Unimplemented();

src/bin/generate-types.js

@@ -10,6 +10,9 @@ import {
 } from 'lodash';
 
 const typeMap = {
+  CompanyType: (data) => {
+    return data.paths['/company/{company_id}'].get.responses['200'].schema.properties;
+  },
   ImagePathType: (data) => {
     return data.definitions['image-path'].type;
   },
@@ -121,9 +124,7 @@ const getPropertyFlowType = (property: Object) => {
 };
 
 const run = async () => {
-  const oas = await got('https://api.stoplight.io/v1/versions/9WaNJfGpnnQ76opqe/export/oas.json', {
-    json: true,
-  });
+  const oas = await got('https://api.stoplight.io/v1/versions/9WaNJfGpnnQ76opqe/export/oas.json').json();
 
   for (const typeName of typeNames) {
     const resourceResolver = typeMap[typeName];
@@ -132,7 +133,7 @@ const run = async () => {
       throw new Error('Unexpected state.');
     }
 
-    const typeDefinition = resourceResolver(oas.body);
+    const typeDefinition = resourceResolver(oas);
 
     if (!typeDefinition) {
       throw new Error('Unexpected state.');

src/types.js

@@ -116,3 +116,14 @@ export type PersonType = {|
   +popularity: number,
   +profilePath: ImagePathType,
 |};
+
+export type CompanyType = {|
+  +description: string,
+  +headquarters: string,
+  +homepage: string,
+  +id: number,
+  +logoPath: string,
+  +name: string,
+  +originCountry: string,
+  +parentCompany: null | CompanyType,
+|};