Exit gracefully when SIGTERM is received

[micro.git] / cmd / micro / micro.go

diff --git a/cmd/micro/micro.go b/cmd/micro/micro.go
index a4d22b70e5db8f95e194c532e0499c33957b49cd..cae723996e29f8b800acd347fc24c8ba7c70331b 100644 (file)
--- a/cmd/micro/micro.go
+++ b/cmd/micro/micro.go
@@ -5,10 +5,12 @@ import (
        "fmt"
        "io/ioutil"
        "os"
+       "os/signal"
        "regexp"
        "runtime"
        "sort"
        "strconv"
+       "syscall"
        "time"
 
        "github.com/go-errors/errors"
@@ -16,6 +18,7 @@ import (
        lua "github.com/yuin/gopher-lua"
        "github.com/zyedidia/micro/v2/internal/action"
        "github.com/zyedidia/micro/v2/internal/buffer"
+       "github.com/zyedidia/micro/v2/internal/clipboard"
        "github.com/zyedidia/micro/v2/internal/config"
        ulua "github.com/zyedidia/micro/v2/internal/lua"
        "github.com/zyedidia/micro/v2/internal/screen"
@@ -26,7 +29,6 @@ import (
 
 var (
        // Event channel
-       events   chan tcell.Event
        autosave chan bool
 
        // Command line flags
@@ -269,9 +271,25 @@ func main() {
                os.Exit(1)
        }
 
+       c := make(chan os.Signal, 1)
+       signal.Notify(c, os.Kill, syscall.SIGTERM)
+
+       go func() {
+               <-c
+               if screen.Screen != nil {
+                       screen.Screen.Fini()
+               }
+               os.Exit(0)
+       }()
+
+       m := clipboard.SetMethod(config.GetGlobalOption("clipboard").(string))
+       clipErr := clipboard.Initialize(m)
+
        defer func() {
                if err := recover(); err != nil {
-                       screen.Screen.Fini()
+                       if screen.Screen != nil {
+                               screen.Screen.Fini()
+                       }
                        fmt.Println("Micro encountered an error:", err)
                        // backup all open buffers
                        for _, b := range buffer.OpenBuffers {
@@ -296,6 +314,11 @@ func main() {
                screen.TermMessage(err)
        }
 
+       err = config.RunPluginFn("preinit")
+       if err != nil {
+               screen.TermMessage(err)
+       }
+
        args := flag.Args()
        b := LoadInput(args)
 
@@ -313,7 +336,16 @@ func main() {
                screen.TermMessage(err)
        }
 
-       events = make(chan tcell.Event)
+       err = config.RunPluginFn("postinit")
+       if err != nil {
+               screen.TermMessage(err)
+       }
+
+       if clipErr != nil {
+               action.InfoBar.Error(clipErr, " or change 'clipboard' option")
+       }
+
+       screen.Events = make(chan tcell.Event)
 
        // Here is the event loop which runs in a separate thread
        go func() {
@@ -322,7 +354,7 @@ func main() {
                        e := screen.Screen.PollEvent()
                        screen.Unlock()
                        if e != nil {
-                               events <- e
+                               screen.Events <- e
                        }
                }
        }()
@@ -335,7 +367,7 @@ func main() {
 
        // wait for initial resize event
        select {
-       case event := <-events:
+       case event := <-screen.Events:
                action.Tabs.HandleEvent(event)
        case <-time.After(10 * time.Millisecond):
                // time out after 10ms
@@ -388,7 +420,7 @@ func DoEvent() {
                }
                ulua.Lock.Unlock()
        case <-shell.CloseTerms:
-       case event = <-events:
+       case event = <-screen.Events:
        case <-screen.DrawChan():
                for len(screen.DrawChan()) > 0 {
                        <-screen.DrawChan()