Minor Formatting Changes and better error handling

This commit is contained in:
aditya-K2 2022-03-16 02:48:53 +05:30
parent 719a659df6
commit b7f7c9cebd
4 changed files with 37 additions and 34 deletions

View File

@ -39,8 +39,11 @@ func downloadImage(url string, imagePath string) (string, error) {
if err == nil { if err == nil {
b, err := os.Create(imagePath) b, err := os.Create(imagePath)
if err == nil { if err == nil {
b.Write(v) if _, err := b.Write(v); err == nil {
return imagePath, nil return imagePath, nil
} else {
return "", errors.New("could Not Write Image")
}
} else { } else {
b.Close() b.Close()
return "", err return "", err
@ -49,5 +52,5 @@ func downloadImage(url string, imagePath string) (string, error) {
return "", err return "", err
} }
} }
return "", errors.New("Image Not Received") return "", errors.New("image Not Received")
} }

View File

@ -27,14 +27,14 @@ func SetNotificationServer(n interface{ Send(string) }) {
Notify = n Notify = n
} }
// Renderer is just a channel on which we will send the Path to the song whose // Renderer is just a channel on which we will send the Path to the song whose
// Image is to be Rendered. This channel is passed to the OpenImage which in turn is called // Image is to be Rendered. This channel is passed to the OpenImage which in turn is called
// by the Start() function as a go routine. // by the Start() function as a go routine.
type Renderer struct { type Renderer struct {
c chan string c chan string
} }
// Returns a new Renderer with a string channel // NewRenderer Returns a new Renderer with a string channel
func NewRenderer() *Renderer { func NewRenderer() *Renderer {
c := make(chan string) c := make(chan string)
return &Renderer{ return &Renderer{
@ -43,13 +43,13 @@ func NewRenderer() *Renderer {
} }
// Send Image Path to Renderer // Send Image Path to Renderer
func (self *Renderer) Send(path string) { func (r *Renderer) Send(path string) {
self.c <- path r.c <- path
} }
// Go Routine that will Be Called and will listen on the channel c // OpenImage Go Routine that will Be Called and will listen on the channel c
// for changes and on getting a string over the channel will open the Image and // for changes and on getting a string over the channel will open the Image and
// keep listening again. This will keep the image blocked ( i.e no need to use time.Sleep() etc. ) // keep listening again. This will keep the image blocked ( i.e. no need to use time.Sleep() etc. )
// and saves resources too. // and saves resources too.
func OpenImage(path string, c chan string) { func OpenImage(path string, c chan string) {
fw, fh := utils.GetFontWidth() fw, fh := utils.GetFontWidth()
@ -57,7 +57,9 @@ func OpenImage(path string, c chan string) {
if path != "stop" { if path != "stop" {
extractedImage := GetImagePath(path) extractedImage := GetImagePath(path)
img2, _ := GetImg(extractedImage) img2, _ := GetImg(extractedImage)
im, _ = ueberzug.NewImage(img2, int(float32(ui.IMG_X)*fw)+viper.GetInt("ADDITIONAL_PADDING_X"), int(float32(ui.IMG_Y)*fh)+viper.GetInt("ADDITIONAL_PADDING_Y")) im, _ = ueberzug.NewImage(img2,
int(float32(ui.ImgX)*fw)+viper.GetInt("ADDITIONAL_PADDING_X"),
int(float32(ui.ImgY)*fh)+viper.GetInt("ADDITIONAL_PADDING_Y"))
} }
d := <-c d := <-c
if im != nil { if im != nil {
@ -70,13 +72,14 @@ func OpenImage(path string, c chan string) {
} }
} }
// Initialises the Renderer and calls the go routine OpenImage and passes the channel // Start Initialises the Renderer and calls the go routine OpenImage and passes the channel
// as argument. // as argument.
func (self *Renderer) Start(path string) { func (r *Renderer) Start(path string) {
go OpenImage(path, self.c) go OpenImage(path, r.c)
} }
// This Function returns the path to the image that is to be rendered it checks first for the image in the cache // GetImagePath This Function returns the path to the image that is to be
// rendered it checks first for the image in the cache
// else it adds the image to the cache and then extracts it and renders it. // else it adds the image to the cache and then extracts it and renders it.
func GetImagePath(path string) string { func GetImagePath(path string) string {
a, err := CONN.ListInfo(path) a, err := CONN.ListInfo(path)
@ -88,7 +91,8 @@ func GetImagePath(path string) string {
imagePath := cache.GenerateName(a[0]["artist"], a[0]["album"]) imagePath := cache.GenerateName(a[0]["artist"], a[0]["album"])
absPath := utils.CheckDirectoryFmt(viper.GetString("MUSIC_DIRECTORY")) + path absPath := utils.CheckDirectoryFmt(viper.GetString("MUSIC_DIRECTORY")) + path
extractedImage = ExtractImageFromFile(absPath, imagePath) extractedImage = ExtractImageFromFile(absPath, imagePath)
if extractedImage == viper.GetString("DEFAULT_IMAGE_PATH") && viper.GetString("GET_COVER_ART_FROM_LAST_FM") == "TRUE" { if extractedImage == viper.GetString("DEFAULT_IMAGE_PATH") &&
viper.GetString("GET_COVER_ART_FROM_LAST_FM") == "TRUE" {
downloadedImage, err := getImageFromLastFM(a[0]["artist"], a[0]["album"], imagePath) downloadedImage, err := getImageFromLastFM(a[0]["artist"], a[0]["album"], imagePath)
if err == nil { if err == nil {
Notify.Send("Image From LastFM") Notify.Send("Image From LastFM")
@ -116,7 +120,8 @@ func GetImg(uri string) (image.Image, error) {
} }
fw, fh := utils.GetFontWidth() fw, fh := utils.GetFontWidth()
img = resize.Resize( img = resize.Resize(
uint(float32(ui.IMG_W)*(fw+float32(viper.GetFloat64("IMAGE_WIDTH_EXTRA_X")))), uint(float32(ui.IMG_H)*(fh+float32(viper.GetFloat64("IMAGE_WIDTH_EXTRA_Y")))), uint(float32(ui.ImgW)*(fw+float32(viper.GetFloat64("IMAGE_WIDTH_EXTRA_X")))),
uint(float32(ui.ImgH)*(fh+float32(viper.GetFloat64("IMAGE_WIDTH_EXTRA_Y")))),
img, img,
resize.Bilinear, resize.Bilinear,
) )

View File

@ -5,7 +5,12 @@ import (
"github.com/gdamore/tcell/v2" "github.com/gdamore/tcell/v2"
) )
var IMG_X, IMG_Y, IMG_W, IMG_H int var (
ImgY int
ImgW int
ImgH int
ImgX int
)
type Application struct { type Application struct {
App *tview.Application App *tview.Application
@ -26,7 +31,7 @@ func NewApplication() *Application {
imagePreviewer := tview.NewBox() imagePreviewer := tview.NewBox()
imagePreviewer.SetBorder(true) imagePreviewer.SetBorder(true)
imagePreviewer.SetDrawFunc(func(s tcell.Screen, x, y, width, height int) (int, int, int, int) { imagePreviewer.SetDrawFunc(func(s tcell.Screen, x, y, width, height int) (int, int, int, int) {
IMG_X, IMG_Y, IMG_W, IMG_H = imagePreviewer.GetRect() ImgX, ImgY, ImgW, ImgH = imagePreviewer.GetRect()
return imagePreviewer.GetInnerRect() return imagePreviewer.GetInnerRect()
}) })

View File

@ -30,8 +30,8 @@ func GetWidth() *winsize {
func GetFontWidth() (float32, float32) { func GetFontWidth() (float32, float32) {
g := GetWidth() g := GetWidth()
fw := (float32(g.Xpixel) / float32(g.Col)) fw := float32(g.Xpixel) / float32(g.Col)
fh := (float32(g.Ypixel) / float32(g.Row)) fh := float32(g.Ypixel) / float32(g.Row)
return fw, fh return fw, fh
} }
@ -109,19 +109,9 @@ func Copy(sourceImage, destinationImage string) error {
} }
} }
func Join(stringSlice []string) string {
var _s string = stringSlice[0]
for i := 1; i < len(stringSlice); i++ {
if _s != "" {
_s += ("/" + stringSlice[i])
}
}
return _s
}
func GetFormattedString(s string, width int) string { func GetFormattedString(s string, width int) string {
if len(s) < width { if len(s) < width {
s += strings.Repeat(" ", (width - len(s))) s += strings.Repeat(" ", width-len(s))
} else { } else {
s = s[:(width - 2)] s = s[:(width - 2)]
s += " " s += " "
@ -172,7 +162,7 @@ func GetMatchedString(a []int, s, color string) string {
func Unique(intSlice []int) []int { func Unique(intSlice []int) []int {
keys := make(map[int]bool) keys := make(map[int]bool)
list := []int{} var list []int
for _, entry := range intSlice { for _, entry := range intSlice {
if _, exists := keys[entry]; !exists { if _, exists := keys[entry]; !exists {
keys[entry] = true keys[entry] = true