tables

Author: victorzheng02

src/components/companies.ts

@@ -0,0 +1,43 @@
+import _ from 'lodash';
+import { openDB } from './db';
+
+interface SQLiteCompanyInfoResponse {
+  company_id: string;
+  description: string;
+  categories: string;
+}
+
+interface CompanyInfo {
+  company_id: string;
+  description: string;
+  categories: string[];
+}
+
+export const getCompanyInfo = async (companyId: string): Promise<CompanyInfo[]> => {
+  const db = await openDB();
+  const res: SQLiteCompanyInfoResponse[] = await db.all(
+    'SELECT * FROM companies WHERE id = ?',
+    companyId,
+  );
+  const cleaned = res.map((row) => {
+    return {
+      ...row,
+      categories: row.categories.split(','),
+    };
+  });
+  return cleaned;
+};
+
+export const getCompaniesByUserId = async (userId: string): Promise<string[]> => {
+  const db = await openDB();
+
+  const res = await db.all('SELECT company_id FROM people_companies WHERE user_id = ?', userId);
+  return res.map((row) => row.company_id);
+};
+
+export const getUsersByCompanyId = async (companyId: string): Promise<string[]> => {
+  const db = await openDB();
+
+  const res = await db.all('SELECT user_id FROM people_companies WHERE company_id = ?', companyId);
+  return res.map((row) => row.user_id);
+};

src/components/db.ts

@@ -165,6 +165,27 @@ const addSQLColumnIfNotExists = async (
   if (columns.length == 0) {
     await db.run(`ALTER TABLE ${tableName} ADD COLUMN ${columnName} ${columnDataType}`);
   }
+const initCompaniesTable = async (db: Database): Promise<void> => {
+  // the company_id will match the crunchbase entity_id, which is a unique identifier for each company
+  // see https://app.swaggerhub.com/apis-docs/Crunchbase/crunchbase-enterprise_api/1.0.3#/Entity/get_entities_organizations__entity_id_
+  await db.run(
+    `
+    CREATE TABLE IF NOT EXISTS companies (
+      company_id VARCHAR(30) PRIMARY KEY NOT NULL,
+      description TEXT
+      categories TEXT
+      `,
+  );
+};
+
+const initPeopleCompaniesTable = async (db: Database): Promise<void> => {
+  await db.run(
+    `
+    CREATE TABLE IF NOT EXISTS people_companies (
+      user_id VARCHAR(255) TEXT NOT NULL,
+      FOREIGNKEY(company_id) REFERENCES companies(company_id) NOT NULL
+      )`,
+  );
 };
 
 const initTables = async (db: Database): Promise<void> => {
@@ -177,6 +198,8 @@ const initTables = async (db: Database): Promise<void> => {
   await initUserCoinTable(db);
   await initUserProfileTable(db);
   await initRpsGameInfo(db);
+  await initCompaniesTable(db);
+  await initPeopleCompaniesTable(db);
 };
 
 export const openDB = async (): Promise<Database> => {
@@ -189,4 +212,4 @@ export const openDB = async (): Promise<Database> => {
     logger.info({ message: 'Initialized database and tables.', where: 'openDB' });
   }
   return db;
-};
+};

src/utils/companyInfo.ts

@@ -4,6 +4,7 @@ import fetch from 'node-fetch';
 const CRUNCHBASE_ORGANIZATION_URL = 'crunchbase.com/organization/';
 const CRUNCHBASE_ORGANIZATION_API_URL = 'https://api.crunchbase.com/api/v4/entities/organizations/';
 
+const fields = ['categories', 'short_description', 'website'];
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
 const getCompanyInfo = async (companyCrunchbaseLink: string) => {
   // check if url contains crunchbase.com/organization
@@ -25,7 +26,9 @@ const getCompanyInfo = async (companyCrunchbaseLink: string) => {
   }
 
   const response = await fetch(
-    `${CRUNCHBASE_ORGANIZATION_API_URL}${companyName}?user_key=${process.env.CRUNCHBASE_API_KEY}&field_ids=categories,short_description`,
+    `${CRUNCHBASE_ORGANIZATION_API_URL}${companyName}?user_key=${
+      vars.CRUNCHBASE_API_KEY
+    }&field_ids=${fields.join(',')}`,
   );
   const data = await response.json();
   if (data.error) {