This seems to work now.
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
Colin 2024-06-12 13:53:24 -04:00
parent a6b6f5cbf6
commit 789f94cbaf
10 changed files with 90 additions and 93 deletions

View File

@ -1 +1,2 @@
A /newfile.txt
A /newfile.txt
A /detectme.txt

View File

@ -1,5 +1 @@
D /file1.txt
A /file2.txt
C /do/match/file3.txt
A /dontmatch.yml
D /dont/match.md
A /detectme.yml

BIN
dist/darwin_amd64_main vendored

Binary file not shown.

BIN
dist/darwin_arm64_main vendored

Binary file not shown.

BIN
dist/linux_amd64_main vendored

Binary file not shown.

Binary file not shown.

BIN
dist/linux_arm64_main vendored

Binary file not shown.

Binary file not shown.

Binary file not shown.

174
main.go
View File

@ -1,7 +1,6 @@
package main
import (
"bufio"
"crypto/md5"
"encoding/json"
"fmt"
@ -24,60 +23,100 @@ type ContainerInfo struct {
Ignores []string
}
var notifiedChanges = struct {
sync.RWMutex
changes map[string]string
}{changes: make(map[string]string)}
type MonitoredContainer struct {
Info ContainerInfo
LastChecked time.Time
}
var apiAddress string
var (
apiAddress string
glitchtipDSN string
notifiedChanges = make(map[string]string)
monitoredContainers = make(map[string]*MonitoredContainer)
mu sync.Mutex
)
func init() {
apiAddress = os.Getenv("API_ADDRESS")
if apiAddress == "" {
apiAddress = "http://localhost:8080"
}
}
func main() {
// Check if GLITCHTIP_DSN environment variable is set
glitchtipDSN := os.Getenv("GLITCHTIP_DSN")
glitchtipDSN = os.Getenv("GLITCHTIP_DSN")
if glitchtipDSN == "" {
log.Fatal("GLITCHTIP_DSN environment variable is not set")
}
}
func main() {
log.Println("Starting Oculus...")
// Ensure the diffs directory exists
err := ensureDiffsDirectory()
err := os.MkdirAll("./diffs", os.ModePerm)
if err != nil {
log.Fatalf("Error creating diffs directory: %v", err)
}
for {
containers, err := fetchContainers()
err := fetchAndMonitorContainers()
if err != nil {
log.Printf("Error fetching containers: %v", err)
time.Sleep(1 * time.Minute)
continue
log.Printf("Error in fetching and monitoring containers: %v", err)
}
var wg sync.WaitGroup
for _, container := range containers {
wg.Add(1)
go func(container ContainerInfo) {
defer wg.Done()
checkAndNotify(container)
}(container)
}
wg.Wait()
cleanupHashes(containers)
// Sleep for 1 minute before checking for new containers again
time.Sleep(1 * time.Minute)
time.Sleep(1 * time.Second)
}
}
func fetchAndMonitorContainers() error {
containers, err := fetchContainers()
if err != nil {
return err
}
for _, container := range containers {
mu.Lock()
if _, exists := monitoredContainers[container.ID]; !exists {
interval, err := time.ParseDuration(container.Interval)
if err != nil {
log.Printf("Invalid interval for container %s (%s): %v", container.Name, container.ID, err)
mu.Unlock()
continue
}
monitoredContainers[container.ID] = &MonitoredContainer{
Info: container,
LastChecked: time.Now().Add(-interval),
}
}
mu.Unlock()
}
var wg sync.WaitGroup
for _, monitoredContainer := range monitoredContainers {
wg.Add(1)
go func(mc *MonitoredContainer) {
defer wg.Done()
interval, err := time.ParseDuration(mc.Info.Interval)
if err != nil {
log.Printf("Invalid interval for container %s (%s): %v", mc.Info.Name, mc.Info.ID, err)
return
}
mu.Lock()
if time.Since(mc.LastChecked) >= interval {
mc.LastChecked = time.Now()
mu.Unlock()
checkAndNotify(mc.Info)
} else {
mu.Unlock()
}
}(monitoredContainer)
}
wg.Wait()
return nil
}
func fetchContainers() ([]ContainerInfo, error) {
resp, err := http.Get(fmt.Sprintf("%s/containers", apiAddress))
if err != nil {
@ -134,9 +173,9 @@ func checkAndNotify(container ContainerInfo) {
diffHash := fmt.Sprintf("%x", md5.Sum([]byte(filteredOutput)))
log.Printf("Diff hash for container %s: %s", container.Name, diffHash)
notifiedChanges.RLock()
lastNotifiedHash, notified := notifiedChanges.changes[container.ID]
notifiedChanges.RUnlock()
mu.Lock()
lastNotifiedHash, notified := notifiedChanges[container.ID]
mu.Unlock()
if !notified || lastNotifiedHash != diffHash {
filename := filepath.Join("diffs", fmt.Sprintf("%s.diff", container.CName))
@ -150,9 +189,9 @@ func checkAndNotify(container ContainerInfo) {
if err != nil {
log.Printf("Error sending notification for container %s: %v", container.ID, err)
} else {
notifiedChanges.Lock()
notifiedChanges.changes[container.ID] = diffHash
notifiedChanges.Unlock()
mu.Lock()
notifiedChanges[container.ID] = diffHash
mu.Unlock()
log.Printf("Notification sent and hash updated for container: %s (%s)", container.Name, container.ID)
}
} else {
@ -172,44 +211,25 @@ func filterDiffOutput(diffOutput, cname string, ignores []string) (string, error
return "", fmt.Errorf("error writing diff to file: %s", err)
}
// Read and filter the diff output
var filteredLines []string
file, err := os.Open(filename)
// Construct the filter command
args := append([]string{filename}, ignores...)
cmd := exec.Command("filter", args...)
fullCommand := fmt.Sprintf("filter %s", strings.Join(cmd.Args, " "))
log.Printf("Running command: %s", fullCommand)
fmt.Printf("Running command: %s\n", fullCommand) // Print the command to stdout for debugging
output, err := cmd.CombinedOutput()
if err != nil {
return "", fmt.Errorf("error opening diff file: %s", err)
}
defer file.Close()
scanner := bufio.NewScanner(file)
for scanner.Scan() {
line := scanner.Text()
ignore := false
for _, pattern := range ignores {
match, err := filepath.Match(pattern, line)
if err != nil {
return "", fmt.Errorf("error matching pattern: %s", err)
}
if match {
ignore = true
break
}
}
if !ignore {
filteredLines = append(filteredLines, line)
}
return "", fmt.Errorf("error running filter command: %s, output: %s", err, output)
}
if err := scanner.Err(); err != nil {
return "", fmt.Errorf("error reading diff file: %s", err)
}
filteredOutput := strings.Join(filteredLines, "\n")
err = ioutil.WriteFile(filename, []byte(filteredOutput), 0644)
// Read the filtered output
filteredOutput, err := ioutil.ReadFile(filename)
if err != nil {
return "", fmt.Errorf("error writing filtered diff to file: %s", err)
return "", fmt.Errorf("error reading filtered diff file: %s", err)
}
return filteredOutput, nil
return string(filteredOutput), nil
}
func writeToFile(filename, content string) error {
@ -233,23 +253,3 @@ func sendNotification(content string) error {
}
return nil
}
func ensureDiffsDirectory() error {
return os.MkdirAll("./diffs", os.ModePerm)
}
func cleanupHashes(containers []ContainerInfo) {
notifiedChanges.Lock()
defer notifiedChanges.Unlock()
activeContainers := make(map[string]bool)
for _, container := range containers {
activeContainers[container.ID] = true
}
for id := range notifiedChanges.changes {
if !activeContainers[id] {
delete(notifiedChanges.changes, id)
}
}
}