rspamd/src/libserver/css/css_rule.hxx

/*-
 * Copyright 2021 Vsevolod Stakhov
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#pragma once

#ifndef RSPAMD_CSS_RULE_HXX
#define RSPAMD_CSS_RULE_HXX

#include "css_value.hxx"
#include "css_property.hxx"
#include "css_parser.hxx"
#include "contrib/ankerl/unordered_dense.h"
#include "libutil/cxx/util.hxx"
#include "libutil/cxx/hash_util.hxx"
#include <vector>
#include <memory>

namespace rspamd::html {
/* Forward declaration */
struct html_block;
}// namespace rspamd::html

namespace rspamd::css {

class css_rule {
	css_property prop;
	using css_values_vec = std::vector<css_value>;
	css_values_vec values;

public:
	/* We must create css rule explicitly from a property and values */
	css_rule() = delete;

	css_rule(const css_rule &other) = delete;

	/* Constructors */
	css_rule(css_rule &&other) noexcept = default;

	explicit css_rule(css_property &&prop, css_values_vec &&values) noexcept
		: prop(prop), values(std::forward<css_values_vec>(values))
	{
	}

	explicit css_rule(const css_property &prop) noexcept
		: prop(prop), values{}
	{
	}

	/* Methods */
	/* Comparison is special, as we care merely about property, not the values */
	auto operator==(const css_rule &other) const
	{
		return prop == other.prop;
	}

	constexpr const css_values_vec &get_values(void) const
	{
		return values;
	}
	constexpr const css_property &get_prop(void) const
	{
		return prop;
	}

	/* Import values from another rules according to the importance */
	void override_values(const css_rule &other);
	void merge_values(const css_rule &other);
	void add_value(const css_value &value);
};

}// namespace rspamd::css

/* Make rules hashable by property */
namespace std {
template<>
class hash<rspamd::css::css_rule> {
public:
	using is_avalanching = void;
	constexpr auto operator()(const rspamd::css::css_rule &rule) const -> auto
	{
		return hash<rspamd::css::css_property>()(rule.get_prop());
	}
};

}// namespace std

namespace rspamd::css {

/**
 * Class that is designed to hold css declaration (a set of rules)
 */
class css_declarations_block {
public:
	using rule_shared_ptr = std::shared_ptr<css_rule>;
	using rule_shared_hash = smart_ptr_hash<css_rule>;
	using rule_shared_eq = smart_ptr_equal<css_rule>;
	enum class merge_type {
		merge_duplicate,
		merge_parent,
		merge_override
	};

	css_declarations_block() = default;
	auto add_rule(rule_shared_ptr rule) -> bool;
	auto merge_block(const css_declarations_block &other,
					 merge_type how = merge_type::merge_duplicate) -> void;
	auto get_rules(void) const -> const auto &
	{
		return rules;
	}

	/**
	 * Returns if a declaration block has some property
	 * @param prop
	 * @return
	 */
	auto has_property(const css_property &prop) const -> bool
	{
		return (rules.find(css_rule{prop}) != rules.end());
	}

	/**
	 * Compile CSS declaration to the html block
	 * @param pool used to carry memory required for html_block
	 * @return html block structure
	 */
	auto compile_to_block(rspamd_mempool_t *pool) const -> rspamd::html::html_block *;

private:
	ankerl::unordered_dense::set<rule_shared_ptr, rule_shared_hash, rule_shared_eq> rules;
};

using css_declarations_block_ptr = std::shared_ptr<css_declarations_block>;

auto process_declaration_tokens(rspamd_mempool_t *pool,
								blocks_gen_functor &&next_token_functor)
	-> css_declarations_block_ptr;

}// namespace rspamd::css

#endif//RSPAMD_CSS_RULE_HXX