switch statement, types fixed

scripts/wg-bot/src/joystream/api_extension.ts

@@ -3,7 +3,7 @@ import { Opening, OpeningId } from "@joystream/types/hiring";
 import { ApiPromise } from "@polkadot/api";
 import { Hash } from "@polkadot/types/interfaces";
 
-export const getOpening = async (api: ApiPromise, group: string, hash: Hash, openingId: number): Promise<OpeningOf> => {
+export const getOpening = async (api: ApiPromise, group: string, hash: Hash, openingId: OpeningId): Promise<OpeningOf> => {
     return api.query[group].openingById.at(hash, openingId)
 }
 

scripts/wg-bot/src/joystream/discord.ts

@@ -3,10 +3,12 @@ import { getWorker, getMember, getMint, getEvents, getBlockHash, getWorkerReward
 import { getHiringOpening, getOpening } from './api_extension';
 import { getMintCapacityChangedEmbed, getOpeningAddedEmbed, getOpeningFilledEmbed, getWorkerRewardAmountUpdatedEmbed } from './embeds';
 
-import { wgEvents, workingGroups, joystreamBlue } from '../config'
+import { wgEvents, workingGroups } from '../config'
 import Discord from 'discord.js';
 import { ApiPromise } from '@polkadot/api';
-import { u32 } from '@polkadot/types';
+import { MintBalanceOf, MintId } from '@joystream/types/mint';
+import { OpeningId } from '@joystream/types/hiring';
+import { WorkerId } from '@joystream/types/working-group';
 
 export const processBlock = async (api: ApiPromise, client: Discord.Client, blockNumber: number) => {
     const hash = await getBlockHash(api, blockNumber);
@@ -14,39 +16,41 @@ export const processBlock = async (api: ApiPromise, client: Discord.Client, bloc
     blockEvents.forEach(async (value: EventRecord, index: number, array: EventRecord[]) => {
         let { section, method, data } = value.event;
         if (wgEvents.includes(method) && Object.keys(workingGroups).includes(section)) {
-            console.log(section);
-            console.log(method);
-            console.log(data.toJSON());
+            console.log(`${blockNumber} ${section} ${method} ${data.toHuman()}`);
 
             const channel: Discord.TextChannel = client.channels.cache.get(workingGroups[section]) as Discord.TextChannel;
             if (channel) {
-
-                if (method === 'MintCapacityChanged') {
-                    const mintId = (data[0] as u32).toNumber()
-                    const minted = (data[1] as u32).toNumber()
-                    const mint = (await getMint(api, hash, mintId)).capacity
-                    channel.send({ embeds: [getMintCapacityChangedEmbed(minted, mint, blockNumber, value)] });
-                } else if (method === 'OpeningAdded') {
-                    const openingId = (data[0] as u32).toNumber()
-                    const openingObject = await getOpening(api, section, hash, openingId)
-                    const hiringOpening = await getHiringOpening(api, hash, openingObject.hiring_opening_id)
-                    const opening = JSON.parse(hiringOpening.human_readable_text.toString())
-                    channel.send({ embeds: [getOpeningAddedEmbed(opening, openingObject, blockNumber, value)] });
-                } else if (method === 'OpeningFilled') {
-                    const openingId = (data[0] as u32).toNumber()
-                    const workerId = Object.values(JSON.parse(data[1].toString()))[0] as number
-                    const worker = (await getWorker(api, section, hash, workerId))
-                    const member = (await getMember(api, worker.member_id))
-                    const openingObject = await getOpening(api, section, hash, openingId)
-                    const hiringOpening = await getHiringOpening(api, hash, openingObject.hiring_opening_id)
-                    const opening = JSON.parse(hiringOpening.human_readable_text.toString())
-                    channel.send({ embeds: [getOpeningFilledEmbed(opening, member, blockNumber, value)] });
-                } else if (method === 'WorkerRewardAmountUpdated') {
-                    const workerId = (data[0] as u32).toNumber()
-                    const worker = await getWorker(api, section, hash, workerId)
-                    const member = await getMember(api, worker.member_id)
-                    const reward = await getWorkerReward(api, hash, worker.reward_relationship.unwrap())
-                    channel.send({ embeds: [getWorkerRewardAmountUpdatedEmbed(reward, member, blockNumber, value)] });
+                switch(method) {
+                    case "MintCapacityChanged":
+                        const mintId = (data[0] as MintId).toNumber();
+                        const minted = (data[1] as MintBalanceOf).toNumber();
+                        const mint = (await getMint(api, hash, mintId)).capacity;
+                        channel.send({ embeds: [getMintCapacityChangedEmbed(minted, mint, blockNumber, value)] });
+                        break;
+                    case "OpeningAdded": 
+                        const addedOpeningId = data[0] as OpeningId;
+                        const addedOpeningObject = await getOpening(api, section, hash, addedOpeningId);
+                        const addedHiringOpening = await getHiringOpening(api, hash, addedOpeningObject.hiring_opening_id);
+                        const addedOpeningText = JSON.parse(addedHiringOpening.human_readable_text.toString());
+                        channel.send({ embeds: [getOpeningAddedEmbed(addedOpeningText, addedOpeningObject, blockNumber, value)] });
+                        break;
+                    case "OpeningFilled": 
+                        const filledOpeningId = data[0] as OpeningId;
+                        const hiredWorkerId = Object.values(JSON.parse(data[1].toString()))[0] as number;
+                        const hiredWorker = await getWorker(api, section, hash, hiredWorkerId);
+                        const hiredMember = await getMember(api, hiredWorker.member_id);
+                        const filledOpeningObject = await getOpening(api, section, hash, filledOpeningId);
+                        const filledHiringOpening = await getHiringOpening(api, hash, filledOpeningObject.hiring_opening_id);
+                        const filledOpeningText = JSON.parse(filledHiringOpening.human_readable_text.toString());
+                        channel.send({ embeds: [getOpeningFilledEmbed(filledOpeningText, hiredMember, blockNumber, value)] });
+                        break;
+                    case "WorkerRewardAmountUpdated": 
+                        const workerId = data[0] as WorkerId;
+                        const worker = await getWorker(api, section, hash, workerId.toNumber());
+                        const member = await getMember(api, worker.member_id);
+                        const reward = await getWorkerReward(api, hash, worker.reward_relationship.unwrap());
+                        channel.send({ embeds: [getWorkerRewardAmountUpdatedEmbed(reward, member, blockNumber, value)] });
+                        break;
                 }
             } else {
                 console.log(`Channel not configured for ${section}`);