Skip to content

Commit

Permalink
move log methods to own file
Browse files Browse the repository at this point in the history
  • Loading branch information
sgsullivan committed Aug 4, 2022
1 parent 48518f6 commit b466c4c
Show file tree
Hide file tree
Showing 2 changed files with 37 additions and 29 deletions.
29 changes: 0 additions & 29 deletions befehl.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ import (
"bytes"
"fmt"
"log"
"os"
"sync"
"time"

Expand Down Expand Up @@ -130,31 +129,3 @@ func (instance *Instance) runPayload(wg *sync.WaitGroup, host string, payload []
panic(err)
}
}

func (instance *Instance) prepareLogDir() error {
logDir := instance.getLogDir()
if !filesystem.PathExists(logDir) {
if err := os.MkdirAll(logDir, os.FileMode(0700)); err != nil {
return fmt.Errorf("failed creating [%s]: %s", logDir, err)
}
}
return nil
}

func (instance *Instance) logPayloadRun(host string, output string) error {
instance.prepareLogDir()
logFilePath := instance.getLogFilePath(host)

logFile, err := os.Create(logFilePath)
if err != nil {
return fmt.Errorf("error creating [%s]: %s", logFilePath, err)
}
defer logFile.Close()

if _, err = logFile.WriteString(output); err != nil {
return fmt.Errorf("error writing to [%s]: %s", logFilePath, err)
}

log.Printf("payload completed on %s! logfile at: %s\n", host, logFilePath)
return nil
}
37 changes: 37 additions & 0 deletions log.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
package befehl

import (
"fmt"
"log"
"os"

"github.com/sgsullivan/befehl/helpers/filesystem"
)

func (instance *Instance) prepareLogDir() error {
logDir := instance.getLogDir()
if !filesystem.PathExists(logDir) {
if err := os.MkdirAll(logDir, os.FileMode(0700)); err != nil {
return fmt.Errorf("failed creating [%s]: %s", logDir, err)
}
}
return nil
}

func (instance *Instance) logPayloadRun(host string, output string) error {
instance.prepareLogDir()
logFilePath := instance.getLogFilePath(host)

logFile, err := os.Create(logFilePath)
if err != nil {
return fmt.Errorf("error creating [%s]: %s", logFilePath, err)
}
defer logFile.Close()

if _, err = logFile.WriteString(output); err != nil {
return fmt.Errorf("error writing to [%s]: %s", logFilePath, err)
}

log.Printf("payload completed on %s! logfile at: %s\n", host, logFilePath)
return nil
}

0 comments on commit b466c4c

Please sign in to comment.