StartRNR/cli/src/queries.rs

use cynic::{GraphQlResponse, QueryBuilder};
use serde::{Deserialize, Serialize};

pub mod search_games;
pub mod tournament_sets;

use schema::schema;

// Types

// HACK: Unfortunately, start.gg seems to use integers for its ID type, whereas
// cynic always assumes that IDs are strings. To get around that, we define a
// new scalar type that serializes to u64.
#[derive(cynic::Scalar, Debug, Copy, Clone)]
pub struct ID(pub u64);

// Wrapper types to differentiate between different types of IDs
#[derive(Debug, Copy, Clone)]
pub struct VideogameId(pub u64);
#[derive(Debug, Copy, Clone)]
pub struct EntrantId(pub u64);

#[derive(cynic::Scalar, Debug, Clone)]
pub struct Timestamp(pub u64);

// Query machinery

pub trait QueryUnwrap<Vars>: QueryBuilder<Vars> {
    type VarsUnwrapped;
    type Unwrapped;

    fn wrap_vars(vars: Self::VarsUnwrapped) -> Vars;

    fn unwrap_response(response: GraphQlResponse<Self>) -> Option<Self::Unwrapped>;
}

// Generic function for running start.gg queries
pub async fn run_query<Builder: 'static, Vars>(
    vars: Builder::VarsUnwrapped,
    auth: &str,
) -> Option<Builder::Unwrapped>
where
    Builder: QueryUnwrap<Vars>,
    Vars: Serialize,
    for<'de> Builder: Deserialize<'de>,
{
    use cynic::http::SurfExt;

    let query = Builder::build(<Builder as QueryUnwrap<Vars>>::wrap_vars(vars));

    let response = surf::post("https://api.start.gg/gql/alpha")
        .header("Authorization", String::from("Bearer ") + auth)
        .run_graphql(query)
        .await;

    <Builder as QueryUnwrap<Vars>>::unwrap_response(response.unwrap())
}