1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
// This file is part of rss.
//
// Copyright © 2015-2021 The rust-syndication Developers
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the MIT License and/or Apache 2.0 License.
use std::io::{BufRead, Write};
use quick_xml::events::attributes::Attributes;
use quick_xml::events::{BytesEnd, BytesStart, BytesText, Event};
use quick_xml::Error as XmlError;
use quick_xml::Reader;
use quick_xml::Writer;
use crate::error::Error;
use crate::toxml::ToXml;
use crate::util::{attr_value, decode, element_text};
/// Represents a category in an RSS feed.
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[derive(Debug, Default, Clone, PartialEq)]
#[cfg_attr(feature = "builders", derive(Builder))]
#[cfg_attr(
feature = "builders",
builder(
setter(into),
default,
build_fn(name = "build_impl", private, error = "never::Never")
)
)]
pub struct Category {
/// The name of the category.
pub name: String,
/// The domain for the category.
pub domain: Option<String>,
}
impl Category {
/// Return the name of this category.
///
/// # Examples
///
/// ```
/// use rss::Category;
///
/// let mut category = Category::default();
/// category.set_name("Technology");
/// assert_eq!(category.name(), "Technology");
/// ```
pub fn name(&self) -> &str {
self.name.as_str()
}
/// Set the name of this category.
///
/// # Examples
///
/// ```
/// use rss::Category;
///
/// let mut category = Category::default();
/// category.set_name("Technology");
/// ```
pub fn set_name<V>(&mut self, name: V)
where
V: Into<String>,
{
self.name = name.into();
}
/// Return the domain of this category.
///
/// # Examples
///
/// ```
/// use rss::Category;
///
/// let mut category = Category::default();
/// category.set_domain("http://example.com".to_string());
/// assert_eq!(category.domain(), Some("http://example.com"));
/// ```
pub fn domain(&self) -> Option<&str> {
self.domain.as_deref()
}
/// Set the domain of this category.
///
/// # Examples
///
/// ```
/// use rss::Category;
///
/// let mut category = Category::default();
/// category.set_domain("http://example.com".to_string());
/// ```
pub fn set_domain<V>(&mut self, domain: V)
where
V: Into<Option<String>>,
{
self.domain = domain.into();
}
}
impl Category {
/// Builds a Category from source XML
pub fn from_xml<R: BufRead>(
reader: &mut Reader<R>,
mut atts: Attributes,
) -> Result<Self, Error> {
let mut category = Category::default();
for attr in atts.with_checks(false).flatten() {
if decode(attr.key.as_ref(), reader)?.as_ref() == "domain" {
category.domain = Some(attr_value(&attr, reader)?.to_string());
break;
}
}
category.name = element_text(reader)?.unwrap_or_default();
Ok(category)
}
}
impl ToXml for Category {
fn to_xml<W: Write>(&self, writer: &mut Writer<W>) -> Result<(), XmlError> {
let name = "category";
let mut element = BytesStart::new(name);
if let Some(ref domain) = self.domain {
element.push_attribute(("domain", &**domain));
}
writer.write_event(Event::Start(element))?;
writer.write_event(Event::Text(BytesText::new(&self.name)))?;
writer.write_event(Event::End(BytesEnd::new(name)))?;
Ok(())
}
}
#[cfg(feature = "builders")]
impl CategoryBuilder {
/// Builds a new `Category`.
pub fn build(&self) -> Category {
self.build_impl().unwrap()
}
}