electrosphere/printer.rb at master · kbcasurf/electrosphere · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
require 'colored'
require 'date'

module Printer
  def enable_color(flag = true)
    @color = flag
  end

  def print_title(s, level = 0)
    pad = " " * (level * 4)
    out_s = "#{pad}[*] #{s}"
    if @color
      puts out_s.bold.blue
    else
      puts out_s
    end
    @logger.info(out_s) if instance_variable_defined?("@logger")
  end

  def print_normal(s, level = 0)
    pad = " " * (level * 4)
    out_s = "#{pad}#{s}"
    puts out_s
    @logger.info(out_s) if instance_variable_defined?("@logger")
  end

  def print_success(s, level = 0)
    pad = " " * (level * 4)
    out_s = "#{pad}[SUCCESS] #{s}"
    if @color
      puts out_s.bold.green
    else
      puts out_s
    end
    @logger.info(out_s) if instance_variable_defined?("@logger")
  end

  def print_error(s, level = 0)
    pad = " " * (level * 4)
    out_s = "#{pad}[ ERROR ] #{s}"
    if @color
      puts out_s.bold.red
    else
      puts out_s
    end
    @logger.error(out_s) if instance_variable_defined?("@logger")
  end

  def print_with_label(s, label, level = 0)
    pad = " " * (level * 4)
    out_s = "#{pad}[#{label}] #{s}"
    puts out_s
    @logger.info(out_s) if instance_variable_defined?("@logger")
  end

  def print_debug(s, level = 0)
    pad = " " * (level * 4)
    now = DateTime.now.strftime('%d/%m/%Y %H:%M:%S.%3N')
    out_s = "#{pad}DEBUG|#{now}| #{s}"
    if @color
      puts out_s.bold.yellow
    else
      puts out_s
    end
    @logger.debug(out_s) if instance_variable_defined?("@logger")
  end

  def print_warning(s, level = 0)
    pad = " " * (level * 4)
    out_s = "#{pad}[WARNING] #{s}"
    if @color
      puts out_s.bold.yellow
    else
      puts out_s
    end
    @logger.info(out_s) if instance_variable_defined?("@logger")
  end

  def print_banner

      banner = <<-END

        ██████████ ████                     █████                                         █████
       ░░███░░░░░█░░███                    ░░███                                         ░░███
        ░███  █ ░  ░███   ██████   ██████  ███████   ████████   ██████   █████  ████████  ░███████    ██████  ████████   ██████
        ░██████    ░███  ███░░███ ███░░███░░░███░   ░░███░░███ ███░░███ ███░░  ░░███░░███ ░███░░███  ███░░███░░███░░███ ███░░███
        ░███░░█    ░███ ░███████ ░███ ░░░   ░███     ░███ ░░░ ░███ ░███░░█████  ░███ ░███ ░███ ░███ ░███████  ░███ ░░░ ░███████
        ░███ ░   █ ░███ ░███░░░  ░███  ███  ░███ ███ ░███     ░███ ░███ ░░░░███ ░███ ░███ ░███ ░███ ░███░░░   ░███     ░███░░░
        ██████████ █████░░██████ ░░██████   ░░█████  █████    ░░██████  ██████  ░███████  ████ █████░░██████  █████    ░░██████
       ░░░░░░░░░░ ░░░░░  ░░░░░░   ░░░░░░     ░░░░░  ░░░░░      ░░░░░░  ░░░░░░   ░███░░░  ░░░░ ░░░░░  ░░░░░░  ░░░░░      ░░░░░░
                                                                                ░███
                                                                                █████
                                                                                ░░░░░
        Electrosphere is a amazing microservice for registering
        vulnerabilities found by nuclei on Conviso Platform
        Developed by @rd-team

      END
    pad = " " * (0 * 4)
    out_s = "#{pad}[SUCCESS] #{banner}"
    if @color
      puts banner.bold.green
    else
      puts banner
    end
    @logger.info(banner) if instance_variable_defined?("@logger")
  end

  def self.included(base)
      #base.instance_variable_set(:@color, true)
  end
end