Compare commits
34 Commits
feat_docs
...
7256763c20
| Author | SHA1 | Date | |
|---|---|---|---|
| 7256763c20 | |||
| 54ddafd0fc | |||
| baeab6daf2 | |||
| 5818349d91 | |||
| b0723ae708 | |||
| 4b43bd1338 | |||
| 4fa1e6e4ef | |||
| 06c48f2927 | |||
| 54c0e8a46b | |||
| 7b4dcdd1d8 | |||
| 72d0aa61cd | |||
| e37a84d77d | |||
| 2d6ca2b0bd | |||
| df2b9e7624 | |||
| 46d1270648 | |||
| 5cbf066ccf | |||
| 515e244592 | |||
| fb4322c040 | |||
| f376d8684b | |||
| d0439394cf | |||
| a91632028c | |||
| bccdcf2ca3 | |||
| 18e3a93a38 | |||
| cfb553c975 | |||
| 1466623070 | |||
| c527d40721 | |||
| 781c096abf | |||
| c94fbd4b48 | |||
| 1f00713c1e | |||
| 12cf423550 | |||
| f99726d3b7 | |||
| 13932e572f | |||
| 20120785bd | |||
| 5af4c963ea |
5
.gitignore
vendored
5
.gitignore
vendored
@@ -1,8 +1,7 @@
|
||||
*.qcow2
|
||||
.envrc
|
||||
result
|
||||
example.json
|
||||
go.sum
|
||||
go.mod
|
||||
gokill
|
||||
./gokill
|
||||
output.md
|
||||
thoughts.md
|
||||
|
||||
122
README.md
Normal file
122
README.md
Normal file
@@ -0,0 +1,122 @@
|
||||
# gokill
|
||||
'gokill' is a tool that completes some actions when a certain event occurs.
|
||||
actions can vary from shuting down the machine to sending mails over erasing data.
|
||||
actions can be triggert by certain conditions like specific outcomes of unix
|
||||
comands or not having internet connection.
|
||||
|
||||
actions and triggers should be easy to extend and handled like plugins. they
|
||||
also should be self documenting.
|
||||
every action and trigger should be testable at anytime as a 'dry-run'.
|
||||
actions can have a 'stage' defined. the lowest stage is started first,
|
||||
and only when all actions on that stage are finished next stage is triggered
|
||||
|
||||
gokill should run as daemon. config should be read from /etc/somename/config.json
|
||||
|
||||
## Config Example
|
||||
``` json
|
||||
[ //list of triggers
|
||||
{
|
||||
"type": "UsbDisconnect",
|
||||
"name": "First Trigger",
|
||||
"options": {
|
||||
"deviceId": "ata-Samsung_SSD_860_EVO_1TB_S4AALKWJDI102",
|
||||
"waitTillConnected": true //only trigger when usb drive was actually attached before
|
||||
}
|
||||
"actions": [ //list of actions that will be executed when triggered
|
||||
{
|
||||
"name": "unixCommand",
|
||||
"options": {
|
||||
"command": "shutdown -h now"
|
||||
},
|
||||
"stage": 2 // defines the order in which actions are triggered.
|
||||
},
|
||||
{
|
||||
"type": "sendMail",
|
||||
"options": {
|
||||
"smtpserver": "domain.org",
|
||||
"port": 667,
|
||||
"recipients": [ "mail1@host.org", "mail2@host.org" ],
|
||||
"message": "kill switch was triggered",
|
||||
"attachments": [ "/path/atachments" ],
|
||||
"pubkeys": "/path/to/keys.pub"
|
||||
},
|
||||
"stage": 1 //this event is triggered first, then the shutdown
|
||||
},
|
||||
]
|
||||
},
|
||||
{
|
||||
"type": "EthernetDisconnect",
|
||||
"name": "Second Trigger",
|
||||
"options": {
|
||||
"interfaceName": "eth0",
|
||||
}
|
||||
"actions": [
|
||||
{
|
||||
"name": "unixCommand",
|
||||
"options": {
|
||||
"command": "env DISPLAY=:0 sudo su -c i3lock someUser"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
```
|
||||
|
||||
## nix support
|
||||
|
||||
gokill enjoys full nix support. gokill exposes a nix flakes that outputs a gokill package, a nixosModule and more.
|
||||
That means you can super easily incorporate gokill into your existing nixosConfigurations.
|
||||
Here is a small example config:
|
||||
|
||||
``` nix
|
||||
{
|
||||
services.gokill.enable = true;
|
||||
services.gokill.triggers = [
|
||||
{
|
||||
type = "EthernetDisconnect";
|
||||
name = "MainTrigger";
|
||||
options = {
|
||||
interfaceName = "eth1";
|
||||
};
|
||||
actions = [
|
||||
{
|
||||
type = "Command";
|
||||
options = {
|
||||
command = "echo hello world";
|
||||
};
|
||||
stage = 1;
|
||||
}
|
||||
];
|
||||
}
|
||||
];
|
||||
}
|
||||
```
|
||||
|
||||
This will automatically configure and enable a systemd running gokill as root user in the background
|
||||
|
||||
## actions
|
||||
- [x] shutdown
|
||||
- [ ] wipe ram
|
||||
- [ ] send mail
|
||||
- [ ] delete data
|
||||
- [ ] shred area
|
||||
- [x] random command
|
||||
- [ ] wordpress post
|
||||
- [ ] ipfs command
|
||||
- [ ] [buskill 'triggers'](https://github.com/BusKill/awesome-buskill-triggers)
|
||||
- [x] [lock-screen](https://github.com/BusKill/buskill-linux/tree/master/triggers)
|
||||
- [x] shutdown
|
||||
- [ ] luks header shredder
|
||||
- [ ] veracrypt self-destruct
|
||||
|
||||
## Triggers
|
||||
- [ ] no internet
|
||||
- [x] [pull usb stick](https://github.com/deepakjois/gousbdrivedetector/blob/master/usbdrivedetector_linux.go)
|
||||
- [x] ethernet unplugged
|
||||
- [ ] power adapter disconnected
|
||||
- [ ] unix command
|
||||
- anyOf
|
||||
- trigger wrapper containing many triggers and fires as soon as one of them
|
||||
is triggered
|
||||
- allOf
|
||||
- [ ] ipfs trigger
|
||||
@@ -7,10 +7,12 @@ import (
|
||||
"unknown.com/gokill/internal"
|
||||
)
|
||||
|
||||
type ActionResultChan chan error
|
||||
|
||||
type Action interface {
|
||||
Execute()
|
||||
DryExecute()
|
||||
Create(internal.ActionConfig, chan bool) (Action, error)
|
||||
Create(internal.ActionConfig, ActionResultChan) (Action, error)
|
||||
}
|
||||
|
||||
type DocumentedAction interface {
|
||||
@@ -23,7 +25,7 @@ type Stage struct {
|
||||
}
|
||||
|
||||
type StagedActions struct {
|
||||
ActionChan chan bool
|
||||
ActionChan ActionResultChan
|
||||
StageCount int
|
||||
Stages []Stage
|
||||
}
|
||||
@@ -40,7 +42,11 @@ func (a StagedActions) executeInternal(f func(Action)) {
|
||||
}
|
||||
|
||||
for range stage.Actions {
|
||||
<-a.ActionChan
|
||||
err := <-a.ActionChan
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error occured on Stage %d: %s\n", idx+1, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -64,11 +70,11 @@ func (a StagedActions) Execute() {
|
||||
a.executeInternal(func(a Action) { a.Execute() })
|
||||
}
|
||||
|
||||
func (a StagedActions) Create(config internal.ActionConfig, c chan bool) (Action, error) {
|
||||
func (a StagedActions) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
return StagedActions{}, nil
|
||||
}
|
||||
|
||||
func NewSingleAction(config internal.ActionConfig, c chan bool) (Action, error) {
|
||||
func NewSingleAction(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
for _, availableAction := range GetAllActions() {
|
||||
if config.Type == availableAction.GetName() {
|
||||
return availableAction.Create(config, c)
|
||||
@@ -83,7 +89,7 @@ func NewAction(config []internal.ActionConfig) (Action, error) {
|
||||
return config[i].Stage < config[j].Stage
|
||||
})
|
||||
|
||||
stagedActions := StagedActions{make(chan bool), 0, []Stage{}}
|
||||
stagedActions := StagedActions{make(ActionResultChan), 0, []Stage{}}
|
||||
|
||||
stageMap := make(map[int][]Action)
|
||||
|
||||
@@ -113,10 +119,13 @@ func NewAction(config []internal.ActionConfig) (Action, error) {
|
||||
|
||||
func GetAllActions() []DocumentedAction {
|
||||
return []DocumentedAction{
|
||||
Printer{},
|
||||
TimeOut{},
|
||||
Command{},
|
||||
Printer{},
|
||||
ShellScript{},
|
||||
Shutdown{},
|
||||
SendMatrix{},
|
||||
SendTelegram{},
|
||||
TimeOut{},
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -9,20 +9,20 @@ import (
|
||||
|
||||
type Printer struct {
|
||||
Message string
|
||||
ActionChan chan bool
|
||||
ActionChan ActionResultChan
|
||||
}
|
||||
|
||||
func (p Printer) Execute() {
|
||||
fmt.Printf("Print action fires. Message: %s\n", p.Message)
|
||||
p.ActionChan <- true
|
||||
p.ActionChan <- nil
|
||||
}
|
||||
|
||||
func (p Printer) DryExecute() {
|
||||
fmt.Printf("Print action fire test. Message: %s\n", p.Message)
|
||||
p.ActionChan <- true
|
||||
p.ActionChan <- nil
|
||||
}
|
||||
|
||||
func (p Printer) Create(config internal.ActionConfig, c chan bool) (Action, error) {
|
||||
func (p Printer) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
var result Printer
|
||||
err := json.Unmarshal(config.Options, &result)
|
||||
|
||||
@@ -39,11 +39,30 @@ func (p Printer) GetName() string {
|
||||
}
|
||||
|
||||
func (p Printer) GetDescription() string {
|
||||
return "When triggered prints the configured message to stdout"
|
||||
return `
|
||||
Prints a given message to stdout.
|
||||
This action is mostly used for debugging purposes.
|
||||
`
|
||||
}
|
||||
|
||||
func (p Printer) GetExample() string {
|
||||
return `
|
||||
{
|
||||
type: "Print",
|
||||
"options": {
|
||||
"message": "Hello World!"
|
||||
}
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p Printer) GetOptions() []internal.ConfigOption {
|
||||
return []internal.ConfigOption{
|
||||
{"message", "string", "Message that should be printed", "\"\""},
|
||||
{
|
||||
Name: "message",
|
||||
Type: "string",
|
||||
Description: "Message that should be printed",
|
||||
Default: "\"\"",
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
226
actions/send_matrix.go
Normal file
226
actions/send_matrix.go
Normal file
@@ -0,0 +1,226 @@
|
||||
package actions
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"encoding/json"
|
||||
|
||||
"context"
|
||||
"errors"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
|
||||
"maunium.net/go/mautrix"
|
||||
"maunium.net/go/mautrix/id"
|
||||
"maunium.net/go/mautrix/crypto/cryptohelper"
|
||||
|
||||
"unknown.com/gokill/internal"
|
||||
)
|
||||
|
||||
type SendMatrix struct {
|
||||
Homeserver string `json:"homeserver"`
|
||||
Username string `json:"username"`
|
||||
Password string `json:"password"`
|
||||
Database string `json:"database"`
|
||||
RoomId string `json:"roomId"`
|
||||
Message string `json:"message"`
|
||||
TestMessage string `json:"testMessage"`
|
||||
ActionChan ActionResultChan
|
||||
}
|
||||
|
||||
func (s SendMatrix) sendMessage(message string) error {
|
||||
client, err := mautrix.NewClient(s.Homeserver, "", "")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
cryptoHelper, err := cryptohelper.NewCryptoHelper(client, []byte("meow"), s.Database)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
cryptoHelper.LoginAs = &mautrix.ReqLogin{
|
||||
Type: mautrix.AuthTypePassword,
|
||||
Identifier: mautrix.UserIdentifier{Type: mautrix.IdentifierTypeUser, User: s.Username},
|
||||
Password: s.Password,
|
||||
}
|
||||
|
||||
err = cryptoHelper.Init()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
client.Crypto = cryptoHelper
|
||||
|
||||
fmt.Println("Matrix Client Now running")
|
||||
syncCtx, cancelSync := context.WithCancel(context.Background())
|
||||
var syncStopWait sync.WaitGroup
|
||||
syncStopWait.Add(1)
|
||||
|
||||
go func() {
|
||||
err = client.SyncWithContext(syncCtx)
|
||||
defer syncStopWait.Done()
|
||||
if err != nil && !errors.Is(err, context.Canceled) {
|
||||
return
|
||||
}
|
||||
}()
|
||||
|
||||
time.Sleep(5 * time.Second)
|
||||
resp, err := client.SendText(id.RoomID(s.RoomId), message)
|
||||
|
||||
if err != nil {
|
||||
return fmt.Errorf("Failed to send event")
|
||||
} else {
|
||||
fmt.Println("Matrix Client: Message sent")
|
||||
fmt.Printf("Matrix Client: event_id: %s", resp.EventID.String())
|
||||
}
|
||||
|
||||
cancelSync()
|
||||
syncStopWait.Wait()
|
||||
err = cryptoHelper.Close()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error closing database")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s SendMatrix) DryExecute() {
|
||||
fmt.Println("SendMatrix: Trying to send test message")
|
||||
err := s.sendMessage(s.TestMessage)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("SendMatrix: failed to send test message")
|
||||
}
|
||||
|
||||
s.ActionChan <- err
|
||||
}
|
||||
|
||||
func (s SendMatrix) Execute() {
|
||||
fmt.Println("SendMatrix: Trying to send message")
|
||||
err := s.sendMessage(s.Message)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("SendMatrix: failed to send message")
|
||||
}
|
||||
|
||||
s.ActionChan <- err
|
||||
}
|
||||
|
||||
func CreateSendMatrix(config internal.ActionConfig, c ActionResultChan) (SendMatrix, error) {
|
||||
result := SendMatrix{}
|
||||
|
||||
err := json.Unmarshal(config.Options, &result)
|
||||
|
||||
if err != nil {
|
||||
return SendMatrix{}, err
|
||||
}
|
||||
|
||||
if result.Homeserver == "" {
|
||||
return SendMatrix{}, internal.OptionMissingError{"homeserver"}
|
||||
}
|
||||
|
||||
if result.Username == "" {
|
||||
return SendMatrix{}, internal.OptionMissingError{"username"}
|
||||
}
|
||||
|
||||
if result.Password == "" {
|
||||
return SendMatrix{}, internal.OptionMissingError{"password"}
|
||||
}
|
||||
|
||||
if result.Database == "" {
|
||||
return SendMatrix{}, internal.OptionMissingError{"database"}
|
||||
}
|
||||
|
||||
if result.RoomId == "" {
|
||||
return SendMatrix{}, internal.OptionMissingError{"roomId"}
|
||||
}
|
||||
|
||||
if result.Message == "" {
|
||||
return SendMatrix{}, internal.OptionMissingError{"message"}
|
||||
}
|
||||
|
||||
if result.TestMessage == "" {
|
||||
return SendMatrix{}, internal.OptionMissingError{"testMessage"}
|
||||
}
|
||||
|
||||
result.ActionChan = c
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (s SendMatrix) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
return CreateSendMatrix(config, c)
|
||||
}
|
||||
|
||||
func (p SendMatrix) GetName() string {
|
||||
return "SendMatrix"
|
||||
}
|
||||
|
||||
func (p SendMatrix) GetDescription() string {
|
||||
return "Sends a message to a given room. The user needs to be part of that room already."
|
||||
}
|
||||
|
||||
func (p SendMatrix) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "SendMatrix",
|
||||
"options": {
|
||||
"homeserver": "matrix.org",
|
||||
"username": "testuser",
|
||||
"password": "super-secret",
|
||||
"database": "/etc/gokill/matrix.db",
|
||||
"roomId": "!Balrthajskensaw:matrix.org",
|
||||
"message": "attention, intruders got my device!",
|
||||
"testMessage": "this is just a test, no worries"
|
||||
}
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p SendMatrix) GetOptions() []internal.ConfigOption {
|
||||
return []internal.ConfigOption{
|
||||
{
|
||||
Name: "homeserver",
|
||||
Type: "string",
|
||||
Description: "homeserver address.",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "username",
|
||||
Type: "string",
|
||||
Description: "username (localpart, wihout homeserver address)",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "password",
|
||||
Type: "string",
|
||||
Description: "password in clear text",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "database",
|
||||
Type: "string",
|
||||
Description: "path to database file, will be created if not existing. this is necessary to sync keys for encryption.",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "roomId",
|
||||
Type: "string",
|
||||
Description: "",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "message",
|
||||
Type: "string",
|
||||
Description: "actual message that should be sent",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "testMessage",
|
||||
Type: "string",
|
||||
Description: "message sent during test run",
|
||||
Default: "",
|
||||
},
|
||||
}
|
||||
}
|
||||
143
actions/send_telegram.go
Normal file
143
actions/send_telegram.go
Normal file
@@ -0,0 +1,143 @@
|
||||
package actions
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"encoding/json"
|
||||
|
||||
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
||||
|
||||
"unknown.com/gokill/internal"
|
||||
)
|
||||
|
||||
type SendTelegram struct {
|
||||
Token string `json:"token"`
|
||||
ChatId string `json:"chatId"`
|
||||
Message string `json:"message"`
|
||||
TestMessage string `json:"testMessage"`
|
||||
ActionChan ActionResultChan
|
||||
}
|
||||
|
||||
func (s SendTelegram) sendMessage(message string) error {
|
||||
bot, err := tgbotapi.NewBotAPI("5221828879:AAGrETcxOpAhzPJUl-fUqMJGUAe6ShSuuRs")
|
||||
if err != nil {
|
||||
return fmt.Errorf("SendTelegram sendMessage error: %s", err)
|
||||
}
|
||||
|
||||
bot.Debug = true
|
||||
|
||||
fmt.Printf("Authorized on account %s", bot.Self.UserName)
|
||||
|
||||
u := tgbotapi.NewUpdate(0)
|
||||
u.Timeout = 60
|
||||
|
||||
msg := tgbotapi.NewMessage(-746157642, message)
|
||||
bot.Send(msg)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s SendTelegram) DryExecute() {
|
||||
fmt.Println("SendTelegram: Trying to send test message")
|
||||
err := s.sendMessage(s.TestMessage)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("SendTelegram: failed to send test message")
|
||||
}
|
||||
|
||||
s.ActionChan <- err
|
||||
}
|
||||
|
||||
func (s SendTelegram) Execute() {
|
||||
fmt.Println("SendTelegram: Trying to send message")
|
||||
err := s.sendMessage(s.Message)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("SendTelegram: failed to send message")
|
||||
}
|
||||
|
||||
s.ActionChan <- err
|
||||
}
|
||||
|
||||
func CreateSendTelegram(config internal.ActionConfig, c ActionResultChan) (SendTelegram, error) {
|
||||
result := SendTelegram{}
|
||||
|
||||
err := json.Unmarshal(config.Options, &result)
|
||||
|
||||
if err != nil {
|
||||
return SendTelegram{}, err
|
||||
}
|
||||
|
||||
if result.Token == "" {
|
||||
return SendTelegram{}, internal.OptionMissingError{"token"}
|
||||
}
|
||||
|
||||
if result.ChatId == "" {
|
||||
return SendTelegram{}, internal.OptionMissingError{"chatId"}
|
||||
}
|
||||
|
||||
if result.Message == "" {
|
||||
return SendTelegram{}, internal.OptionMissingError{"message"}
|
||||
}
|
||||
|
||||
if result.TestMessage == "" {
|
||||
return SendTelegram{}, internal.OptionMissingError{"testMessage"}
|
||||
}
|
||||
|
||||
result.ActionChan = c
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (s SendTelegram) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
return CreateSendTelegram(config, c)
|
||||
}
|
||||
|
||||
func (p SendTelegram) GetName() string {
|
||||
return "SendTelegram"
|
||||
}
|
||||
|
||||
func (p SendTelegram) GetDescription() string {
|
||||
return "Sends a message to a given room. The user needs to be part of that room already."
|
||||
}
|
||||
|
||||
func (p SendTelegram) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "SendTelegram",
|
||||
"options": {
|
||||
"token": "5349923487:FFGrETxa0pA29d02Akslw-lkwjdA92KAH2",
|
||||
"chatId": "-832345892",
|
||||
"message": "attention, intruders got my device!",
|
||||
"testMessage": "this is just a test, no worries"
|
||||
}
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p SendTelegram) GetOptions() []internal.ConfigOption {
|
||||
return []internal.ConfigOption{
|
||||
{
|
||||
Name: "token",
|
||||
Type: "string",
|
||||
Description: "telegram bot token (ask botfather)",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "chatId",
|
||||
Type: "int",
|
||||
Description: "chatId of group or chat you want the message be sent to.",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "message",
|
||||
Type: "string",
|
||||
Description: "actual message that should be sent",
|
||||
Default: "",
|
||||
},
|
||||
{
|
||||
Name: "testMessage",
|
||||
Type: "string",
|
||||
Description: "message sent during test run",
|
||||
Default: "",
|
||||
},
|
||||
}
|
||||
}
|
||||
119
actions/shell_script.go
Normal file
119
actions/shell_script.go
Normal file
@@ -0,0 +1,119 @@
|
||||
package actions
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os/exec"
|
||||
"os"
|
||||
|
||||
"unknown.com/gokill/internal"
|
||||
)
|
||||
|
||||
type ShellScript struct {
|
||||
Path string `json:"path"`
|
||||
ActionChan ActionResultChan
|
||||
}
|
||||
|
||||
func isExecutableFile(path string) bool {
|
||||
fi, err := os.Lstat(path)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("Test executing Shellscript Failed.")
|
||||
return false
|
||||
}
|
||||
|
||||
mode := fi.Mode()
|
||||
|
||||
//TODO: should check if current user can execute
|
||||
if mode&01111 == 0 {
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func (c ShellScript) DryExecute() {
|
||||
fmt.Printf("Test Executing ShellScript:\n%s\n", c.Path)
|
||||
|
||||
_, err := os.Open(c.Path)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("Test executing Shellscript Failed.")
|
||||
c.ActionChan <- err
|
||||
return
|
||||
}
|
||||
|
||||
if !isExecutableFile(c.Path) {
|
||||
fmt.Println("Test executing Shellscript Failed.")
|
||||
c.ActionChan <- fmt.Errorf("File is not executable: %s", c.Path)
|
||||
return
|
||||
}
|
||||
|
||||
c.ActionChan <- nil
|
||||
}
|
||||
|
||||
func (c ShellScript) Execute() {
|
||||
if !isExecutableFile(c.Path) {
|
||||
fmt.Println("Test executing Shellscript Failed.")
|
||||
c.ActionChan <- fmt.Errorf("File is not executable: %s", c.Path)
|
||||
return
|
||||
}
|
||||
|
||||
cmd := exec.Command("/bin/sh", c.Path)
|
||||
|
||||
stdout, err := cmd.Output()
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
c.ActionChan <- err
|
||||
}
|
||||
|
||||
fmt.Println(string(stdout[:]))
|
||||
c.ActionChan <- nil
|
||||
}
|
||||
|
||||
func CreateShellScript(config internal.ActionConfig, c ActionResultChan) (ShellScript, error) {
|
||||
result := ShellScript{}
|
||||
|
||||
err := json.Unmarshal(config.Options, &result)
|
||||
|
||||
if err != nil {
|
||||
return ShellScript{}, err
|
||||
}
|
||||
|
||||
if result.Path == "" {
|
||||
return ShellScript{}, internal.OptionMissingError{"path"}
|
||||
}
|
||||
|
||||
result.ActionChan = c
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (cc ShellScript) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
return CreateShellScript(config, c)
|
||||
}
|
||||
|
||||
func (p ShellScript) GetName() string {
|
||||
return "ShellScript"
|
||||
}
|
||||
|
||||
func (p ShellScript) GetDescription() string {
|
||||
return "Executes the given shell script."
|
||||
}
|
||||
|
||||
func (p ShellScript) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "ShellScript",
|
||||
"options": {
|
||||
"path": "/path/to/file.sh"
|
||||
}
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p ShellScript) GetOptions() []internal.ConfigOption {
|
||||
return []internal.ConfigOption{
|
||||
{"path", "string", "path to script to execute", ""},
|
||||
}
|
||||
}
|
||||
@@ -3,33 +3,46 @@ package actions
|
||||
import (
|
||||
"fmt"
|
||||
"os/exec"
|
||||
"encoding/json"
|
||||
|
||||
"unknown.com/gokill/internal"
|
||||
)
|
||||
|
||||
type Shutdown struct {
|
||||
ActionChan chan bool
|
||||
Timeout string `json:"time"`
|
||||
ActionChan ActionResultChan
|
||||
}
|
||||
|
||||
func (s Shutdown) DryExecute() {
|
||||
fmt.Printf("shutdown -h %s\n", s.Timeout)
|
||||
fmt.Println("Test Shutdown executed...")
|
||||
|
||||
s.ActionChan <- true
|
||||
|
||||
s.ActionChan <- nil
|
||||
}
|
||||
|
||||
func (s Shutdown) Execute() {
|
||||
if err := exec.Command("shutdown", "-h", "now").Run(); err != nil {
|
||||
if err := exec.Command("shutdown", "-h", s.Timeout).Run(); err != nil {
|
||||
fmt.Println("Failed to initiate shutdown:", err)
|
||||
}
|
||||
|
||||
fmt.Println("Shutdown executed...")
|
||||
|
||||
s.ActionChan <- true
|
||||
s.ActionChan <- nil
|
||||
}
|
||||
|
||||
func (s Shutdown) Create(config internal.ActionConfig, c chan bool) (Action, error) {
|
||||
return Shutdown{c}, nil
|
||||
func (s Shutdown) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
result := Shutdown{
|
||||
Timeout: "now",
|
||||
}
|
||||
|
||||
err := json.Unmarshal(config.Options, &result)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("Parsing Shutdown options failed.")
|
||||
return Shutdown{}, err
|
||||
}
|
||||
|
||||
result.ActionChan = c
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (p Shutdown) GetName() string {
|
||||
@@ -37,9 +50,27 @@ func (p Shutdown) GetName() string {
|
||||
}
|
||||
|
||||
func (p Shutdown) GetDescription() string {
|
||||
return "When triggered shuts down the machine"
|
||||
return "Shutsdown the machine by perfoming a ```shutdown -h now```"
|
||||
}
|
||||
|
||||
func (p Shutdown) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "Shutdown",
|
||||
"options": {
|
||||
"time": "+5" //wait 5 minutes before shutdown
|
||||
}
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p Shutdown) GetOptions() []internal.ConfigOption {
|
||||
return []internal.ConfigOption{}
|
||||
return []internal.ConfigOption{
|
||||
{
|
||||
Name: "time",
|
||||
Type: "string",
|
||||
Description: "TIME parameter passed to shutdown as follows ```shutdown -h TIME```",
|
||||
Default: "now",
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
@@ -10,7 +10,7 @@ import (
|
||||
|
||||
type TimeOut struct {
|
||||
Duration time.Duration
|
||||
ActionChan chan bool
|
||||
ActionChan ActionResultChan
|
||||
}
|
||||
|
||||
func (t TimeOut) DryExecute() {
|
||||
@@ -20,10 +20,10 @@ func (t TimeOut) DryExecute() {
|
||||
func (t TimeOut) Execute() {
|
||||
fmt.Printf("Waiting %d seconds\n", t.Duration)
|
||||
time.Sleep(time.Duration(t.Duration) * time.Second)
|
||||
t.ActionChan <- true
|
||||
t.ActionChan <- nil
|
||||
}
|
||||
|
||||
func (t TimeOut) Create(config internal.ActionConfig, c chan bool) (Action, error) {
|
||||
func (t TimeOut) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
var result TimeOut
|
||||
err := json.Unmarshal(config.Options, &result)
|
||||
|
||||
@@ -40,7 +40,21 @@ func (p TimeOut) GetName() string {
|
||||
}
|
||||
|
||||
func (p TimeOut) GetDescription() string {
|
||||
return "When triggered waits given duration before continuing with next stage"
|
||||
return `
|
||||
Waits given duration in seconds.
|
||||
This can be used to wait a certain amount of time before continuing to the next Stage
|
||||
`
|
||||
}
|
||||
|
||||
func (p TimeOut) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "Timeout",
|
||||
"options": {
|
||||
"duration": 5
|
||||
}
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p TimeOut) GetOptions() []internal.ConfigOption {
|
||||
|
||||
@@ -11,12 +11,37 @@ import (
|
||||
|
||||
type Command struct {
|
||||
Command string `json:"command"`
|
||||
ActionChan chan bool
|
||||
ActionChan ActionResultChan
|
||||
}
|
||||
|
||||
func isCommandAvailable(name string) bool {
|
||||
cmd := exec.Command("/bin/sh", "-c", "command -v "+name)
|
||||
if err := cmd.Run(); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func (c Command) DryExecute() {
|
||||
fmt.Printf("Test Executing Command:\n%s ", c.Command)
|
||||
c.ActionChan <- true
|
||||
fmt.Printf("Test Executing Command:\n%s\n", c.Command)
|
||||
command, _, err := c.splitCommandString()
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error during argument parsing of command '%s'\n", c.Command)
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
|
||||
isAvailable := isCommandAvailable(command)
|
||||
|
||||
if !isAvailable {
|
||||
fmt.Printf("Command %s not found\n", command)
|
||||
c.ActionChan <- fmt.Errorf("Command %s not found!", command)
|
||||
return
|
||||
}
|
||||
|
||||
c.ActionChan <- nil
|
||||
}
|
||||
|
||||
func (c Command) splitCommandString() (string, []string, error) {
|
||||
@@ -38,8 +63,7 @@ func (c Command) Execute() {
|
||||
fmt.Println("Executing command: ", c.Command)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
c.ActionChan <- false
|
||||
c.ActionChan <- err
|
||||
return
|
||||
}
|
||||
|
||||
@@ -49,14 +73,15 @@ func (c Command) Execute() {
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
c.ActionChan <- err
|
||||
}
|
||||
|
||||
fmt.Println(string(stdout[:]))
|
||||
|
||||
c.ActionChan <- true
|
||||
c.ActionChan <- nil
|
||||
}
|
||||
|
||||
func CreateCommand(config internal.ActionConfig, c chan bool) (Command, error) {
|
||||
func CreateCommand(config internal.ActionConfig, c ActionResultChan) (Command, error) {
|
||||
result := Command{}
|
||||
|
||||
err := json.Unmarshal(config.Options, &result)
|
||||
@@ -74,7 +99,7 @@ func CreateCommand(config internal.ActionConfig, c chan bool) (Command, error) {
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (cc Command) Create(config internal.ActionConfig, c chan bool) (Action, error) {
|
||||
func (cc Command) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
||||
return CreateCommand(config, c)
|
||||
}
|
||||
|
||||
@@ -83,7 +108,18 @@ func (p Command) GetName() string {
|
||||
}
|
||||
|
||||
func (p Command) GetDescription() string {
|
||||
return "When triggered executes given command"
|
||||
return "Invoces given command using exec."
|
||||
}
|
||||
|
||||
func (p Command) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "Command",
|
||||
"options": {
|
||||
"command": "srm /path/to/file"
|
||||
}
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p Command) GetOptions() []internal.ConfigOption {
|
||||
|
||||
102
cmd/docbuilder/docbuilder.go
Normal file
102
cmd/docbuilder/docbuilder.go
Normal file
@@ -0,0 +1,102 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
"os"
|
||||
"flag"
|
||||
|
||||
"unknown.com/gokill/actions"
|
||||
"unknown.com/gokill/triggers"
|
||||
"unknown.com/gokill/internal"
|
||||
)
|
||||
|
||||
func getMarkdown(documenter internal.Documenter) string {
|
||||
var result string
|
||||
result += fmt.Sprintf("# %v\n%v\n\n", documenter.GetName(), documenter.GetDescription())
|
||||
|
||||
result += fmt.Sprintf("*Example:*\n``` json\n%v\n```\n## Options:\n", documenter.GetExample())
|
||||
|
||||
for _, opt := range documenter.GetOptions() {
|
||||
sanitizedDefault := "\"\""
|
||||
|
||||
if len(opt.Default) > 0 {
|
||||
sanitizedDefault = opt.Default
|
||||
}
|
||||
|
||||
result += fmt.Sprintf("### %v\n%v \n\n*Type:* %v \n\n*Default:* ```%v``` \n",
|
||||
opt.Name, opt.Description, opt.Type, sanitizedDefault)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func writeToFile(path string, documenter internal.Documenter) error {
|
||||
fileName := fmt.Sprintf("%s/%s.md", path, documenter.GetName())
|
||||
|
||||
f, err := os.Create(fileName)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return err
|
||||
}
|
||||
|
||||
defer f.Close()
|
||||
|
||||
_, err = f.WriteString(getMarkdown(documenter))
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func writeDocumentersToFiles(destination string) {
|
||||
writeFolder := func(typeName string, documenters []internal.Documenter) {
|
||||
path := fmt.Sprintf("%s/%s", destination, typeName)
|
||||
_ = os.Mkdir(path, os.ModePerm)
|
||||
for _, documenter := range documenters {
|
||||
writeToFile(path, documenter)
|
||||
}
|
||||
}
|
||||
|
||||
actions := actions.GetDocumenters()
|
||||
writeFolder("actions", actions)
|
||||
|
||||
triggers := triggers.GetDocumenters()
|
||||
writeFolder("triggers", triggers)
|
||||
}
|
||||
|
||||
func printDocumentersSummary() {
|
||||
result := fmt.Sprintf("- [Triggers](triggers/README.md)\n")
|
||||
for _, trigger := range triggers.GetDocumenters() {
|
||||
result += fmt.Sprintf("\t- [%s](triggers/%s.md)\n", trigger.GetName(), trigger.GetName())
|
||||
}
|
||||
|
||||
result += fmt.Sprintf("- [Actions](actions/README.md)\n")
|
||||
for _, action := range actions.GetDocumenters() {
|
||||
result += fmt.Sprintf("\t- [%s](actions/%s.md)\n", action.GetName(), action.GetName())
|
||||
}
|
||||
|
||||
fmt.Print(result)
|
||||
}
|
||||
|
||||
|
||||
func main() {
|
||||
outputPath := flag.String("output", "", "path where docs/ shoud be created")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
if *outputPath == "" {
|
||||
printDocumentersSummary()
|
||||
return
|
||||
}
|
||||
|
||||
if len(*outputPath) > 1 {
|
||||
*outputPath = strings.TrimSuffix(*outputPath, "/")
|
||||
}
|
||||
|
||||
writeDocumentersToFiles(*outputPath)
|
||||
}
|
||||
1
docs/.gitignore
vendored
Normal file
1
docs/.gitignore
vendored
Normal file
@@ -0,0 +1 @@
|
||||
book/*
|
||||
4
docs/SUMMARY.md
Normal file
4
docs/SUMMARY.md
Normal file
@@ -0,0 +1,4 @@
|
||||
# Summary
|
||||
|
||||
- [gokill](./README.md)
|
||||
@GOKILL_OPTIONS@
|
||||
21
docs/actions/README.md
Normal file
21
docs/actions/README.md
Normal file
@@ -0,0 +1,21 @@
|
||||
# Actions
|
||||
|
||||
Actions are executed when their parent Trigger got triggered.
|
||||
They then perform some certain task depending on the specific action.
|
||||
Those can vary from shutding down the machine, removing a file or running a bash command.
|
||||
**Some Actions may cause permanent damage to the system. This is intended but should be used with caution.**
|
||||
|
||||
Actions can have a ```Stage``` assigned to define in which order they should run.
|
||||
The lowest stage is executed first and only when finished the next stage is executed.
|
||||
Actions on the same Stage run concurrently.
|
||||
|
||||
Actions have the following syntax:
|
||||
``` json
|
||||
{
|
||||
"type": "SomeAction",
|
||||
"options": { //each action defines its own options
|
||||
"firstOption": "someValue",
|
||||
"stage": 2 //this (positive) number defines the order of multiple actions
|
||||
}
|
||||
}
|
||||
```
|
||||
10
docs/book.toml
Normal file
10
docs/book.toml
Normal file
@@ -0,0 +1,10 @@
|
||||
[book]
|
||||
authors = []
|
||||
language = "en"
|
||||
multilingual = false
|
||||
src = "."
|
||||
title = "gokill docs"
|
||||
|
||||
[output.html.fold]
|
||||
enable = true
|
||||
level = 0
|
||||
33
docs/default.nix
Normal file
33
docs/default.nix
Normal file
@@ -0,0 +1,33 @@
|
||||
{ pkgs, lib, self, ... }:
|
||||
|
||||
with lib;
|
||||
let
|
||||
docbuilder = self.packages.x86_64-linux.gokill-docbuilder;
|
||||
|
||||
prepareMD = ''
|
||||
# Copy inputs into the build directory
|
||||
cp -r --no-preserve=all $inputs/* ./
|
||||
cp ${../README.md} ./README.md
|
||||
|
||||
${docbuilder}/bin/docbuilder --output ./
|
||||
substituteInPlace ./SUMMARY.md \
|
||||
--replace "@GOKILL_OPTIONS@" "$(${docbuilder}/bin/docbuilder)"
|
||||
|
||||
cat ./SUMMARY.md
|
||||
'';
|
||||
in
|
||||
pkgs.stdenv.mkDerivation {
|
||||
name = "gokill-docs";
|
||||
phases = [ "buildPhase" ];
|
||||
buildInputs = [ pkgs.mdbook ];
|
||||
|
||||
inputs = sourceFilesBySuffices ./. [ ".md" ".toml" ];
|
||||
|
||||
buildPhase = ''
|
||||
dest=$out/share/doc
|
||||
mkdir -p $dest
|
||||
${prepareMD}
|
||||
mdbook build
|
||||
cp -r ./book/* $dest
|
||||
'';
|
||||
}
|
||||
19
docs/triggers/README.md
Normal file
19
docs/triggers/README.md
Normal file
@@ -0,0 +1,19 @@
|
||||
# Triggers
|
||||
|
||||
Triggers wait for certain events and execute the actions defined for them.
|
||||
There are different Triggers for different use cases.
|
||||
For example ```UsbDisconnect``` is triggered when a certain Usb Drive is unplugged.
|
||||
If you want your actions to be triggered when an ethernet cable is pulled use ```EthernetDisconnect``` instead.
|
||||
|
||||
Triggers have the following syntax:
|
||||
``` json
|
||||
{
|
||||
"type": "SomeTrigger",
|
||||
"name": "MyFirstTrigger",
|
||||
"options": { //each trigger defines its own options
|
||||
"firstOption": 23,
|
||||
"secondOption": "foo"
|
||||
},
|
||||
"actions": [] //list actions that should be executed here
|
||||
}
|
||||
```
|
||||
42
flake.lock
generated
42
flake.lock
generated
@@ -2,11 +2,11 @@
|
||||
"nodes": {
|
||||
"nixpkgs": {
|
||||
"locked": {
|
||||
"lastModified": 1697915759,
|
||||
"narHash": "sha256-WyMj5jGcecD+KC8gEs+wFth1J1wjisZf8kVZH13f1Zo=",
|
||||
"lastModified": 1698553279,
|
||||
"narHash": "sha256-T/9P8yBSLcqo/v+FTOBK+0rjzjPMctVymZydbvR/Fak=",
|
||||
"owner": "nixos",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "51d906d2341c9e866e48c2efcaac0f2d70bfd43e",
|
||||
"rev": "90e85bc7c1a6fc0760a94ace129d3a1c61c3d035",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
@@ -18,7 +18,41 @@
|
||||
},
|
||||
"root": {
|
||||
"inputs": {
|
||||
"nixpkgs": "nixpkgs"
|
||||
"nixpkgs": "nixpkgs",
|
||||
"utils": "utils"
|
||||
}
|
||||
},
|
||||
"systems": {
|
||||
"locked": {
|
||||
"lastModified": 1681028828,
|
||||
"narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"utils": {
|
||||
"inputs": {
|
||||
"systems": "systems"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1694529238,
|
||||
"narHash": "sha256-zsNZZGTGnMOf9YpHKJqMSsa0dXbfmxeoJ7xHlrt+xmY=",
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"rev": "ff7b65b44d01cf9ba6a71320833626af21126384",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"type": "github"
|
||||
}
|
||||
}
|
||||
},
|
||||
|
||||
151
flake.nix
151
flake.nix
@@ -4,118 +4,100 @@
|
||||
#nixpkgs for testing framework
|
||||
inputs.nixpkgs.url = "github:nixos/nixpkgs/nixpkgs-unstable";
|
||||
|
||||
outputs = { self, nixpkgs, ... }:
|
||||
inputs.utils.url = "github:numtide/flake-utils";
|
||||
|
||||
outputs = { self, nixpkgs, utils, ... }:
|
||||
|
||||
nixpkgs.lib.attrsets.recursiveUpdate
|
||||
(utils.lib.eachSystem (utils.lib.defaultSystems) ( system:
|
||||
let
|
||||
forAllSystems = nixpkgs.lib.genAttrs [ "x86_64-linux" ];
|
||||
pkgs = nixpkgs.legacyPackages.${system};
|
||||
in
|
||||
{
|
||||
packages.x86_64-linux.gokill = nixpkgs.legacyPackages.x86_64-linux.buildGoModule rec {
|
||||
devShells.default = pkgs.mkShell {
|
||||
packages = with pkgs; [
|
||||
go
|
||||
gotools
|
||||
mdbook
|
||||
olm
|
||||
];
|
||||
};
|
||||
|
||||
packages = {
|
||||
gokill = pkgs.buildGoModule rec {
|
||||
pname = "gokill";
|
||||
version = "1.0";
|
||||
vendorHash = "sha256-aKEOMeW9QVSLsSuHV4b1khqM0rRrMjJ6Eu5RjY+6V8k=";
|
||||
vendorHash = "sha256-8xHjVwNskgiSoDKbOpL2phHub1F21ios1t9BcZB944o=";
|
||||
src = ./.;
|
||||
|
||||
buildInputs = [
|
||||
pkgs.olm
|
||||
];
|
||||
|
||||
postInstall = ''
|
||||
'';
|
||||
};
|
||||
|
||||
packages.x86_64-linux.default = self.packages.x86_64-linux.gokill;
|
||||
gokill-docbuilder = pkgs.buildGoModule rec {
|
||||
pname = "docbuilder";
|
||||
version = "1.0";
|
||||
vendorHash = "sha256-8xHjVwNskgiSoDKbOpL2phHub1F21ios1t9BcZB944o=";
|
||||
buildFLags = "-o . $dest/cmd/gokill/docbuilder";
|
||||
src = ./.;
|
||||
|
||||
nixosModules.gokill = { config, lib, pkgs, ... }:
|
||||
let
|
||||
cfg = config.services.gokill;
|
||||
configFile = pkgs.writeText "config.json" ''${cfg.extraConfig}'';
|
||||
gokill-pkg = self.packages.x86_64-linux.gokill;
|
||||
in
|
||||
{
|
||||
options = {
|
||||
services.gokill = {
|
||||
enable = lib.mkOption {
|
||||
default = false;
|
||||
type = lib.types.bool;
|
||||
description = lib.mdDoc ''
|
||||
Enables gokill daemon
|
||||
buildInputs = [
|
||||
pkgs.olm
|
||||
];
|
||||
|
||||
postInstall = ''
|
||||
'';
|
||||
};
|
||||
|
||||
extraConfig = lib.mkOption {
|
||||
type = lib.types.str;
|
||||
description = lib.mdDoc ''
|
||||
gokill config.json
|
||||
'';
|
||||
|
||||
docs = pkgs.callPackage (import ./docs/default.nix) { self = self; };
|
||||
|
||||
default = self.packages.${system}.gokill;
|
||||
|
||||
};
|
||||
|
||||
apps = {
|
||||
docs = {
|
||||
type = "app";
|
||||
program = builtins.toString (pkgs.writeScript "docs" ''
|
||||
${pkgs.python3}/bin/python3 -m http.server --directory ${self.packages."${system}".docs}/share/doc'');
|
||||
};
|
||||
};
|
||||
|
||||
config = lib.mkIf cfg.enable {
|
||||
systemd.services.gokill = {
|
||||
description = "gokill daemon";
|
||||
serviceConfig = {
|
||||
Type = "simple";
|
||||
ExecStart = "${gokill-pkg}/bin/gokill -c ${configFile}";
|
||||
Restart = "on-failure";
|
||||
};
|
||||
|
||||
wantedBy = [ "default.target" ];
|
||||
};
|
||||
};
|
||||
};
|
||||
})) ({
|
||||
nixosModules.gokill = import ./nixos-modules/gokill.nix { self = self; };
|
||||
|
||||
packages.x86_64-linux.testVm =
|
||||
let
|
||||
nixos = nixpkgs.lib.nixosSystem {
|
||||
system = "x86_64-linux";
|
||||
specialArgs = { inherit self; };
|
||||
modules = [
|
||||
self.nixosModules.gokill
|
||||
{
|
||||
services.gokill.enable = true;
|
||||
services.gokill.extraConfig = ''
|
||||
[
|
||||
services.gokill.testRun = false;
|
||||
services.gokill.triggers = [
|
||||
{
|
||||
"type": "Timeout",
|
||||
"name": "custom timeout",
|
||||
"options": {
|
||||
"duration": 30
|
||||
},
|
||||
"actions": [
|
||||
type = "Timeout";
|
||||
name = "custom timeout";
|
||||
options = {
|
||||
duration = 10;
|
||||
};
|
||||
actions = [
|
||||
{
|
||||
"type": "Print",
|
||||
"options": {
|
||||
"message": "Stage 1 triggered. Waiting 25 seconds"
|
||||
},
|
||||
"stage": 1
|
||||
},
|
||||
{
|
||||
"type": "Timeout",
|
||||
"options": {
|
||||
"duration": 20
|
||||
},
|
||||
"stage": 1
|
||||
},
|
||||
{
|
||||
"type": "Timeout",
|
||||
"options": {
|
||||
"duration": 5
|
||||
},
|
||||
"stage": 2
|
||||
},
|
||||
{
|
||||
"type": "Print",
|
||||
"options": {
|
||||
"message": "Shutdown in 5 seconds..."
|
||||
},
|
||||
"stage": 2
|
||||
},
|
||||
{
|
||||
"type": "Shutdown",
|
||||
"options": {
|
||||
},
|
||||
"stage": 3
|
||||
type = "Shutdown";
|
||||
options = {
|
||||
};
|
||||
stage = 2;
|
||||
}
|
||||
]
|
||||
];
|
||||
}
|
||||
]
|
||||
'';
|
||||
];
|
||||
users.users.root.password = "root";
|
||||
virtualisation.vmVariant.virtualisation.graphics = false;
|
||||
}
|
||||
@@ -131,5 +113,14 @@
|
||||
${self.packages."x86_64-linux".testVm}/bin/run-nixos-vm
|
||||
'');
|
||||
};
|
||||
|
||||
checks.x86_64-linux = let
|
||||
checkArgs = {
|
||||
pkgs = nixpkgs.legacyPackages."x86_64-linux";
|
||||
inherit self;
|
||||
};
|
||||
in {
|
||||
gokill = import ./test/test.nix checkArgs;
|
||||
};
|
||||
}) ;
|
||||
}
|
||||
|
||||
24
go.mod
24
go.mod
@@ -1,3 +1,25 @@
|
||||
module unknown.com/gokill
|
||||
|
||||
go 1.20
|
||||
go 1.21.3
|
||||
|
||||
require (
|
||||
github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1
|
||||
github.com/mattn/go-sqlite3 v1.14.17
|
||||
maunium.net/go/mautrix v0.16.1
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/mattn/go-colorable v0.1.12 // indirect
|
||||
github.com/mattn/go-isatty v0.0.14 // indirect
|
||||
github.com/rs/zerolog v1.30.0 // indirect
|
||||
github.com/tidwall/gjson v1.16.0 // indirect
|
||||
github.com/tidwall/match v1.1.1 // indirect
|
||||
github.com/tidwall/pretty v1.2.0 // indirect
|
||||
github.com/tidwall/sjson v1.2.5 // indirect
|
||||
go.mau.fi/util v0.1.0 // indirect
|
||||
golang.org/x/crypto v0.13.0 // indirect
|
||||
golang.org/x/exp v0.0.0-20230905200255-921286631fa9 // indirect
|
||||
golang.org/x/net v0.15.0 // indirect
|
||||
golang.org/x/sys v0.12.0 // indirect
|
||||
maunium.net/go/maulogger/v2 v2.4.1 // indirect
|
||||
)
|
||||
|
||||
49
go.sum
Normal file
49
go.sum
Normal file
@@ -0,0 +1,49 @@
|
||||
github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60=
|
||||
github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM=
|
||||
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1 h1:wG8n/XJQ07TmjbITcGiUaOtXxdrINDz1b0J1w0SzqDc=
|
||||
github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1/go.mod h1:A2S0CWkNylc2phvKXWBBdD3K0iGnDBGbzRpISP2zBl8=
|
||||
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
|
||||
github.com/mattn/go-colorable v0.1.12 h1:jF+Du6AlPIjs2BiUiQlKOX0rt3SujHxPnksPKZbaA40=
|
||||
github.com/mattn/go-colorable v0.1.12/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4=
|
||||
github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y=
|
||||
github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=
|
||||
github.com/mattn/go-sqlite3 v1.14.17 h1:mCRHCLDUBXgpKAqIKsaAaAsrAlbkeomtRFKXh2L6YIM=
|
||||
github.com/mattn/go-sqlite3 v1.14.17/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
|
||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg=
|
||||
github.com/rs/zerolog v1.30.0 h1:SymVODrcRsaRaSInD9yQtKbtWqwsfoPcRff/oRXLj4c=
|
||||
github.com/rs/zerolog v1.30.0/go.mod h1:/tk+P47gFdPXq4QYjvCmT5/Gsug2nagsFWBWhAiSi1w=
|
||||
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
|
||||
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||
github.com/tidwall/gjson v1.14.2/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
|
||||
github.com/tidwall/gjson v1.16.0 h1:SyXa+dsSPpUlcwEDuKuEBJEz5vzTvOea+9rjyYodQFg=
|
||||
github.com/tidwall/gjson v1.16.0/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
|
||||
github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
|
||||
github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
|
||||
github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs=
|
||||
github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
|
||||
github.com/tidwall/sjson v1.2.5 h1:kLy8mja+1c9jlljvWTlSazM7cKDRfJuR/bOJhcY5NcY=
|
||||
github.com/tidwall/sjson v1.2.5/go.mod h1:Fvgq9kS/6ociJEDnK0Fk1cpYF4FIW6ZF7LAe+6jwd28=
|
||||
go.mau.fi/util v0.1.0 h1:BwIFWIOEeO7lsiI2eWKFkWTfc5yQmoe+0FYyOFVyaoE=
|
||||
go.mau.fi/util v0.1.0/go.mod h1:AxuJUMCxpzgJ5eV9JbPWKRH8aAJJidxetNdUj7qcb84=
|
||||
golang.org/x/crypto v0.13.0 h1:mvySKfSWJ+UKUii46M40LOvyWfN0s2U+46/jDd0e6Ck=
|
||||
golang.org/x/crypto v0.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliYc=
|
||||
golang.org/x/exp v0.0.0-20230905200255-921286631fa9 h1:GoHiUyI/Tp2nVkLI2mCxVkOjsbSXD66ic0XW0js0R9g=
|
||||
golang.org/x/exp v0.0.0-20230905200255-921286631fa9/go.mod h1:S2oDrQGGwySpoQPVqRShND87VCbxmc6bL1Yd2oYrm6k=
|
||||
golang.org/x/net v0.15.0 h1:ugBLEUaxABaB5AJqW9enI0ACdci2RUd4eP51NTBvuJ8=
|
||||
golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk=
|
||||
golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.12.0 h1:CM0HF96J0hcLAwsHPJZjfdNzs0gftsLfgKt57wWHJ0o=
|
||||
golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
maunium.net/go/maulogger/v2 v2.4.1 h1:N7zSdd0mZkB2m2JtFUsiGTQQAdP0YeFWT7YMc80yAL8=
|
||||
maunium.net/go/maulogger/v2 v2.4.1/go.mod h1:omPuYwYBILeVQobz8uO3XC8DIRuEb5rXYlQSuqrbCho=
|
||||
maunium.net/go/mautrix v0.16.1 h1:Wb3CvOCe8A/NLsFeZYxKrgXKiqeZUQEBD1zqm7n/kWk=
|
||||
maunium.net/go/mautrix v0.16.1/go.mod h1:2Jf15tulVtr6LxoiRL4smRXwpkGWUNfBFhwh/aXDBuk=
|
||||
@@ -36,5 +36,6 @@ type ConfigOption struct {
|
||||
type Documenter interface {
|
||||
GetName() string
|
||||
GetDescription() string
|
||||
GetExample() string
|
||||
GetOptions() []ConfigOption
|
||||
}
|
||||
|
||||
82
nixos-modules/gokill.nix
Normal file
82
nixos-modules/gokill.nix
Normal file
@@ -0,0 +1,82 @@
|
||||
flake: { config, lib, pkgs, self, ... }:
|
||||
let
|
||||
cfg = config.services.gokill;
|
||||
configFile = pkgs.writeText "config.json" (builtins.toJSON cfg.triggers);
|
||||
gokill-pkg = self.packages.x86_64-linux.gokill;
|
||||
testRun = if cfg.testRun then "-t" else "";
|
||||
in
|
||||
{
|
||||
options = with lib; {
|
||||
services.gokill = {
|
||||
enable = mkOption {
|
||||
default = false;
|
||||
type = types.bool;
|
||||
description = mdDoc ''
|
||||
Enables gokill daemon
|
||||
'';
|
||||
};
|
||||
|
||||
testRun = mkOption {
|
||||
default = false;
|
||||
type = types.bool;
|
||||
description = mdDoc ''
|
||||
When set to true gokill is performing a test run
|
||||
'';
|
||||
};
|
||||
|
||||
triggers = mkOption {
|
||||
description = "list of triggers";
|
||||
default = [];
|
||||
type = with types; types.listOf ( submodule {
|
||||
options = {
|
||||
type = mkOption {
|
||||
type = types.str;
|
||||
};
|
||||
|
||||
name = mkOption {
|
||||
type = types.str;
|
||||
};
|
||||
|
||||
options = mkOption {
|
||||
type = types.attrs;
|
||||
};
|
||||
|
||||
actions = mkOption {
|
||||
description = "list of actions";
|
||||
type = with types; types.listOf ( submodule {
|
||||
options = {
|
||||
type = mkOption {
|
||||
type = types.str;
|
||||
};
|
||||
|
||||
options = mkOption {
|
||||
type = types.attrs;
|
||||
};
|
||||
|
||||
stage = mkOption {
|
||||
type = types.int;
|
||||
};
|
||||
};
|
||||
});
|
||||
};
|
||||
};
|
||||
});
|
||||
};
|
||||
};
|
||||
};
|
||||
|
||||
config = lib.mkIf cfg.enable {
|
||||
systemd.services.gokill = {
|
||||
description = "gokill daemon";
|
||||
serviceConfig = {
|
||||
Type = "simple";
|
||||
ExecStart = "${gokill-pkg}/bin/gokill -c ${configFile} ${testRun}";
|
||||
Restart = "on-failure";
|
||||
};
|
||||
|
||||
wantedBy = [ "default.target" ];
|
||||
};
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
21
test/lib.nix
Normal file
21
test/lib.nix
Normal file
@@ -0,0 +1,21 @@
|
||||
# tests/lib.nix
|
||||
# based on https://blog.thalheim.io/2023/01/08/how-to-use-nixos-testing-framework-with-flakes/
|
||||
# The first argument to this function is the test module itself
|
||||
test:
|
||||
# These arguments are provided by `flake.nix` on import, see checkArgs
|
||||
{ pkgs, self}:
|
||||
let
|
||||
inherit (pkgs) lib;
|
||||
# this imports the nixos library that contains our testing framework
|
||||
nixos-lib = import (pkgs.path + "/nixos/lib") {};
|
||||
in
|
||||
(nixos-lib.runTest {
|
||||
hostPkgs = pkgs;
|
||||
# This speeds up the evaluation by skipping evaluating documentation (optional)
|
||||
defaults.documentation.enable = lib.mkDefault false;
|
||||
# This makes `self` available in the NixOS configuration of our virtual machines.
|
||||
# This is useful for referencing modules or packages from your own flake
|
||||
# as well as importing from other flakes.
|
||||
node.specialArgs = { inherit self; };
|
||||
imports = [ test ];
|
||||
}).config.result
|
||||
40
test/test.nix
Normal file
40
test/test.nix
Normal file
@@ -0,0 +1,40 @@
|
||||
(import ./lib.nix) {
|
||||
name = "gokill-base-test";
|
||||
nodes = {
|
||||
node1 = { self, pkgs, ... }: {
|
||||
imports = [ self.nixosModules.gokill ];
|
||||
|
||||
services.gokill = {
|
||||
enable = true;
|
||||
triggers = [
|
||||
{
|
||||
type = "Timeout";
|
||||
name = "custom timeout";
|
||||
options = {
|
||||
duration = 3;
|
||||
};
|
||||
actions = [
|
||||
{
|
||||
type = "Command";
|
||||
options = {
|
||||
command = "echo hello world";
|
||||
};
|
||||
stage = 2;
|
||||
}
|
||||
];
|
||||
}
|
||||
];
|
||||
};
|
||||
};
|
||||
};
|
||||
|
||||
testScript = ''
|
||||
import time
|
||||
start_all() # wait for our service to start
|
||||
node1.wait_for_unit("gokill")
|
||||
time.sleep(4)
|
||||
output = node1.succeed("journalctl -u gokill.service | tail -n 2 | head -n 1")
|
||||
# Check if our webserver returns the expected result
|
||||
assert "hello world" in output
|
||||
'';
|
||||
}
|
||||
@@ -3,7 +3,7 @@ package triggers
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"unknown.com/gokill/actions"
|
||||
@@ -17,7 +17,7 @@ type EthernetDisconnect struct {
|
||||
}
|
||||
|
||||
func isEthernetConnected(deviceName string) bool {
|
||||
content, err := ioutil.ReadFile(fmt.Sprintf("/sys/class/net/%s/operstate", deviceName))
|
||||
content, err := os.ReadFile(fmt.Sprintf("/sys/class/net/%s/operstate", deviceName))
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
@@ -87,6 +87,21 @@ func (p EthernetDisconnect) GetDescription() string {
|
||||
return "Triggers if Ethernetcable is disconnected."
|
||||
}
|
||||
|
||||
func (p EthernetDisconnect) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "EthernetDisconnect",
|
||||
"name": "Example Trigger",
|
||||
"options": {
|
||||
"interfaceName": "eth0",
|
||||
"waitTillConnected": true
|
||||
},
|
||||
"actions": [
|
||||
]
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p EthernetDisconnect) GetOptions() []internal.ConfigOption {
|
||||
return []internal.ConfigOption{
|
||||
{"waitTillConnected", "bool", "Only trigger when device was connected before", "true"},
|
||||
|
||||
@@ -1,5 +1,4 @@
|
||||
package triggers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
@@ -45,7 +44,21 @@ func (p TimeOut) GetName() string {
|
||||
}
|
||||
|
||||
func (p TimeOut) GetDescription() string {
|
||||
return "Triggers after given duration."
|
||||
return "Triggers after given duration. Mostly used for debugging."
|
||||
}
|
||||
|
||||
func (p TimeOut) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "Timeout",
|
||||
"name": "Example Trigger",
|
||||
"options": {
|
||||
"duration": 5
|
||||
},
|
||||
"actions": [
|
||||
]
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
func (p TimeOut) GetOptions() []internal.ConfigOption {
|
||||
|
||||
@@ -28,8 +28,8 @@ func NewTrigger(config internal.KillSwitchConfig) (Trigger, error) {
|
||||
|
||||
func GetAllTriggers() []DocumentedTrigger {
|
||||
return []DocumentedTrigger{
|
||||
TimeOut{},
|
||||
EthernetDisconnect{},
|
||||
TimeOut{},
|
||||
UsbDisconnect{},
|
||||
}
|
||||
}
|
||||
|
||||
@@ -85,9 +85,28 @@ func (p UsbDisconnect) GetName() string {
|
||||
}
|
||||
|
||||
func (p UsbDisconnect) GetDescription() string {
|
||||
return "Triggers when given usb drive is disconnected"
|
||||
return `
|
||||
Triggers when given usb drive is disconnected.
|
||||
Currently it simply checks that the file /dev/disk/by-id/$deviceId exists.
|
||||
`
|
||||
}
|
||||
|
||||
func (p UsbDisconnect) GetExample() string {
|
||||
return `
|
||||
{
|
||||
"type": "UsbDisconnect",
|
||||
"name": "Example Trigger",
|
||||
"options": {
|
||||
"deviceId": "ata-Samsung_SSD_860_EVO_1TB_S4AALKWJDI102",
|
||||
"waitTillConnected": true
|
||||
},
|
||||
"actions": [
|
||||
]
|
||||
}
|
||||
`
|
||||
}
|
||||
|
||||
|
||||
func (p UsbDisconnect) GetOptions() []internal.ConfigOption {
|
||||
return []internal.ConfigOption{
|
||||
{"waitTillConnected", "bool", "Only trigger when device was connected before", "true"},
|
||||
|
||||
Reference in New Issue
Block a user