import re
from .config.const import *

tranny = f'<img loading="lazy" data-bs-toggle="tooltip" alt=":marseytrain:" title=":marseytrain:" src="{SITE_FULL_IMAGES}/e/marseytrain.webp">'
trannie = f'<img loading="lazy" data-bs-toggle="tooltip" alt=":!marseytrain:" title=":!marseytrain:" src="{SITE_FULL_IMAGES}/e/marseytrain.webp">'
troon = f'<img loading="lazy" data-bs-toggle="tooltip" alt=":marseytrain2:" title=":marseytrain2:" src="{SITE_FULL_IMAGES}/e/marseytrain2.webp">'

#DON'T ADD ANY FILTERS WITH ' OR ", VECTOR FOR HTML INJECTION
SLURS = {
	'tranny': tranny,
	'trannie': trannie,
	'troon': troon,
	'(?<!\\bs)nigger': 'BIPOC',
	'negroid': 'BIPOC',
	'nignog': 'BIPOC',
	'nig nog': 'BIPOC',
	'(?<!\w)nog(?!\w)': 'BIPOC',
	'niglet': 'BIPOClet',
	'negress': 'BIPOCette',
	'faggot': 'cute twink',
	'fag': 'strag',
	'(?<!\w)spic(?!\w)': 'hard-working American',
	'(?<!\w)spics(?!\w)': 'hard-working Americans',
	'kike': 'jewish chad',
	'(?<!\w)heeb': 'jewish chad',
	'daisys destruction': 'Cars 2',
	'daisy destruction': 'Cars 2',
	'pajeet': 'sexy Indian dude',
	'dyke(?!\w)': 'cute butch',
	'dykes': 'cute butches',
	'sheboon': 'ebony cupcake',
	'gaslight': 'gaslamp',
}

#DON'T ADD ANY FILTERS WITH ' OR ", VECTOR FOR HTML INJECTION
if SITE_NAME == 'rDrama':
	SLURS |= {
		'retarded': 'r-slurred',
		' a retard': ' an r-slur',
		'retard': 'r-slur',
		'(?<!\w)pedo(?!\w)': 'p-do',
		'pedophile': 'p-dophile',
		'kill youself': 'keep yourself safe',
		'kill yourself': 'keep yourself safe',
		'kill yourselves': 'keep yourselves safe',
		'(?<!\w)kys(?!\w)': 'keep yourself safe',
		'latinos': 'latinx',
		'latino': 'latinx',
		'latinas': 'latinx',
		'latina': 'latinx',
		'hispanics': 'latinx',
		'hispanic': 'latinx',
		'autistic': 'neurodivergent',
		'autist': 'neurodivergent',
		'gamer': 'g*mer',
		'journalist': 'journ*list',
		'journalism': 'journ*lism',
		'fake and gay': 'fake and straight',
		'(?<!\w)rapist': 'male feminist',
		'its almost if': 'Im an r-slur but',
		'its almost as if': 'Im an r-slur but',
		'its almost like': 'Im an r-slur but',
		'krayon(?! \()': 'krayon (sister toucher)',
		'discord': 'groomercord',
		'nigga': 'neighbor',
		'(?<!cow)tools(?!\w)': 'cowtools',
		'upvote': 'upmarsey',
		'downvote': 'downmarsey',
	}

PROFANITIES = {
	'motherfucker': 'motherlover',
	'fuck': 'frick',
	'(?<!\w)ass(?!\w)': 'butt',
	'shitting': 'pooping',
	'damn': 'darn',
	'bitch(?!\w)': 'b-word',
	'toilet': 'potty',
	'(?<!\w)asshole': 'butthole',
	'(?<!\w)rape': 'r*pe',
	'(?<!\w)hell(?!\w)': 'heck',
	'(?<!\w)sex(?!\w)': 's*x',
	'(?<!\w)cum(?!\w)': 'c*m',
	'(?<!\w)dick': 'peepee',
	'cock(?!\w)': 'peepee',
	'cocks': 'peepees',
	'penis': 'peepee',
	'pussy': 'kitty',
	'pussies': 'kitties',
	'cunt': 'c*nt',
}



slur_single_words = "|".join([slur.lower() for slur in SLURS.keys()])
profanity_single_words = "|".join([profanity.lower() for profanity in PROFANITIES.keys()])
slur_regex = re.compile(f"<[^>]*>|{slur_single_words}", flags=re.I|re.A)
profanity_regex = re.compile(f"<[^>]*>|{profanity_single_words}", flags=re.I|re.A)

SLURS_FOR_REPLACING = {}
for k, val in SLURS.items():
	newkey = k.split('(?!')[0]
	if ')' in newkey:
		newkey = newkey.split(')')[1]
	SLURS_FOR_REPLACING[newkey] = val

PROFANITIES_FOR_REPLACING = {}
for k, val in PROFANITIES.items():
	newkey = k.split('(?!')[0]
	if ')' in newkey:
		newkey = newkey.split(')')[1]
	PROFANITIES_FOR_REPLACING[newkey] = val



def sub_matcher(match, X_FOR_REPLACING):
	group_num = 0
	match_str = match.group(group_num)
	if match_str.startswith('<'):
		return match_str
	else:
		repl = X_FOR_REPLACING[match_str.lower()]
		if "<img" not in repl:
			if match_str.isupper():
				return repl.upper()
			if match_str[0].isupper():
				return repl[0].upper() + repl[1:]
		return repl

def sub_matcher_slurs(match):
	return sub_matcher(match, SLURS_FOR_REPLACING)

def sub_matcher_profanities(match):
	return sub_matcher(match, PROFANITIES_FOR_REPLACING)



def censor_slurs_profanities(body, logged_user, is_plain=False):
	if not body: return ""

	if '<pre>' in body or '<code>' in body:
			return body

	if not logged_user or logged_user == 'chat' or logged_user.slurreplacer:
		body = slur_regex.sub(sub_matcher_slurs, body)

	if SITE_NAME == 'rDrama':
		if not logged_user or logged_user == 'chat' or logged_user.profanityreplacer:
			body = profanity_regex.sub(sub_matcher_profanities, body)

	if is_plain:
		body = body.replace(tranny, ':marseytrain:')
		body = body.replace(trannie, ':!marseytrain:')
		body = body.replace(troon, ':marseytrain2:')

	return body