iska
95 lines
2.8 KiB
Objective-C
95 lines
2.8 KiB
Objective-C
//
|
|
// HTMLSanitizingPolicyBuilder.m
|
|
// HTMLKit
|
|
//
|
|
// Created by Iska on 26.05.18.
|
|
// Copyright © 2018 BrainCookie. All rights reserved.
|
|
//
|
|
|
|
#import "HTMLSanitizingPolicyBuilder.h"
|
|
|
|
@interface HTMLSanitizingPolicyBuilder()
|
|
{
|
|
NSMutableDictionary<NSString *, NSMutableArray<HTMLElementPolicy *> *> * elementPolicies;
|
|
NSMutableDictionary<NSString *, HTMLAttributePolicy *> * attributePolicies;
|
|
NSMutableDictionary<NSString *, NSNumber *> * textContainers;
|
|
}
|
|
@end
|
|
|
|
@implementation HTMLSanitizingPolicyBuilder
|
|
|
|
- (HTMLSanitizingPolicyBuilder *)allowElements:(NSArray<NSString *> *)elementNames
|
|
{
|
|
return [self allowPolicy:HTMLElementPolicy.identity onElements:elementNames];
|
|
}
|
|
|
|
- (HTMLSanitizingPolicyBuilder *)disallowElements:(NSArray<NSString *> *)elementNames
|
|
{
|
|
return [self allowPolicy:HTMLElementPolicy.rejectAll onElements:elementNames];
|
|
}
|
|
|
|
- (HTMLSanitizingPolicyBuilder *)allowPolicy:(HTMLElementPolicy *)policy onElements:(NSArray<NSString *> *)elementNames
|
|
{
|
|
for (NSString *name in elementNames) {
|
|
NSMutableArray<HTMLElementPolicy *> *list = elementPolicies[name];
|
|
if (list == nil) {
|
|
list = [NSMutableArray new];
|
|
}
|
|
[list addObject:policy];
|
|
elementPolicies[name] = list;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (HTMLSanitizingPolicyBuilder *)allowCommonInlineFormattingElements
|
|
{
|
|
return [self allowElements:@[@"b", @"i", @"font", @"s", @"u", @"o", @"sup", @"sub", @"ins", @"del",
|
|
@"strong", @"strike", @"tt", @"code", @"big", @"small", @"br", @"span", @"em"]];
|
|
}
|
|
|
|
- (HTMLSanitizingPolicyBuilder *)allowCommonBlockElements
|
|
{
|
|
return [self allowElements:@[@"p", @"div", @"h1", @"h2", @"h3", @"h4", @"h5", @"h6", @"ul", @"ol", @"li",
|
|
@"blockquote"]];
|
|
}
|
|
|
|
- (HTMLSanitizingPolicyBuilder *)allowTextInElements:(NSArray<NSString *> *)elementNames;
|
|
{
|
|
for (NSString *name in elementNames) {
|
|
textContainers[name] = @YES;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (HTMLSanitizingPolicyBuilder *)disallowTextInElements:(NSArray<NSString *> *)elementNames
|
|
{
|
|
for (NSString *name in elementNames) {
|
|
textContainers[name] = @NO;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
//- (HTMLSanitizingPolicyBuilder *)allowAttributes:(NSArray<NSString *> *)attributeNames
|
|
// onElements:(NSArray<NSString *> *)elementNames
|
|
//{
|
|
// for (NSString *attribute in attributeNames) {
|
|
// [self allowAttributePolicy:HTMLAttributePolicy.identity onElements:elementNames];
|
|
// }
|
|
// return self;
|
|
//}
|
|
//
|
|
//- (HTMLSanitizingPolicyBuilder *)disallowAttributes:(NSArray<NSString *> *)attributeName
|
|
// onElements:(NSArray<NSString *> *)elementNames;
|
|
//
|
|
//- (HTMLSanitizingPolicyBuilder *)allowAttributePolicy:(HTMLAttributePolicy *)policy onElements:(NSArray<NSString *> *)elementNames
|
|
//{
|
|
// return self;
|
|
//}
|
|
//
|
|
//- (HTMLSanitizingPolicyBuilder *)disallowAttributePolicy:(HTMLAttributePolicy *)policy onElements:(NSArray<NSString *> *)elementNames
|
|
//{
|
|
// return self;
|
|
//}
|
|
|
|
@end
|