From a6d42a1291f17b35b9ebd9f0d68c53bc3d6fd164 Mon Sep 17 00:00:00 2001 From: jolheiser Date: Wed, 20 Dec 2023 23:18:45 -0600 Subject: [PATCH] initial commit Signed-off-by: jolheiser --- LICENSE | 19 +++++++ README.md | 7 +++ example.nix | 5 ++ go.mod | 20 +++++++ go.sum | 78 ++++++++++++++++++++++++++++ main.go | 115 +++++++++++++++++++++++++++++++++++++++++ model.go | 146 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 7 files changed, 390 insertions(+) create mode 100644 LICENSE create mode 100644 README.md create mode 100644 example.nix create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go create mode 100644 model.go diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..56f09ee --- /dev/null +++ b/LICENSE @@ -0,0 +1,19 @@ +Copyright (c) 2023 John Olheiser + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/README.md b/README.md new file mode 100644 index 0000000..cdc858a --- /dev/null +++ b/README.md @@ -0,0 +1,7 @@ +# nixpl + +A ver basic side-by-side nix repl. + +## License + +[MIT](LICENSE) diff --git a/example.nix b/example.nix new file mode 100644 index 0000000..7535e45 --- /dev/null +++ b/example.nix @@ -0,0 +1,5 @@ +let + foo = "bar"; +in { + baz = foo; +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..130aa97 --- /dev/null +++ b/go.mod @@ -0,0 +1,20 @@ +module go.jolheiser.com/nixpl + +go 1.21.4 + +require ( + git.sr.ht/~rockorager/tcell-term v0.10.0 + github.com/fsnotify/fsnotify v1.7.0 + github.com/gdamore/tcell/v2 v2.6.1-0.20231203215052-2917c3801e73 +) + +require ( + github.com/creack/pty v1.1.17 // indirect + github.com/gdamore/encoding v1.0.0 // indirect + github.com/lucasb-eyer/go-colorful v1.2.0 // indirect + github.com/mattn/go-runewidth v0.0.14 // indirect + github.com/rivo/uniseg v0.4.3 // indirect + golang.org/x/sys v0.11.0 // indirect + golang.org/x/term v0.9.0 // indirect + golang.org/x/text v0.12.0 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..ba4c54d --- /dev/null +++ b/go.sum @@ -0,0 +1,78 @@ +git.sr.ht/~rockorager/tcell-term v0.10.0 h1:BqxJjtCMmLIfS6fdIal8TSiH3qPiYTjATuIRIWNVPbo= +git.sr.ht/~rockorager/tcell-term v0.10.0/go.mod h1:Snxh5CrziiA2CjyLOZ6tGAg5vMPlE+REMWT3rtKuyyQ= +github.com/creack/pty v1.1.17 h1:QeVUsEDNrLBW4tMgZHvxy18sKtr6VI492kBhUfhDJNI= +github.com/creack/pty v1.1.17/go.mod h1:MOBLtS5ELjhRRrroQr9kyvTxUAFNvYEK993ew/Vr4O4= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA= +github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM= +github.com/gdamore/encoding v1.0.0 h1:+7OoQ1Bc6eTm5niUzBa0Ctsh6JbMW6Ra+YNuAtDBdko= +github.com/gdamore/encoding v1.0.0/go.mod h1:alR0ol34c49FCSBLjhosxzcPHQbf2trDkoo5dl+VrEg= +github.com/gdamore/tcell/v2 v2.5.4/go.mod h1:dZgRy5v4iMobMEcWNYBtREnDZAT9DYmfqIkrgEMxLyw= +github.com/gdamore/tcell/v2 v2.6.1-0.20231203215052-2917c3801e73 h1:SeDV6ZUSVlTAUUPdMzPXgMyj96z+whQJRRUff8dIeic= +github.com/gdamore/tcell/v2 v2.6.1-0.20231203215052-2917c3801e73/go.mod h1:pwzJMyH4Hd0AZMJkWQ+/g01dDvYWEvmJuaiRU71Xl8k= +github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY= +github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0= +github.com/mattn/go-runewidth v0.0.14 h1:+xnbZSEeDbOIg5/mE6JF0w6n9duR1l3/WmbinWVwUuU= +github.com/mattn/go-runewidth v0.0.14/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/rivo/uniseg v0.4.3 h1:utMvzDsuh3suAEnhH0RdHmoPbU648o6CvXxTx4SBMOw= +github.com/rivo/uniseg v0.4.3/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8= +github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= +golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= +golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= +golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.9.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM= +golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/term v0.0.0-20220722155259-a9ba230a4035/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k= +golang.org/x/term v0.9.0 h1:GRRCnKYhdQrD8kfRAdQ6Zcw1P0OcELxGLKJvtjVMZ28= +golang.org/x/term v0.9.0/go.mod h1:M6DEAAIenWoTxdKrOltXcmDY3rSplQUkrvaDU5FcQyo= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/text v0.5.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/text v0.12.0 h1:k+n5B8goJNdU7hSvEtMUz3d1Q6D/XW4COJSJR6fN0mc= +golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= +golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/main.go b/main.go new file mode 100644 index 0000000..a7959ce --- /dev/null +++ b/main.go @@ -0,0 +1,115 @@ +package main + +import ( + "bytes" + "flag" + "fmt" + "os" + "os/exec" + "path/filepath" + "strings" + + "github.com/fsnotify/fsnotify" +) + +func main() { + fs := flag.NewFlagSet("nixpl", flag.ExitOnError) + if err := fs.Parse(os.Args[1:]); err != nil { + panic(err) + } + + tmp, err := os.MkdirTemp(os.TempDir(), "nixpl-*") + if err != nil { + fmt.Fprintf(os.Stderr, "%v\n", err) + os.Exit(1) + } + defer os.RemoveAll(tmp) + + apn := filepath.Join(tmp, "repl.nix") + fi, err := os.Create(apn) + if err != nil { + fmt.Fprintf(os.Stderr, "%v\n", err) + os.Exit(1) + } + defer fi.Close() + + if fs.Arg(0) != "" { + content, _ := os.ReadFile(fs.Arg(0)) + fi.Write(content) + fi.Sync() + } + + m := newModel() + go watch(apn, m) + cmd := exec.Command(os.Getenv("EDITOR"), apn) + err = m.term.Start(cmd) + if err != nil { + panic(err) + } + for { + ev := m.s.PollEvent() + if ev == nil { + break + } + m.Update(ev) + } +} + +func watch(file string, m *model) { + watcher, err := fsnotify.NewWatcher() + if err != nil { + fmt.Fprintf(os.Stderr, "%v\n", err) + os.Exit(1) + } + defer watcher.Close() + watcher.Add(file) + + fn := func() { + content, err := os.ReadFile(file) + if err != nil { + return + } + var evalBuf bytes.Buffer + var errBuf bytes.Buffer + eval := exec.Command("nix", "eval", "--expr", string(content)) + eval.Stdout = &evalBuf + eval.Stderr = &errBuf + if err := eval.Run(); err != nil { + if errBuf.Len() > 0 { + m.UpdateResult(errBuf.String()) + } + return + } + + result := evalBuf.String() + + if _, err := exec.LookPath("alejandra"); err == nil { + var formatBuf bytes.Buffer + format := exec.Command("alejandra", "-q", "-") + format.Stdin = strings.NewReader(result) + format.Stdout = &formatBuf + if err := format.Run(); err != nil { + return + } + result = formatBuf.String() + } + m.UpdateResult(result) + } + + fn() + for { + select { + case event, ok := <-watcher.Events: + if !ok { + return + } + if event.Has(fsnotify.Write) { + fn() + } + case _, ok := <-watcher.Errors: + if !ok { + return + } + } + } +} diff --git a/model.go b/model.go new file mode 100644 index 0000000..38b1b26 --- /dev/null +++ b/model.go @@ -0,0 +1,146 @@ +package main + +import ( + "fmt" + "os" + + tcellterm "git.sr.ht/~rockorager/tcell-term" + "github.com/gdamore/tcell/v2" + "github.com/gdamore/tcell/v2/views" +) + +type model struct { + term *tcellterm.VT + s tcell.Screen + termView views.View + title *views.TextBar + titleView views.View + result *views.TextArea + resultView views.View +} + +func newModel() *model { + var err error + m := &model{} + m.s, err = tcell.NewScreen() + if err != nil { + fmt.Fprintf(os.Stderr, "%v\n", err) + os.Exit(1) + } + if err = m.s.Init(); err != nil { + fmt.Fprintf(os.Stderr, "%v\n", err) + os.Exit(1) + } + m.s.EnablePaste() + w, _ := m.s.Size() + + m.title = views.NewTextBar() + m.title.SetCenter( + "nixpl", + tcell.StyleDefault.Foreground(tcell.ColorBlue). + Bold(true). + Underline(true), + ) + + m.titleView = views.NewViewPort(m.s, 0, 0, -1, 2) + m.title.SetView(m.titleView) + + m.termView = views.NewViewPort(m.s, 0, 2, w/2, -1) + m.term = tcellterm.New() + m.term.SetSurface(m.termView) + m.term.Attach(m.HandleEvent) + + m.resultView = views.NewViewPort(m.s, w/2, 2, w/2, -1) + m.result = views.NewTextArea() + m.result.SetView(m.resultView) + + m.s.EnableMouse() + + return m +} + +// Update is the main event handler. It should only be called by the main thread +func (m *model) Update(ev tcell.Event) { + switch ev := ev.(type) { + case *tcell.EventKey: + switch ev.Key() { + case tcell.KeyCtrlC: + m.term.Close() + m.s.Fini() + return + } + if m.term != nil { + m.term.HandleEvent(ev) + } + m.term.Draw() + m.result.Draw() + m.s.Show() + case *tcell.EventResize: + w, _ := m.s.Size() + if m.term != nil { + m.termView.Resize(0, 2, w/2, -1) + m.term.Resize(m.termView.Size()) + } + m.titleView.Resize(0, 0, -1, 2) + m.title.Resize() + m.resultView.Resize(w/2, 2, w/2, -1) + m.result.Resize() + m.title.Draw() + m.term.Draw() + m.result.Draw() + m.s.Sync() + return + case *tcellterm.EventRedraw: + m.term.Draw() + m.title.Draw() + m.result.Draw() + + row, col, style, vis := m.term.Cursor() + if vis { + m.s.SetCursorStyle(style) + m.s.ShowCursor(col, row+2) + + } else { + m.s.HideCursor() + } + m.s.Show() + return + case *tcellterm.EventClosed: + m.s.Clear() + m.s.Fini() + return + case *tcell.EventPaste: + m.term.HandleEvent(ev) + return + case *tcell.EventMouse: + // Translate the coordinates to our global coordinates (y-2) + x, y := ev.Position() + if y-2 < 0 { + // Event is outside our view + return + } + e := tcell.NewEventMouse(x, y-2, ev.Buttons(), ev.Modifiers()) + m.term.HandleEvent(e) + return + case *tcellterm.EventMouseMode: + m.s.EnableMouse(ev.Flags()...) + case *tcellterm.EventPanic: + m.s.Clear() + m.s.Fini() + fmt.Println(ev.Error) + } + return +} + +// HandleEvent is used to handle events from underlying widgets. Any events +// which redraw must be executed in the main goroutine by posting the event back +// to tcell +func (m *model) HandleEvent(ev tcell.Event) { + m.s.PostEvent(ev) +} + +func (m *model) UpdateResult(result string) { + m.result.SetContent(result) + m.result.Draw() + m.s.Sync() +}