Fazendaaa/AnilistBot

import { queryAnilist } from 'endeavor';
import { IAnilistContext, INotFoundContext, ISearchContext } from '.';
import { IMinimumInline } from '../../telegram/';
import { ICharacters, ICharactersQueryPage, IMedia, IMediaQueryPage, IStaff, IStaffQueryPage, IStudios, IStudiosQueryPage
} from '../queries';
import characters from '../queries/charactersSearch.gql';
import media from '../queries/mediaSearch.gql';
import staff from '../queries/staffSearch.gql';
import studio from '../queries/studioSearch.gql';
import { charactersInfo, mediaInfo, notFoundInfo, staffInfo, studiosInfo } from './info';

const searchAnilist = async ({ search, query, page }: IAnilistContext): Promise<Object | Error> => queryAnilist({
    query,
    variables: { page, perPage: 2,  search: ('' !== search) ? search : null  }
});

const staffSearch = async ({ search, page, translation }: ISearchContext): Promise<IMinimumInline[]> => {
    const searched = <IStaffQueryPage> await searchAnilist({ search, page, query: staff });
    const curriedStaffInfo = ((input: IStaff) => staffInfo({ staff: input, translation }));

    return searched.data.Page.staff.map(curriedStaffInfo);
};

const studiosSearch = async ({ search, page, translation }: ISearchContext): Promise<IMinimumInline[]> => {
    const searched = <IStudiosQueryPage> await searchAnilist({ search, page, query: studio });
    const curriedStudiosInfo = ((studios: IStudios) => studiosInfo({ studios, translation }));

    return searched.data.Page.studios.map(curriedStudiosInfo);
};

const charactersSearch = async ({ search, page, translation }: ISearchContext): Promise<IMinimumInline[]> => {
    const searched = <ICharactersQueryPage> await searchAnilist({ search, page, query: characters });
    const curriedCharactersInfo = ((input: ICharacters) => charactersInfo({ characters: input, translation }));

    return searched.data.Page.characters.map(curriedCharactersInfo);
};

const mediaSearch = async ({ search, page, translation }: ISearchContext): Promise<IMinimumInline[]> => {
    const searched = <IMediaQueryPage> await searchAnilist({ search, page, query: media });
    const curriedMediaInfo = ((input: IMedia) => mediaInfo({ media: input, translation }));

    return searched.data.Page.media.map(curriedMediaInfo);
};

export const allSearch = async ({ search, page, translation }: ISearchContext): Promise<IMinimumInline[]> => {
    const first = await mediaSearch({ search, translation, page });
    const second = (first.length === 0) ? await charactersSearch({ search, translation, page }) : [];
    const third = (second.length === 0) ? await studiosSearch({ search, translation, page }) : [];
    const fourth = (third.length === 0) ? await staffSearch({ search, translation, page }) : [];

    return [
        ...first,
        ...second,
        ...third,
        ...fourth
    ];
};

export const notFoundSearch = ({ search, translation }: INotFoundContext): IMinimumInline[] => [ notFoundInfo({ search, translation }) ];