rudimentary slideshow functionality

This commit is contained in:
andrzej 2024-10-29 16:18:21 +01:00
parent ab5b679896
commit 80a49ca59a
7 changed files with 57 additions and 35 deletions

View File

@ -9,6 +9,7 @@ import (
type Config struct { type Config struct {
ImageFilters ImageFilters
Root string Root string
Duration int
} }
func (config *Config) load() { func (config *Config) load() {

View File

@ -3,5 +3,6 @@
"colorize":[ 247,40,60 ], "colorize":[ 247,40,60 ],
"contrast":-35, "contrast":-35,
"gamma":0.8, "gamma":0.8,
"root":"/home/andrzej/bgs" "root":"/home/andrzej/bgs",
"duration":10
} }

View File

@ -1,6 +1,7 @@
package main package main
import ( import (
"fmt"
"math/rand" "math/rand"
"os" "os"
"regexp" "regexp"
@ -12,25 +13,24 @@ func getRandomFile(dir string) (path string, filename string, error error) {
return "", "", err return "", "", err
} }
// for _, file := range files { fmt.Println(file.Name(), file.IsDir())
// }
var randomIndex int var randomIndex int
var randomImg os.DirEntry var randomImg os.DirEntry
for { for {
randomIndex = rand.Intn(len(files)) randomIndex = rand.Intn(len(files))
randomImg = files[randomIndex] randomImg = files[randomIndex]
//re-roll if you get a directory //re-roll if you get a directory
if files[randomIndex].IsDir() == false { if files[randomIndex].IsDir() {
break fmt.Printf("%v is a directory\n", randomImg.Name())
continue
} }
//re-roll if not a recognised image type //re-roll if not a recognised image type
isImage, _ := regexp.Match(".jpg$|.jpeg$|.png$|.bmp$|.webp$", []byte(randomImg.Name())) isImage, _ := regexp.Match(".jpg$|.jpeg$|.png$|.bmp$|.webp$", []byte(randomImg.Name()))
if isImage { if !isImage {
break fmt.Printf("%v is not a recognised image format\n", randomImg.Name())
} continue
} }
return dir + "/" + randomImg.Name(), randomImg.Name(), nil return dir + "/" + randomImg.Name(), randomImg.Name(), nil
}
} }
func fileExistsInCache(filename string) (bool, error) { func fileExistsInCache(filename string) (bool, error) {

View File

@ -1,6 +1,7 @@
package main package main
import ( import (
"fmt"
"image" "image"
"log" "log"
"os" "os"
@ -38,6 +39,7 @@ func pickRandomImage(dir string) (string, error) {
curr := wd + "/images/" + filenameNoExt curr := wd + "/images/" + filenameNoExt
if !fileExists { if !fileExists {
img, err := loadImage(randomImg) img, err := loadImage(randomImg)
fmt.Printf("failed to load image %v", randomImg)
if err != nil { if err != nil {
return "", err return "", err
} }
@ -52,6 +54,7 @@ func pickRandomImage(dir string) (string, error) {
func loadImage(filename string) (image.Image, error) { func loadImage(filename string) (image.Image, error) {
f, err := os.Open(filename) f, err := os.Open(filename)
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer f.Close() defer f.Close()

25
main.go
View File

@ -2,7 +2,7 @@ package main
import ( import (
"fmt" "fmt"
"log" // "log"
"os" "os"
"sync" "sync"
) )
@ -10,25 +10,24 @@ import (
var config Config var config Config
func main() { func main() {
var err error
config.load() config.load()
fmt.Printf("%+v\n", config) fmt.Printf("%+v\n", config)
args := []string{""} //TODO: make this logic interface more neatly with the slideshow channel
if len(os.Args) > 0 { var dir string
args = os.Args[1:] args := os.Args[1:]
if len(args) > 0 {
dir = args[0]
} else {
dir = ""
} }
curr, err := pickRandomImage(args[0])
if err != nil {
log.Fatal("failed to pick image!", err)
}
hyprpaperSet(curr)
var waitGroup sync.WaitGroup var waitGroup sync.WaitGroup
waitGroup.Add(1) waitGroup.Add(1)
go server(&waitGroup) slideshowDir := make(chan string)
go server(slideshowDir)
go slideshow(slideshowDir)
slideshowDir <- dir
waitGroup.Wait() waitGroup.Wait()
} }

View File

@ -6,11 +6,10 @@ import (
"os" "os"
"os/signal" "os/signal"
"strings" "strings"
"sync"
"syscall" "syscall"
) )
func server(waitGroup *sync.WaitGroup) { func server(slideshowDir chan string) {
sockfile := "/tmp/bg-go.sock" sockfile := "/tmp/bg-go.sock"
//Create a Unix domain socket and listen for incoming connections. //Create a Unix domain socket and listen for incoming connections.
@ -18,6 +17,7 @@ func server(waitGroup *sync.WaitGroup) {
if err != nil { if err != nil {
panic(err) panic(err)
} }
log.Println("listening on unix domain socket...")
//cleanup sockfile //cleanup sockfile
//TODO: this only works when the program is killed from outside. Make it work via defer() too? //TODO: this only works when the program is killed from outside. Make it work via defer() too?
@ -51,17 +51,13 @@ func server(waitGroup *sync.WaitGroup) {
} }
str := strings.TrimSpace(string(buf[:n])) str := strings.TrimSpace(string(buf[:n]))
newbg, err := pickRandomImage(str) slideshowDir <- str
if err != nil { // newbg, err := pickRandomImage(str)
panic(err)
}
hyprpaperSet(newbg)
// // Echo the data back to the connection
// _, err = conn.Write(buf[:n])
// if err != nil { // if err != nil {
// log.Fatal(err) // panic(err)
// } // }
// hyprpaperSet(newbg)
}(conn) }(conn)
} }

View File

@ -1,5 +1,27 @@
package main package main
func slideshow(dir string) { import (
"log"
"time"
)
func slideshow(ch <-chan string) {
dir := <-ch
ticker := time.NewTicker(time.Duration(config.Duration) * time.Minute)
go func() {
for {
img, err := pickRandomImage(dir)
if err != nil {
panic(err)
}
hyprpaperSet(img)
select {
case dir = <-ch:
log.Println("directory set!")
continue
case <-ticker.C:
continue
}
}
}()
} }