privacore-open-source-search-engine/UrlComponent.h

#ifndef GB_URLCOMPONENT_H
#define GB_URLCOMPONENT_H

#include <string>

class UrlComponent {
public:
	enum Type {
		TYPE_PATH,
		TYPE_QUERY
	};

	UrlComponent( Type type, const char *pos, size_t len, char separator, bool isFirst );

	void print() const;

	Type getType() const {
		return m_type;
	}

	const std::string& getString() const {
		return m_componentStr;
	}

	char getSeparator() const {
		return m_separator;
	}

	void setSeparator( char separator ) {
		m_separator = separator;
	}

	std::string getKey() const {
		return m_key;
	}

	bool hasValue() const {
		return ( m_keyLen != m_componentStr.size() );
	}

	const char* getValue() const {
		if ( m_keyLen == m_componentStr.size() ) {
			return m_componentStr.c_str();
		}

		return m_componentStr.c_str() + m_keyLen + 1;
	}

	size_t getValueLen() const {
		if ( m_keyLen == m_componentStr.size() ) {
			return m_componentStr.size();
		}

		return m_componentStr.size() - m_keyLen - 1;
	}

	bool isFirst() const {
		return m_isFirst;
	}

	bool isDeleted() const {
		return m_deleted;
	}

	void setDeleted() {
		m_deleted = true;
	}

	class Matcher;
	class Validator;

	static void normalize( std::string *component );

private:
	Type m_type;

	std::string m_componentStr;

	char m_separator;
	bool m_isFirst;

	std::string m_key;
	size_t m_keyLen;

	bool m_deleted;
};

enum MatchCriteria {
	MATCH_DEFAULT = 0,
	MATCH_CASE = 1,
	MATCH_PARTIAL = 2,
	MATCH_PREFIX = 4,
	MATCH_SUFFIX = 8
};

inline MatchCriteria operator|( MatchCriteria a, MatchCriteria b ) {
	return static_cast<MatchCriteria>( static_cast<int>(a) | static_cast<int>(b) );
}

inline MatchCriteria operator&( MatchCriteria a, MatchCriteria b ) {
	return static_cast<MatchCriteria>( static_cast<int>(a) & static_cast<int>(b) );
}

class UrlComponent::Matcher {
public:
	Matcher( const char *param, MatchCriteria matchCriteria = MATCH_DEFAULT );

	void print() const;

	MatchCriteria getMatchCriteria() const {
		return m_matchCriteria;
	}

	const std::string& getParam() const {
		return m_param;
	}

	bool isMatching( const UrlComponent &urlPart ) const; 

private:
	std::string m_param;
	MatchCriteria m_matchCriteria;
	bool m_matchPartial;
	bool m_matchCase;
	bool m_matchPrefix;
	bool m_matchSuffix;
};

// allowed characters
enum AllowCriteria {
	ALLOW_ALL = 0,
	ALLOW_DIGIT = 1, // allow digit
	ALLOW_HEX = 2, // allow hex
	ALLOW_ALPHA = 4, // allow alpha (lower/upper)
	ALLOW_ALPHA_LOWER = 8, // allow alpha lower
	ALLOW_ALPHA_UPPER = 16, // allow alpha upper
	ALLOW_PUNCTUATION = 32 // allow punctuation
};

inline AllowCriteria operator|( AllowCriteria a, AllowCriteria b ) {
	return static_cast<AllowCriteria>( static_cast<int>(a) | static_cast<int>(b) );
}

inline AllowCriteria operator&( AllowCriteria a, AllowCriteria b ) {
	return static_cast<AllowCriteria>( static_cast<int>(a) & static_cast<int>(b) );
}

// mandatory characters
enum MandatoryCriteria {
	MANDATORY_NONE = 0,
	MANDATORY_DIGIT = 1, // must have digit
	MANDATORY_ALPHA_HEX = 2, // must have alpha hex
	MANDATORY_ALPHA = 4, // must have alpha (lower/upper)
	MANDATORY_ALPHA_LOWER = 8, // must have alpha lower
	MANDATORY_ALPHA_UPPER = 16, // must have alpha upper
	MANDATORY_PUNCTUATION = 32 // must have punctuation
};

inline MandatoryCriteria operator|( MandatoryCriteria a, MandatoryCriteria b ) {
	return static_cast<MandatoryCriteria>( static_cast<int>(a) | static_cast<int>(b) );
}

inline MandatoryCriteria operator&( MandatoryCriteria a, MandatoryCriteria b ) {
	return static_cast<MandatoryCriteria>( static_cast<int>(a) & static_cast<int>(b) );
}

class UrlComponent::Validator {
public:
	Validator(size_t minLength, size_t maxLength, AllowCriteria allowCriteria, MandatoryCriteria mandatoryCriteria);
	Validator(size_t minLength, size_t maxLength, bool allowEmpty, AllowCriteria allowCriteria, MandatoryCriteria mandatoryCriteria);

	void print() const;

	bool allowEmptyValue() const {
		return m_allowEmpty;
	}

	bool isValid(const UrlComponent &urlPart) const;

	bool isValid(const char *value, size_t valueLen) const;

private:
	size_t m_minLength;
	size_t m_maxLength;
	bool m_allowEmpty;

	AllowCriteria m_allowCriteria;
	bool m_allowAlpha;
	bool m_allowAlphaLower;
	bool m_allowAlphaUpper;
	bool m_allowAlphaHex;
	bool m_allowDigit;
	bool m_allowPunctuation;

	MandatoryCriteria m_mandatoryCriteria;
	bool m_mandatoryAlpha;
	bool m_mandatoryAlphaLower;
	bool m_mandatoryAlphaUpper;
	bool m_mandatoryAlphaHex;
	bool m_mandatoryDigit;
	bool m_mandatoryPunctuation;
};

#endif // GB_URLCOMPONENT_H