Parse only subdomain result from amass and make it hosts file

parse.lua

local filename = arg[1]

if not filename then
    print("Usage: lua parse.lua file:domain.com")
    os.exit(1)
end

local hostname = filename

local file = io.open(filename, "r")
if not file then
    print("Failed to open file:", filename)
    os.exit(1)
end

local data = file:read("*a")
file:close()

local results = {}

for domain, record_type in data:gmatch("([%w%.%-]+%.[a-zA-Z]+) %(FQDN%)%s*%-%-%>%s*([%w_]+)%s*%-%-%>") do
    -- Check if domain ends with the base domain
    if domain:match("%." .. hostname .. "$") or domain == hostname then
        results[domain] = record_type
    end
end

for domain, record_type in pairs(results) do
    -- print(string.format("%s => %s", domain, record_type))
    print(string.format("0.0.0.0 %s", domain))
end