Initial commit
This commit is contained in:
commit
74c912dda7
59
main.go
Normal file
59
main.go
Normal file
|
|
@ -0,0 +1,59 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"log"
|
||||||
|
|
||||||
|
app "./pkg"
|
||||||
|
)
|
||||||
|
|
||||||
|
// CLI flags
|
||||||
|
var (
|
||||||
|
title string
|
||||||
|
icon string
|
||||||
|
browser string
|
||||||
|
browserExec string
|
||||||
|
url string
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
flag.StringVar(&title, "title", "", "App launcher title, default uses the web page title")
|
||||||
|
flag.StringVar(&icon, "icon", "", "App icon image (filesystem or URL), default uses the site favicon")
|
||||||
|
flag.StringVar(&browser, "browser", "", "Browser executable, full path or command, default firefox or chromium")
|
||||||
|
flag.StringVar(&browserExec, "exec", "", "Manually provide the browser exec string, like `firefox --ssb %s` or `chromium --app=%s`, "+
|
||||||
|
"default is auto-detected based on Firefox and Chrome/ium.")
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
// Detect defaults.
|
||||||
|
if browser == "" {
|
||||||
|
detected, err := app.DetectBrowser()
|
||||||
|
if err != nil {
|
||||||
|
panic("Failed to detect your web browser. Provide the -browser or -exec options to set one manually")
|
||||||
|
}
|
||||||
|
browser = detected
|
||||||
|
log.Printf("Detected browser: %s", browser)
|
||||||
|
}
|
||||||
|
if browserExec == "" {
|
||||||
|
detected, err := app.DetectExec(browser)
|
||||||
|
if err != nil {
|
||||||
|
panic("Failed to detect the browser exec line. Provide the -exec option to set one manually")
|
||||||
|
}
|
||||||
|
browserExec = detected
|
||||||
|
log.Printf("Browser exec line for PWA: '%s'", browserExec)
|
||||||
|
}
|
||||||
|
|
||||||
|
url = flag.Arg(0)
|
||||||
|
if url == "" {
|
||||||
|
panic("Usage: pwa-launcher [options] <url>")
|
||||||
|
}
|
||||||
|
|
||||||
|
app.Run(app.Parameters{
|
||||||
|
Title: title,
|
||||||
|
URL: url,
|
||||||
|
Exec: browserExec,
|
||||||
|
Icon: icon,
|
||||||
|
})
|
||||||
|
}
|
||||||
85
pkg/app.go
Normal file
85
pkg/app.go
Normal file
|
|
@ -0,0 +1,85 @@
|
||||||
|
package app
|
||||||
|
|
||||||
|
import (
|
||||||
|
"image"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Parameters to run the pwa-launcher app.
|
||||||
|
type Parameters struct {
|
||||||
|
Title string
|
||||||
|
Icon string
|
||||||
|
iconData image.Image
|
||||||
|
URL string
|
||||||
|
Exec string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run the main app logic.
|
||||||
|
func Run(p Parameters) error {
|
||||||
|
// If no title or icon given, parse the site.
|
||||||
|
if p.Title == "" || p.Icon == "" {
|
||||||
|
insights, err := Parse(p.URL)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Insights error: %s", err)
|
||||||
|
} else {
|
||||||
|
log.Printf("insights: %+v", insights)
|
||||||
|
if p.Title == "" && insights.Title != "" {
|
||||||
|
p.Title = insights.Title
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if icons, err := DetectIcons(p.URL); err == nil {
|
||||||
|
log.Printf("detected icons: %+v", icons)
|
||||||
|
insights.Icons = append(insights.Icons, icons...)
|
||||||
|
} else {
|
||||||
|
log.Printf("DetectIcons error: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("Final insights: %+v", insights)
|
||||||
|
|
||||||
|
// Select the best icon.
|
||||||
|
if p.Icon == "" {
|
||||||
|
icon, err := BestIcon(insights)
|
||||||
|
if err != nil {
|
||||||
|
panic("No suitable app icon found, provide one manually with -icon")
|
||||||
|
}
|
||||||
|
p.iconData = icon
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Do we need to get or download an icon?
|
||||||
|
if p.iconData == nil && p.Icon != "" {
|
||||||
|
if strings.HasPrefix(p.Icon, "http:") || strings.HasPrefix(p.Icon, "https:") {
|
||||||
|
// Download an icon from the web.
|
||||||
|
png, err := ParseWebPNG(p.Icon)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Couldn't download -icon from %s: %s", p.Icon, err)
|
||||||
|
}
|
||||||
|
p.iconData = png
|
||||||
|
} else {
|
||||||
|
fh, err := os.Open(p.Icon)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
png, err := ParsePNG(fh)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
p.iconData = png
|
||||||
|
fh.Close()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Missing a title?
|
||||||
|
if p.Title == "" {
|
||||||
|
panic("Couldn't detect a page title, provide one with the -title option")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Install the icon and launcher.
|
||||||
|
Install(p)
|
||||||
|
log.Printf("Launcher installed successfully.")
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
35
pkg/browsers.go
Normal file
35
pkg/browsers.go
Normal file
|
|
@ -0,0 +1,35 @@
|
||||||
|
package app
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Preferred browsers list.
|
||||||
|
var browsers = []string{
|
||||||
|
"/usr/bin/firefox",
|
||||||
|
"/usr/bin/chromium",
|
||||||
|
"/usr/bin/google-chrome",
|
||||||
|
}
|
||||||
|
|
||||||
|
// DetectBrowser auto-detects a preferred web browser and returns its path.
|
||||||
|
func DetectBrowser() (string, error) {
|
||||||
|
for _, path := range browsers {
|
||||||
|
if _, err := os.Stat(path); err == nil {
|
||||||
|
return path, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", errors.New("failed to auto-detect a web browser")
|
||||||
|
}
|
||||||
|
|
||||||
|
// DetectExec auto-detects the exec line syntax for the selected browser.
|
||||||
|
func DetectExec(browser string) (string, error) {
|
||||||
|
if strings.Contains(browser, "firefox") {
|
||||||
|
return fmt.Sprintf("%s --ssb %%s", browser), nil
|
||||||
|
} else if strings.Contains(browser, "chromium") || strings.Contains(browser, "chrome") {
|
||||||
|
return fmt.Sprintf("%s --app=%%s", browser), nil
|
||||||
|
}
|
||||||
|
return "", errors.New("failed to auto-detect the browser exec line")
|
||||||
|
}
|
||||||
138
pkg/image.go
Normal file
138
pkg/image.go
Normal file
|
|
@ -0,0 +1,138 @@
|
||||||
|
package app
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"image"
|
||||||
|
"image/png"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
// BestIcon selects the best PNG icon image.
|
||||||
|
func BestIcon(i Insights) (image.Image, error) {
|
||||||
|
// Ensure all icons are loaded.
|
||||||
|
for _, icon := range i.Icons {
|
||||||
|
if icon.Data == nil {
|
||||||
|
img, err := ParseWebPNG(icon.URL)
|
||||||
|
if err == nil {
|
||||||
|
icon.Data = img
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the biggest icon.
|
||||||
|
var biggest image.Image
|
||||||
|
var maxSize int
|
||||||
|
for _, icon := range i.Icons {
|
||||||
|
w := icon.Width()
|
||||||
|
if w > maxSize && icon.Data != nil {
|
||||||
|
biggest = icon.Data
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if biggest == nil {
|
||||||
|
return nil, errors.New("no suitable icon available")
|
||||||
|
}
|
||||||
|
return biggest, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParsePNG parses a PNG image.
|
||||||
|
func ParsePNG(r io.Reader) (image.Image, error) {
|
||||||
|
img, err := png.Decode(r)
|
||||||
|
return img, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseWebPNG parses a PNG image by HTTP URL.
|
||||||
|
func ParseWebPNG(url string) (image.Image, error) {
|
||||||
|
resp, err := http.Get(AddScheme(url))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
return ParsePNG(resp.Body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IcoToPNG converts a .ico file into PNG images.
|
||||||
|
func IcoToPNG(r io.Reader) ([]image.Image, error) {
|
||||||
|
var result = []image.Image{}
|
||||||
|
bin, _ := ioutil.ReadAll(r)
|
||||||
|
|
||||||
|
// Check if the image is already a PNG format.
|
||||||
|
if string(bin[1:4]) == "PNG" {
|
||||||
|
fmt.Printf("IcoToPNG: ico is already a png format!")
|
||||||
|
buf := bytes.NewBuffer(bin)
|
||||||
|
img, err := png.Decode(buf)
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
return []image.Image{img}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ensure we have icotool available.
|
||||||
|
if !HasICOTool() {
|
||||||
|
return result, errors.New("icotool not available")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a temp directory to extract this icon.
|
||||||
|
dir, err := ioutil.TempDir("", "pwa-launcher")
|
||||||
|
if err != nil {
|
||||||
|
return result, fmt.Errorf("TempDir: %s", err)
|
||||||
|
}
|
||||||
|
log.Printf("Temp dir to extract .ico file: %s", dir)
|
||||||
|
// defer os.RemoveAll(dir)
|
||||||
|
|
||||||
|
// Write the .ico binary to the source file.
|
||||||
|
icoFile := filepath.Join(dir, "favicon.ico")
|
||||||
|
fh, err := os.Create(icoFile)
|
||||||
|
if err != nil {
|
||||||
|
return result, fmt.Errorf("Write %s: %s", icoFile, err)
|
||||||
|
}
|
||||||
|
fh.Write(bin)
|
||||||
|
fh.Close()
|
||||||
|
|
||||||
|
// Run the commands.
|
||||||
|
cmd := exec.Command(
|
||||||
|
"icotool", "-x", "-o", dir, icoFile,
|
||||||
|
)
|
||||||
|
stdout, err := cmd.CombinedOutput()
|
||||||
|
fmt.Println(stdout)
|
||||||
|
|
||||||
|
// Glom the PNG images.
|
||||||
|
files, _ := filepath.Glob(filepath.Join(dir, "*.png"))
|
||||||
|
for _, file := range files {
|
||||||
|
fh, err := os.Open(file)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if img, err := ParsePNG(fh); err == nil {
|
||||||
|
result = append(result, img)
|
||||||
|
}
|
||||||
|
|
||||||
|
fh.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// HasICOTool checks if the icotool binary is available.
|
||||||
|
func HasICOTool() bool {
|
||||||
|
_, err := exec.LookPath("icotool")
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf(
|
||||||
|
"********\n" +
|
||||||
|
"WARNING: command `icotool` was not found on your system.\n" +
|
||||||
|
"Extracting PNG images from .ico files will not be supported.\n" +
|
||||||
|
"To remedy this, install icoutils, e.g. `sudo apt install icoutils`\n" +
|
||||||
|
"********\n",
|
||||||
|
)
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
219
pkg/net.go
Normal file
219
pkg/net.go
Normal file
|
|
@ -0,0 +1,219 @@
|
||||||
|
package app
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"image"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"golang.org/x/net/html"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Insights parsed from a web URL.
|
||||||
|
type Insights struct {
|
||||||
|
Title string
|
||||||
|
Icons []Icon
|
||||||
|
}
|
||||||
|
|
||||||
|
// Icon found for the web URL.
|
||||||
|
type Icon struct {
|
||||||
|
Size string
|
||||||
|
URL string
|
||||||
|
Data image.Image // raw image data, if available
|
||||||
|
}
|
||||||
|
|
||||||
|
// Width returns the icon's width.
|
||||||
|
func (i Icon) Width() int {
|
||||||
|
parts := strings.Split(i.Size, "x")
|
||||||
|
if len(parts) > 1 {
|
||||||
|
w, _ := strconv.Atoi(parts[0])
|
||||||
|
return w
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse a web page and return page insights.
|
||||||
|
func Parse(url string) (Insights, error) {
|
||||||
|
log.Printf("### Parse HTML on %s for Title and Icon URLs", url)
|
||||||
|
var (
|
||||||
|
result = Insights{
|
||||||
|
Icons: []Icon{},
|
||||||
|
}
|
||||||
|
inTag = ""
|
||||||
|
)
|
||||||
|
|
||||||
|
resp, err := http.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
return result, fmt.Errorf("HTTP error: %s", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
z := html.NewTokenizer(resp.Body)
|
||||||
|
parsing := true
|
||||||
|
for parsing {
|
||||||
|
tt := z.Next()
|
||||||
|
token := z.Token()
|
||||||
|
switch tt {
|
||||||
|
case html.ErrorToken:
|
||||||
|
fmt.Printf("error: %s\n", z.Err())
|
||||||
|
if z.Err() == io.EOF {
|
||||||
|
// successful error condition
|
||||||
|
parsing = false
|
||||||
|
break
|
||||||
|
}
|
||||||
|
return result, fmt.Errorf("HTML parsing error: %s", z.Err())
|
||||||
|
|
||||||
|
case html.TextToken:
|
||||||
|
if inTag == "title" && result.Title == "" {
|
||||||
|
result.Title += token.Data
|
||||||
|
}
|
||||||
|
|
||||||
|
case html.StartTagToken:
|
||||||
|
inTag = token.Data
|
||||||
|
|
||||||
|
// Looking for <link rel="shortcut icon">
|
||||||
|
if token.Data == "link" {
|
||||||
|
attr := AttrDict(token)
|
||||||
|
rel, _ := attr["rel"]
|
||||||
|
sizes, _ := attr["sizes"]
|
||||||
|
href, _ := attr["href"]
|
||||||
|
|
||||||
|
// Ensure "//" URIs start with "https://"
|
||||||
|
href = AddScheme(href)
|
||||||
|
|
||||||
|
if rel == "shortcut icon" {
|
||||||
|
log.Printf(`Found <link rel="shortcut icon"> URL: %s`, href)
|
||||||
|
if sizes == "" {
|
||||||
|
sizes = "16x16"
|
||||||
|
}
|
||||||
|
|
||||||
|
// If an ico file, extract the PNGs.
|
||||||
|
if filepath.Ext(href) == ".ico" {
|
||||||
|
log.Printf("The favicon is a .ico file, extracting PNGs")
|
||||||
|
if resp, err := http.Get(href); err == nil {
|
||||||
|
if pngs, err := IcoToPNG(resp.Body); err == nil {
|
||||||
|
for _, png := range pngs {
|
||||||
|
size := png.Bounds().Size()
|
||||||
|
result.Icons = append(result.Icons, Icon{
|
||||||
|
Size: fmt.Sprintf("%dx%d", size.X, size.Y),
|
||||||
|
URL: href,
|
||||||
|
Data: png,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
resp.Body.Close()
|
||||||
|
continue
|
||||||
|
} else {
|
||||||
|
log.Printf("Error extracting PNG from %s: %s", href, err)
|
||||||
|
}
|
||||||
|
resp.Body.Close()
|
||||||
|
} else {
|
||||||
|
log.Printf("HTTP error downloading %s: %s", href, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result.Icons = append(result.Icons, Icon{
|
||||||
|
Size: sizes,
|
||||||
|
URL: href,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case html.EndTagToken, html.SelfClosingTagToken:
|
||||||
|
inTag = ""
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DetectIcons checks well-known icon URLs on a base domain.
|
||||||
|
func DetectIcons(weburl string) ([]Icon, error) {
|
||||||
|
var result = []Icon{}
|
||||||
|
|
||||||
|
uri, err := url.Parse(weburl)
|
||||||
|
if err != nil {
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
|
||||||
|
baseURL := uri.Scheme + "://" + uri.Host
|
||||||
|
log.Printf("### Auto-detecting Icon URLs from site %s", baseURL)
|
||||||
|
|
||||||
|
tryURI := []string{
|
||||||
|
"/apple-touch-icon.png",
|
||||||
|
"/apple-touch-icon-180x180.png",
|
||||||
|
"/apple-touch-icon-152x152.png",
|
||||||
|
"/apple-touch-icon-144x144.png",
|
||||||
|
"/apple-touch-icon-120x120.png",
|
||||||
|
"/apple-touch-icon-114x114.png",
|
||||||
|
"/apple-touch-icon-76x76.png",
|
||||||
|
"/apple-touch-icon-72x72.png",
|
||||||
|
"/apple-touch-icon-57x57.png",
|
||||||
|
"/apple-touch-icon-60x60.png",
|
||||||
|
"/favicon.ico",
|
||||||
|
}
|
||||||
|
for _, uri := range tryURI {
|
||||||
|
resp, err := http.Get(baseURL + uri)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
// PNG images?
|
||||||
|
if filepath.Ext(uri) == ".png" {
|
||||||
|
png, err := ParsePNG(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
size := png.Bounds().Size()
|
||||||
|
result = append(result, Icon{
|
||||||
|
Size: fmt.Sprintf("%dx%d", size.X, size.Y),
|
||||||
|
URL: baseURL + uri,
|
||||||
|
})
|
||||||
|
} else if filepath.Ext(uri) == ".ico" {
|
||||||
|
// Extract the PNG images from the icon.
|
||||||
|
pngs, err := IcoToPNG(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, png := range pngs {
|
||||||
|
size := png.Bounds().Size()
|
||||||
|
result = append(result, Icon{
|
||||||
|
Size: fmt.Sprintf("%dx%d", size.X, size.Y),
|
||||||
|
URL: baseURL + uri,
|
||||||
|
Data: png,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("Found icon: %s", uri)
|
||||||
|
_ = resp
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// AttrDict converts an HTML Token attributes list into a hash map.
|
||||||
|
func AttrDict(token html.Token) map[string]string {
|
||||||
|
var result = map[string]string{}
|
||||||
|
for _, attr := range token.Attr {
|
||||||
|
result[attr.Key] = attr.Val
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddScheme ensures an HTTP URL has a valid scheme, converting "//" into
|
||||||
|
// "https://"
|
||||||
|
func AddScheme(uri string) string {
|
||||||
|
if strings.HasPrefix(uri, "//") {
|
||||||
|
return "https:" + uri
|
||||||
|
}
|
||||||
|
return uri
|
||||||
|
}
|
||||||
101
pkg/templates.go
Normal file
101
pkg/templates.go
Normal file
|
|
@ -0,0 +1,101 @@
|
||||||
|
package app
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"image/png"
|
||||||
|
"log"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Launcher is the source code template for a desktop launcher.
|
||||||
|
const Launcher = `[Desktop Entry]
|
||||||
|
Version=1.0
|
||||||
|
Name={{ .Name }}
|
||||||
|
GenericName={{ .Name }}
|
||||||
|
Comment=Progressive Web App
|
||||||
|
Exec={{ .Exec }}
|
||||||
|
Icon={{ .Icon }}
|
||||||
|
Terminal=false
|
||||||
|
Type=Application
|
||||||
|
StartupNotify=true
|
||||||
|
Categories=Network;
|
||||||
|
`
|
||||||
|
|
||||||
|
// Vars for the launcher template.
|
||||||
|
type Vars struct {
|
||||||
|
Name string
|
||||||
|
Exec string
|
||||||
|
Icon string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Install the launcher and icon.
|
||||||
|
func Install(p Parameters) {
|
||||||
|
var (
|
||||||
|
HOME = os.Getenv("HOME")
|
||||||
|
basename = UID(p)
|
||||||
|
)
|
||||||
|
|
||||||
|
if HOME == "" {
|
||||||
|
panic("No $HOME variable found for current user!")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ensure directories exist.
|
||||||
|
for _, path := range []string{
|
||||||
|
filepath.Join(HOME, ".local", "share", "applications"),
|
||||||
|
filepath.Join(HOME, ".config", "pwa-launcher"),
|
||||||
|
} {
|
||||||
|
err := os.MkdirAll(path, 0755)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Failed to mkdir %s: %s", path, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write the PNG image.
|
||||||
|
var iconPath = filepath.Join(HOME, ".config", "pwa-launcher", basename+".png")
|
||||||
|
log.Printf("Write icon to: %s", iconPath)
|
||||||
|
icoFH, err := os.Create(iconPath)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
defer icoFH.Close()
|
||||||
|
if err := png.Encode(icoFH, p.iconData); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write the desktop launcher.
|
||||||
|
var launcherPath = filepath.Join(HOME, ".local", "share", "applications", basename+".desktop")
|
||||||
|
log.Printf("Write launcher to: %s", launcherPath)
|
||||||
|
|
||||||
|
fh, err := os.Create(launcherPath)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
defer fh.Close()
|
||||||
|
t := template.Must(template.New("launcher").Parse(Launcher))
|
||||||
|
err = t.Execute(fh, Vars{
|
||||||
|
Name: p.Title,
|
||||||
|
Exec: fmt.Sprintf(p.Exec, p.URL),
|
||||||
|
Icon: iconPath,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// UID generates a unique name for the launcher and icon files.
|
||||||
|
func UID(p Parameters) string {
|
||||||
|
var name = "pwa-"
|
||||||
|
|
||||||
|
uri, err := url.Parse(p.URL)
|
||||||
|
if err == nil {
|
||||||
|
name += strings.Replace(uri.Host, ":", "_", -1) + "-"
|
||||||
|
}
|
||||||
|
|
||||||
|
name += fmt.Sprintf("%d", time.Now().Unix())
|
||||||
|
return name
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user