selenium_java_generator.rb

require 'open-uri'

# use it for give vars and functions name
$counter = 1000;

$array_rules_not_include = ["https://", "about:blank", "_blank", "rel=", "target="]

$special_char_to_match = /[!@#$%^&*().?":{}\[\]|<>;]/

$java_class_name = "SeleniumJavaGenerator"

class SeleniumJavaGenerator
        # check if string has latters
    def self.letters?(string)
        string.chars.any? { |char| ('a'..'z').include? char.downcase }
    end

    # Create java body with selenium import libs 
    def self.create_java_body()
        java_body = "import org.openqa.selenium.JavascriptExecutor;\n"
        java_body += "import org.openqa.selenium.WebDriver;\n"
        java_body += "import org.openqa.selenium.WebElement;\n"
        java_body += "import org.openqa.selenium.support.FindBy;\n\n\n"
        java_body += "public class #{$java_class_name} {\n"
        return java_body
    end

    # Read web page through url
    def self.read_web_page_url(url)
        open(url) do |file|
            read_html_by_lines(file)
        end
    end

    # Read web page through file
    def self.read_web_page_file(file_name)
        File.open(file_name, "a+") do |file|
            allText = file.readlines()
            read_html_by_lines(allText)
        end
    end


    # take html line by line to read
    def self.read_html_by_lines(html_source)
        print_html = create_java_body()
        html_source.each{ |line| 
            if line.match(/(class=|id=)/)
                new_java_code = create_web_element_object(line)
                if new_java_code.length > 0
                    print_html += new_java_code + "\n"
                end
            end
        }
        write_java_class(print_html + "}\n")      
    end


    # Read html line and search for id or class to create object
    def self.create_web_element_object(line)
        newLine = line.scan(/(\S+)=["']?((?:.(?!["']?\s+(?:\S+)=|[>"']))+.)["']?/)
        tag_name = line.scan(/<(\w+)\s+\w+.*?>/)
        java_code = ""
        index = 0
        while index < newLine.size
            current_item = newLine[index]  
            index2 = 0
            while index2 < current_item.size
                $counter += 1
                current_item2 =  newLine[index][index2]
                if current_item2 === ("id")
                    java_code = "//\t" + line + "\n"
                    java_code += check_and_create(newLine[index][index2 + 1], "id", nil)
                elsif current_item2 === ("class")
                    java_code = "//\t" + line + "\n"
                    if tag_name[0] != nil
                        java_code += check_and_create(newLine[index][index2 + 1], "css", tag_name[0][0])
                    end
                end
                index2 += 1 
            end
            index += 1
        end
        return java_code
    end


    def self.check_conditions(tag_check)
        $array_rules_not_include.each { |condition| 
            if tag_check.include?(condition)      
                return false
            end
        }
        return true
    end


    def self.check_and_create(item_class_name, find_type, tag_name)
        collector = ""
        
        if item_class_name != nil
            if  item_class_name.length > 3 &&
                check_conditions(item_class_name) &&
                letters?(item_class_name) &&
                !item_class_name.match($special_char_to_match)
                
                # check if first char is number if return do not continue !
                if item_class_name[0].match(/[0-9]/)
                    return "";
                end
            
                rename_item = item_class_name;
        
                # for css will rename to add tag name in the first name
                if(find_type == "css")
                    rename_item = css_element(item_class_name, tag_name)
                end
        
                
        
                ### <Create on the following format>
                
                # First: @FindBy([type]="[id | class]")
                collector += "\t\t@FindBy(#{find_type}=\"#{rename_item}\")" + "\n"
                
                # Second: Create the object and name it in camel case
                camel_case_item = camel_case(item_class_name, true)
                collector += "\t\tWebElement #{camel_case_item}_#{$counter};" + "\n\n"
        
                # Third: Create function method to start click
                collector += create_java_click_method(camel_case_item)
            end
        end

        return collector
    end


    def self.create_java_click_method(method_name)
        method = "\t\tpublic void clickOn#{camel_case(method_name, false)}_#{$counter}(){\n"
        method += "\t\t\t#{method_name}_#{$counter}.click();\n"
        method += "\t\t}\n\n"
        return method
    end


    # create camel case method
    def self.camel_case(item_name, downcase)
        
        # will take the string and convert to camel case
        new_item_name = item_name.gsub(/\s|\-|\__|\_/, ' ').split.map(&:capitalize)*''
        
        # make first latter down case
        if downcase
            new_item_name[0] = new_item_name[0].downcase
        end

        return new_item_name
    end


    def self.css_element(item_name, tag_name)
        return tag_name + "." + item_name.rstrip.lstrip.gsub(/(\s)+/, '.')
    end

    def self.write_java_class(html)
        puts "... Creating Java Class ...."
        File.open("./#{$java_class_name}.java", "a+") do |file|
            file.truncate(0)
            file << html
        end
    end
end