booru-rs 0.3.0

//! Models for Safebooru API responses.
//!
//! This module contains the data structures for deserializing
//! responses from the Safebooru API.

use core::fmt;
use serde::Deserialize;

/// A post from Safebooru.
///
/// This struct represents a single image post from Safebooru.
/// Safebooru is a SFW-only booru site.
#[derive(Deserialize, Debug, Clone, PartialEq)]
pub struct SafebooruPost {
    pub id: u32,
    pub score: Option<u32>,
    /// This can be `null` for really recent posts
    pub height: u32,
    pub width: u32,
    pub hash: String,
    pub tags: String,
    pub image: String,
    /// Directory number where the image is stored
    pub directory: u32,
    /// Full URL to the image file
    pub file_url: String,
    /// URL to the preview/thumbnail image
    pub preview_url: String,
    /// URL to the sample (resized) image
    pub sample_url: String,
    /// Source URL for the original artwork
    #[serde(default)]
    pub source: String,
    /// This is basically equivalent to `updated_at` in a Danbooru post. Except
    /// that it's provided as a UNIX timestamp. Safebooru provides no `created_at`
    /// field.
    pub change: u32,
    pub rating: SafebooruRating,
}

/// Post rating classification for Safebooru.
///
/// While Safebooru is primarily a SFW site, the rating field
/// can contain other values for deleted/hidden content.
#[derive(Deserialize, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[serde(rename_all = "lowercase")]
pub enum SafebooruRating {
    Safe,
    General,
    // Yes there are explicit and questionable posts. Though you only need to care
    // about them if you're querying for deleted content.
    Questionable,
    Explicit,
}

impl From<SafebooruRating> for String {
    fn from(rating: SafebooruRating) -> String {
        rating.to_string()
    }
}

impl fmt::Display for SafebooruRating {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let tag = format!("{:?}", self).to_lowercase();
        write!(f, "{tag}")
    }
}