diff --git a/report-window/use-survey.ts b/report-window/use-survey.ts index 84a588f..ae7bf70 100644 --- a/report-window/use-survey.ts +++ b/report-window/use-survey.ts @@ -2,7 +2,7 @@ import * as React from 'react'; import * as Survey from 'survey-react'; import generateSurveyQuestions from './generate-survey-questions'; import RawAnswers from './raw-answers'; -import verbs from './verbs'; +import verbs, { v } from './verbs'; export default function useSurvey( hosts: string[], @@ -17,13 +17,7 @@ export default function useSurvey( options: { name: string; value?: string } ) { if (verbs[options.name.toLowerCase()]) { - options.value = verbs[options.name.toLowerCase()][sender.valuesHash.zaimek]; - if (options.name[0] == options.name[0].toUpperCase()) { - options.value = [ - options.value[0].toUpperCase(), - ...options.value.slice(1), - ].join(''); - } + options.value = v(options.name, sender.valuesHash.zaimek); } }); survey.onComplete.add(onComplete); diff --git a/report-window/verbs.ts b/report-window/verbs.ts index 288de97..1519192 100644 --- a/report-window/verbs.ts +++ b/report-window/verbs.ts @@ -1,4 +1,4 @@ -export default { +const words = { zrobiłem: ['zrobiłem', 'zrobiłam', 'zrobiłom', 'zrobiliśmy'], szukałem: ['szukałem', 'szukałam', 'szukałom', 'szukaliśmy'], znalazłem: ['znalazłem', 'znalazłam', 'znalazłom', 'znaleźliśmy'], @@ -17,3 +17,13 @@ export default { widzę: ['widzę', 'widzę', 'widzę', 'widzimy'], widziałem: ['widziałem', 'widziałam', 'widziałom', 'widzieliśmy'], } as { [key: string]: string[] }; + +export default words; + +export function v(key: string, index: number) { + let result = words[key.toLowerCase()][index] || key; + if (key[0] == key[0].toUpperCase()) { + result = [result[0].toUpperCase(), ...result.slice(1)].join(''); + } + return result; +}