forked from bettercap/bettercap
/
ticker.go
100 lines (80 loc) · 1.97 KB
/
ticker.go
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
package modules
import (
"time"
"github.com/bettercap/bettercap/log"
"github.com/bettercap/bettercap/session"
)
type Ticker struct {
session.SessionModule
Period time.Duration
Commands []string
}
func NewTicker(s *session.Session) *Ticker {
t := &Ticker{
SessionModule: session.NewSessionModule("ticker", s),
}
t.AddParam(session.NewStringParameter("ticker.commands",
"clear; net.show",
"",
"List of commands separated by a ;"))
t.AddParam(session.NewIntParameter("ticker.period",
"1",
"Ticker period in seconds"))
t.AddHandler(session.NewModuleHandler("ticker on", "",
"Start the ticker.",
func(args []string) error {
return t.Start()
}))
t.AddHandler(session.NewModuleHandler("ticker off", "",
"Stop the ticker.",
func(args []string) error {
return t.Stop()
}))
return t
}
func (t *Ticker) Name() string {
return "ticker"
}
func (t *Ticker) Description() string {
return "A module to execute one or more commands every given amount of seconds."
}
func (t *Ticker) Author() string {
return "Simone Margaritelli <evilsocket@protonmail.com>"
}
func (t *Ticker) Configure() error {
var err error
var commands string
var period int
if err, commands = t.StringParam("ticker.commands"); err != nil {
return err
} else if err, period = t.IntParam("ticker.period"); err != nil {
return err
}
t.Commands = session.ParseCommands(commands)
t.Period = time.Duration(period) * time.Second
return nil
}
func (t *Ticker) Start() error {
if t.Running() == true {
return session.ErrAlreadyStarted
} else if err := t.Configure(); err != nil {
return err
}
return t.SetRunning(true, func() {
log.Info("Ticker running with period %.fs.", t.Period.Seconds())
tick := time.Tick(t.Period)
for range tick {
if t.Running() == false {
break
}
for _, cmd := range t.Commands {
if err := t.Session.Run(cmd); err != nil {
log.Error("%s", err)
}
}
}
})
}
func (t *Ticker) Stop() error {
return t.SetRunning(false, nil)
}