iska
57 lines
1.0 KiB
Objective-C
57 lines
1.0 KiB
Objective-C
//
|
|
// HTMLAttributePolicy.m
|
|
// HTMLKit
|
|
//
|
|
// Created by Iska on 28.05.18.
|
|
// Copyright © 2018 BrainCookie. All rights reserved.
|
|
//
|
|
|
|
#import "HTMLAttributePolicy.h"
|
|
|
|
@interface HTMLAttributePolicy()
|
|
{
|
|
SanitizeAttrbiuteValueBlock _block;
|
|
}
|
|
@end
|
|
|
|
@implementation HTMLAttributePolicy
|
|
|
|
+ (instancetype)identity
|
|
{
|
|
return [self policyWithBlock:^NSString * _Nullable (NSString * _Nullable value, NSString * _Nonnull key) {
|
|
return value;
|
|
}];
|
|
}
|
|
|
|
+ (instancetype)rejectAll
|
|
{
|
|
return [self policyWithBlock:^NSString * _Nullable (NSString * _Nullable value, NSString * _Nonnull key) {
|
|
return nil;
|
|
}];
|
|
}
|
|
|
|
+ (instancetype)policyWithBlock:(SanitizeAttrbiuteValueBlock)block
|
|
{
|
|
HTMLAttributePolicy *policy = [[HTMLAttributePolicy alloc] initWithBlock:block];
|
|
return policy;
|
|
}
|
|
|
|
- (instancetype)initWithBlock:(SanitizeAttrbiuteValueBlock)block
|
|
{
|
|
self = [super init];
|
|
if (self) {
|
|
_block = block;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (NSString *)sanitizeValue:(NSString *)value forKey:(NSString *)key
|
|
{
|
|
if (_block) {
|
|
return _block(value, key);
|
|
}
|
|
return value;
|
|
}
|
|
|
|
@end
|