require 'erb'
require 'yaml'

def generate_badge array
    badge = ""
    array.each { |t|
        case t
        when 'linux'
            badge = badge + "![](./images/linux.png)"
        when 'windows'
            badge = badge + "![](./images/windows.png)"
        when 'macos'
            badge = badge + "![](./images/apple.png)"
        when 'firefox'
            badge = badge + "![](./images/firefox.png)"
        when 'safari'
            badge = badge + "![](./images/safari.png)"
        when 'chrome'
            badge = badge + "![](./images/chrome.png)"
        end
    }
    
    return badge
end

template = %q{
<h1 align="center">
<br>
<a href=""><img src="https://user-images.githubusercontent.com/13212227/104400969-9f3d9280-5596-11eb-80f4-864effae95fc.png" alt="" width="500px;"></a>
<br>
<img src="https://img.shields.io/github/last-commit/hahwul/WebHackersWeapons?style=flat"> 
<img src="https://img.shields.io/badge/PRs-welcome-cyan">
<img src="https://github.com/hahwul/WebHackersWeapons/workflows/Build/badge.svg">
<a href="https://twitter.com/intent/follow?screen_name=hahwul"><img src="https://img.shields.io/twitter/follow/hahwul?style=flat&logo=twitter"></a>
</h1>
A collection of awesome tools used by Web hackers. Happy hacking , Happy bug-hunting

## Family project
[![WebHackersWeapons](https://img.shields.io/github/stars/hahwul/WebHackersWeapons?label=WebHackersWeapons)](https://github.com/hahwul/WebHackersWeapons)
[![MobileHackersWeapons](https://img.shields.io/github/stars/hahwul/MobileHackersWeapons?label=MobileHackersWeapons)](https://github.com/hahwul/MobileHackersWeapons)

## Table of Contents
- [Weapons](#weapons)
    - Tools
    - [Bookmarklets](https://github.com/hahwul/WebHackersWeapons/tree/master/Bookmarklets)
    - [Browser Extensions](https://github.com/hahwul/WebHackersWeapons/tree/master/Browser%20Extensions)
    - [Burp and ZAP Extensions](https://github.com/hahwul/WebHackersWeapons/tree/master/Burp%20and%20ZAP%20Extensions)
- [Contribute](https://github.com/hahwul/WebHackersWeapons/blob/master/CONTRIBUTING.md) 
- [Thanks to contributor](#thanks-to-contributor)

## Weapons
### Tools
<%= tools %>

### Bookmarklets
<%= bookmarklets %>

### Browser Addons
<%= browser_addons %>

### Burpsuite and ZAP Addons
<%= tool_addons %>

## Thanks to (Contributor)
I would like to thank everyone who helped with this project 👍😎 
![](/images/CONTRIBUTORS.svg)

}.gsub(/^  /, '')

head = "| Type | Name | Description | Badges | Popularity |\n"
head = head + "| --- | --- | --- | --- | --- |"
tools = head + "\n"
bookmarklets = head + "\n"
browser_addons = head + "\n"
tool_addons = head + "\n"

Dir.entries("./weapons/").each do | name |
    begin
        data = YAML.load(File.open("./weapons/#{name}"))
        name = data['name']
        popularity = "x"

        if data['url'].length > 0 
            name = "[#{name}](#{data['url']})"
        end

        if data['url'].include? "github.com"
            split_result = data['url'].split "//github.com/"
            popularity = "![](https://img.shields.io/github/stars/#{split_result[1]})"
        end
        badge = generate_badge(data['platform'])
        line = "|#{data['types']}|#{name}|#{data['description']}|#{badge}|#{popularity}|"
        case data['category'] 
        when 'tool'
            tools = tools + line + "\n"
        when 'tool-addon'
            tool_addons = tool_addons + line + "\n"
        when 'browser-addon'
            browser_addons = browser_addons + line + "\n"
        when 'bookmarklet'
            bookmarklets = bookmarklets + line + "\n"
        else
            puts name
        end
    rescue => e 
        puts e
    end
end

markdown = ERB.new(template, trim_mode: "%<>")
puts markdown.result