Skip to content

Commit

Permalink
Modify the copy
Browse files Browse the repository at this point in the history
  • Loading branch information
nrathaus committed May 14, 2024
1 parent 06e58ed commit a8be50d
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions lib/core/dictionary.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,23 +154,23 @@ def generate(self, files=[], is_blacklist=False):
if DOMAIN_TAG in new_line:
for hostname in hostnames:
split_hostnames = hostname.split(".")
new_line = new_line.replace(DOMAIN_TAG, hostname)
final_lines.append(new_line)
final_line = new_line.replace(DOMAIN_TAG, hostname)
final_lines.append(final_line)

if len(split_hostnames) > 1:
# We go from 1 dot to .. n .. as we want to return from www.somesite.co.uk:
# www.somesite.co.uk, somesite.co.uk, co.uk
for dots in range(1, len(split_hostnames)):
new_hostname = ".".join(split_hostnames[dots:])
new_line = new_line.replace(DOMAIN_TAG, new_hostname)
final_lines.append(new_line)
final_line = new_line.replace(DOMAIN_TAG, new_hostname)
final_lines.append(final_line)

# We go from n dot to .. 1 .. as we want to return from www.somesite.co.uk:
# www.somesite.co, www.somesite, www
for dots in range(1, len(split_hostnames)):
new_hostname = ".".join(split_hostnames[:dots])
new_line = new_line.replace(DOMAIN_TAG, new_hostname)
final_lines.append(new_line)
final_line = new_line.replace(DOMAIN_TAG, new_hostname)
final_lines.append(final_line)

new_lines = final_lines[:]

Expand All @@ -182,8 +182,8 @@ def generate(self, files=[], is_blacklist=False):
for new_line in new_lines:
if EXTENSION_TAG in new_line.lower():
for extension in options["extensions"]:
new_line = re_ext_tag.sub(extension, line)
final_lines.append(new_line)
final_line = re_ext_tag.sub(extension, line)
final_lines.append(final_line)

# If neither was triggered, just copy new_lines to our final outcome
if not domain_tag_triggered and not extension_tag_triggered:
Expand Down

0 comments on commit a8be50d

Please sign in to comment.