108 lines
2.4 KiB
Nim
108 lines
2.4 KiB
Nim
import os
|
|
import osproc
|
|
import strutils
|
|
import sequtils
|
|
import random
|
|
|
|
import ../common
|
|
import ../parser
|
|
import ../output
|
|
|
|
var UNSPLASH_KEY = ""
|
|
var BG_DIR = "/tmp/"
|
|
var LAST_FILE = ""
|
|
var LAST = ""
|
|
var GET_FROM_UNSPLASH = false
|
|
|
|
const UNSPLASH_URL = "https://api.unsplash.com/photos/random?query=$QUERY&orientation=landscape"
|
|
|
|
type
|
|
Note* = object
|
|
urgency*: Urgency
|
|
title*: string
|
|
content*: string
|
|
timeout*: int
|
|
Urgency* = enum
|
|
Normal = "normal"
|
|
Low = "low"
|
|
Urgent = "urgent"
|
|
Critical = "critical"
|
|
|
|
proc newNote*(): Note =
|
|
return Note(urgency: Normal, title: "Notification", content: "Hello, I am a notifications", timeout: 2000)
|
|
|
|
proc `$`*(n: Note): string =
|
|
let str = "notify-send -u $U $T $C -t $N"
|
|
.replace("$U", $n.urgency)
|
|
.replace("$T", n.title.escape)
|
|
.replace("$C", n.content.escape)
|
|
.replace("$N", $n.timeout)
|
|
return str
|
|
|
|
proc send(n: Note) =
|
|
discard execCmdEx($n)
|
|
|
|
proc getFromUnsplash(q: var string): string =
|
|
createDir(BG_DIR & "/unsplash")
|
|
echo "Getting from Unsplash"
|
|
q = q.replace(" ","%20")
|
|
let uri = UNSPLASH_URL.replace("$QUERY",q)
|
|
|
|
proc getFiles(dir: string): seq[string] =
|
|
var files: seq[string] = @[]
|
|
for file in walkDir(dir):
|
|
if file.path.endsWith(".jpg"): files.add(file.path)
|
|
elif file.kind == pcDir:
|
|
files = files.concat(getFiles(file.path))
|
|
return files
|
|
|
|
proc getLast() =
|
|
LAST = readFile(LAST_FILE).strip()
|
|
|
|
proc getImageFromDir(): string =
|
|
echo "Getting Random file from " & BG_DIR
|
|
var img_files = getFiles(BG_DIR).filter(proc(f: string): bool = f != LAST)
|
|
img_files.shuffle()
|
|
let img_file = img_files[0]
|
|
|
|
echo "Found : ", img_file
|
|
|
|
proc setLast() =
|
|
var n: Note = newNote()
|
|
n.title = "Setting Background to Last"
|
|
n.content = LAST
|
|
n.send()
|
|
let feh = "feh --bg-fill " & LAST.escape
|
|
echo feh
|
|
discard execCmdEx(feh)
|
|
|
|
proc getDesign(): Info =
|
|
var data = newInfo("Wallpapurr")
|
|
return data
|
|
|
|
proc queryPrompt(): string =
|
|
let data = getDesign()
|
|
let output = data.outputData()
|
|
return output
|
|
|
|
proc go*() =
|
|
var args = parseWallpapurrArgs()
|
|
UNSPLASH_KEY = myConfig.unsplash_key
|
|
BG_DIR = myConfig.bg_dir
|
|
LAST_FILE = BG_DIR & "/last.txt"
|
|
getLast()
|
|
var img = ""
|
|
if args.query != "" or args.from_unsplash:
|
|
if args.query == "":
|
|
args.query = queryPrompt()
|
|
echo "Query: ", args.query
|
|
img = getFromUnsplash(args.query)
|
|
elif args.last:
|
|
setLast()
|
|
else:
|
|
img = getImageFromDir()
|
|
echo img
|
|
|
|
|
|
|
|
|