some more examples

Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
Jeff Carr 2019-05-29 09:54:28 -07:00
parent a06fd3c4a7
commit 6a65efc40d
7 changed files with 258 additions and 0 deletions

1
.gitignore vendored
View File

@ -6,6 +6,7 @@ example-systray/example-systray
example-UI-table/example-UI-table
example-dnssecsocket/example-dnssecsocket
example-table/example-table
example-aminal/example-aminal
test1/test1
test2/test2

7
example-aminal/Makefile Normal file
View File

@ -0,0 +1,7 @@
build:
go build
example-aminal
build-windows:
env GOOS=windows GOARCH=amd64 go build

122
example-aminal/config.go Normal file
View File

@ -0,0 +1,122 @@
package main
import (
"flag"
"fmt"
"io/ioutil"
"os"
"os/user"
"path/filepath"
"github.com/liamg/aminal/config"
"github.com/liamg/aminal/version"
)
func getActuallyProvidedFlags() map[string]bool {
result := make(map[string]bool)
flag.Visit(func(f *flag.Flag) {
result[f.Name] = true
})
return result
}
func getConfig() *config.Config {
showVersion := false
ignoreConfig := false
shell := ""
debugMode := false
slomo := false
if flag.Parsed() == false {
flag.BoolVar(&showVersion, "version", showVersion, "Output version information")
flag.BoolVar(&ignoreConfig, "ignore-config", ignoreConfig, "Ignore user config files and use defaults")
flag.StringVar(&shell, "shell", shell, "Specify the shell to use")
flag.BoolVar(&debugMode, "debug", debugMode, "Enable debug logging")
flag.BoolVar(&slomo, "slomo", slomo, "Render in slow motion (useful for debugging)")
flag.Parse() // actual parsing and fetching flags from the command line
}
actuallyProvidedFlags := getActuallyProvidedFlags()
if showVersion {
v := version.Version
if v == "" {
v = "development"
}
fmt.Println(v)
os.Exit(0)
}
var conf *config.Config
if ignoreConfig {
conf = &config.DefaultConfig
} else {
conf = loadConfigFile()
}
// Override values in the configuration file with the values specified in the command line, if any.
if actuallyProvidedFlags["shell"] {
conf.Shell = shell
}
if actuallyProvidedFlags["debug"] {
conf.DebugMode = debugMode
}
if actuallyProvidedFlags["slomo"] {
conf.Slomo = slomo
}
return conf
}
func loadConfigFile() *config.Config {
usr, err := user.Current()
if err != nil {
fmt.Printf("Failed to get current user information: %s\n", err)
return &config.DefaultConfig
}
home := usr.HomeDir
if home == "" {
return &config.DefaultConfig
}
places := []string{}
xdgHome := os.Getenv("XDG_CONFIG_HOME")
if xdgHome != "" {
places = append(places, filepath.Join(xdgHome, "aminal/config.toml"))
}
places = append(places, filepath.Join(home, ".config/aminal/config.toml"))
places = append(places, filepath.Join(home, ".aminal.toml"))
for _, place := range places {
if b, err := ioutil.ReadFile(place); err == nil {
if c, err := config.Parse(b); err == nil {
return c
}
fmt.Printf("Invalid config at %s: %s\n", place, err)
}
}
if b, err := config.DefaultConfig.Encode(); err != nil {
fmt.Printf("Failed to encode config file: %s\n", err)
} else {
err = os.MkdirAll(filepath.Dir(places[0]), 0744)
if err != nil {
fmt.Printf("Failed to create config file directory: %s\n", err)
} else {
if err = ioutil.WriteFile(places[0], b, 0644); err != nil {
fmt.Printf("Failed to encode config file: %s\n", err)
}
}
}
return &config.DefaultConfig
}

25
example-aminal/logger.go Normal file
View File

@ -0,0 +1,25 @@
package main
import (
"fmt"
"github.com/liamg/aminal/config"
"go.uber.org/zap"
)
func getLogger(conf *config.Config) (*zap.SugaredLogger, error) {
var logger *zap.Logger
var err error
if conf.DebugMode {
logger, err = zap.NewDevelopment()
} else {
loggerConfig := zap.NewProductionConfig()
loggerConfig.Encoding = "console"
logger, err = loggerConfig.Build()
}
if err != nil {
return nil, fmt.Errorf("Failed to create logger: %s", err)
}
return logger.Sugar(), nil
}

80
example-aminal/main.go Normal file
View File

@ -0,0 +1,80 @@
package main
import (
"fmt"
"github.com/liamg/aminal/gui"
"github.com/liamg/aminal/platform"
"github.com/liamg/aminal/terminal"
"github.com/riywo/loginshell"
"os"
"runtime"
)
type callback func(terminal *terminal.Terminal, g *gui.GUI)
func init() {
runtime.LockOSThread()
}
func main() {
initialize(nil)
}
func initialize(unitTestfunc callback) {
conf := getConfig()
logger, err := getLogger(conf)
if err != nil {
fmt.Printf("Failed to create logger: %s\n", err)
os.Exit(1)
}
defer logger.Sync()
logger.Infof("Allocating pty...")
pty, err := platform.NewPty(80, 25)
if err != nil {
logger.Fatalf("Failed to allocate pty: %s", err)
}
shellStr := conf.Shell
if shellStr == "" {
loginShell, err := loginshell.Shell()
if err != nil {
logger.Fatalf("Failed to ascertain your shell: %s", err)
}
shellStr = loginShell
}
os.Setenv("TERM", "xterm-256color") // controversial! easier than installing terminfo everywhere, but obviously going to be slightly different to xterm functionality, so we'll see...
os.Setenv("COLORTERM", "truecolor")
guestProcess, err := pty.CreateGuestProcess(shellStr)
if err != nil {
pty.Close()
logger.Fatalf("Failed to start your shell: %s", err)
}
defer guestProcess.Close()
logger.Infof("Creating terminal...")
terminal := terminal.New(pty, logger, conf)
g, err := gui.New(conf, terminal, logger)
if err != nil {
logger.Fatalf("Cannot start: %s", err)
}
if unitTestfunc != nil {
go unitTestfunc(terminal, g)
} else {
go func() {
if err := guestProcess.Wait(); err != nil {
logger.Fatalf("Failed to wait for guest process: %s", err)
}
g.Close()
}()
}
if err := g.Render(); err != nil {
logger.Fatalf("Render error: %s", err)
}
}

2
example-test/Makefile Normal file
View File

@ -0,0 +1,2 @@
all:
go test --race

21
example-test/ex_test.go Normal file
View File

@ -0,0 +1,21 @@
package ex
import "sync"
import "testing"
type Item struct{}
func TestAppend(t *testing.T) {
var list []Item
n := 2
wg := sync.WaitGroup{}
wg.Add(n)
for i := 0; i < n; i++ {
go func() {
defer wg.Done()
list = append(list, Item{})
}()
}
wg.Wait()
}