81 lines
1.5 KiB
Go
81 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"sync"
|
|
"syscall"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func serve(cCtx *cli.Context) error {
|
|
ctx, cancel := context.WithCancel(cCtx.Context)
|
|
|
|
go func() {
|
|
sig := make(chan os.Signal, 1)
|
|
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
|
|
<-sig
|
|
cancel()
|
|
}()
|
|
|
|
rawLogger, err := zap.NewDevelopment()
|
|
if err != nil {
|
|
log.Fatalf("Error initializing logger: %v", err)
|
|
}
|
|
logger = rawLogger.Sugar()
|
|
|
|
logger.Info("Running serve command, flags passed:")
|
|
for _, flag := range ServeFlags {
|
|
logger.Infof("-%s %v", flag.Names()[0], cCtx.Value(flag.Names()[0]))
|
|
}
|
|
|
|
name := cCtx.String(NameFlag)
|
|
port := cCtx.Int(PortFlag)
|
|
apiModules := cCtx.String(APIModulesFlag)
|
|
|
|
cli, err := start(cCtx, name, port, apiModules)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer cli.stop()
|
|
|
|
// Retrieve for messages
|
|
var wg sync.WaitGroup
|
|
msgCh := make(chan string)
|
|
|
|
wg.Add(1)
|
|
go cli.retrieveMessagesLoop(ctx, RetrieveInterval, msgCh, &wg)
|
|
|
|
// Send and accept contact request
|
|
dest := cCtx.String(AddFlag)
|
|
if dest != "" {
|
|
err := cli.sendContactRequest(cCtx, dest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
go func() {
|
|
msgID := <-msgCh
|
|
err = cli.sendContactRequestAcceptance(cCtx, msgID)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
return
|
|
}
|
|
}()
|
|
|
|
// Send message if mutual contact exists
|
|
sem := make(chan struct{}, 1)
|
|
wg.Add(1)
|
|
go cli.sendMessageLoop(ctx, SendInterval, &wg, sem, cancel)
|
|
|
|
wg.Wait()
|
|
logger.Info("Exiting")
|
|
|
|
return nil
|
|
}
|