134 lines
2.9 KiB
Go
134 lines
2.9 KiB
Go
// Copyright 2018 The Chromium Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
// TODO(jophba):
|
|
// Add response messages from receiver
|
|
|
|
// Inject JS into viewURL to using .Eval and .Bind to send and receiver presentation connection messages
|
|
|
|
import (
|
|
"context"
|
|
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
|
|
"osp"
|
|
|
|
mdns "github.com/grandcat/zeroconf"
|
|
"github.com/zserge/webview"
|
|
)
|
|
|
|
func runServer(ctx context.Context, mdnsInstanceName string, port int) {
|
|
tlsCert, err := osp.GenerateTlsCert()
|
|
if err != nil {
|
|
log.Fatalln("Failed to generate TLS cert:", err.Error())
|
|
}
|
|
osp.RunPresentationReceiver(ctx, mdnsInstanceName, port, *tlsCert, viewUrl)
|
|
}
|
|
|
|
func browseMdns(ctx context.Context) {
|
|
entries, err := osp.BrowseMdns(ctx)
|
|
if err != nil {
|
|
log.Fatalf("Failed to browse mDNS: %v\n", err)
|
|
}
|
|
for entry := range entries {
|
|
log.Println(fmt.Sprintf("%s at %s(%s or %s):%d", entry.Instance, entry.HostName, entry.AddrIPv4, entry.AddrIPv6, entry.Port))
|
|
}
|
|
}
|
|
|
|
func getMdnsHost(entry *mdns.ServiceEntry) string {
|
|
for _, ipv6 := range entry.AddrIPv6 {
|
|
log.Printf("Choosing IPv6 address [%s]\n", ipv6)
|
|
return fmt.Sprintf("[%s]", ipv6)
|
|
}
|
|
for _, ipv4 := range entry.AddrIPv4 {
|
|
log.Printf("Choosing IPv4 address %s\n", ipv4)
|
|
return fmt.Sprintf("%s", ipv4)
|
|
}
|
|
|
|
// This shouldn't happen
|
|
log.Printf("No IP address found. Falling back to hostname %s\n", entry.HostName)
|
|
return entry.HostName
|
|
}
|
|
|
|
func flingUrl(ctx context.Context, target string, url string) {
|
|
log.Printf("Search for %s\n", target)
|
|
entries, err := osp.LookupMdns(ctx, target)
|
|
if err != nil {
|
|
log.Fatalf("Failed to browse mDNS: %v\n", err)
|
|
}
|
|
for entry := range entries {
|
|
log.Printf("Fling %s to %s:%d\n", url, entry.HostName, entry.Port)
|
|
host := getMdnsHost(entry)
|
|
err := osp.StartPresentation(ctx, host, entry.Port, url)
|
|
if err != nil {
|
|
log.Fatalln("Failed to start presentation.")
|
|
}
|
|
break
|
|
}
|
|
}
|
|
|
|
func viewUrl(url string) {
|
|
web := webview.New(webview.Settings{
|
|
// Title: "Open Screen",
|
|
URL: url,
|
|
Width: 800,
|
|
Height: 600,
|
|
Resizable: true,
|
|
})
|
|
web.Dispatch(func() {
|
|
// web.SetFullscreen(true)
|
|
})
|
|
web.Dispatch(func() {
|
|
// web.Eval();
|
|
// web.Bind();
|
|
})
|
|
web.Run()
|
|
}
|
|
|
|
func main() {
|
|
port := flag.Int("port", 10000, "port")
|
|
flag.Parse()
|
|
args := flag.Args()
|
|
if len(args) < 1 {
|
|
log.Fatalln("Usage: osp server name; osp browse; osp fling url; osp view url")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
cmd := args[0]
|
|
switch cmd {
|
|
case "server":
|
|
if len(args) < 2 {
|
|
log.Fatalln("Usage: osp server name")
|
|
}
|
|
mdnsInstanceName := args[1]
|
|
runServer(ctx, mdnsInstanceName, *port)
|
|
|
|
case "browse":
|
|
browseMdns(ctx)
|
|
|
|
case "fling":
|
|
if len(args) < 3 {
|
|
log.Fatalln("Usage: osp fling target url")
|
|
}
|
|
target := args[1]
|
|
url := args[2]
|
|
|
|
flingUrl(ctx, target, url)
|
|
|
|
case "view":
|
|
if len(args) < 2 {
|
|
log.Fatalln("Usage: osp view url")
|
|
}
|
|
|
|
url := args[1]
|
|
|
|
viewUrl(url)
|
|
}
|
|
}
|