93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
package publicip
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
"github.com/qdm12/gluetun/internal/constants"
|
|
"github.com/qdm12/gluetun/internal/publicip/ipinfo"
|
|
)
|
|
|
|
func (l *Loop) Run(ctx context.Context, done chan<- struct{}) {
|
|
defer close(done)
|
|
|
|
select {
|
|
case <-l.start:
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
|
|
for ctx.Err() == nil {
|
|
getCtx, getCancel := context.WithCancel(ctx)
|
|
defer getCancel()
|
|
|
|
resultCh := make(chan ipinfo.Response)
|
|
errorCh := make(chan error)
|
|
go func() {
|
|
result, err := l.fetcher.FetchInfo(getCtx, nil)
|
|
if err != nil {
|
|
if getCtx.Err() == nil {
|
|
errorCh <- err
|
|
}
|
|
return
|
|
}
|
|
resultCh <- result
|
|
}()
|
|
|
|
if l.userTrigger {
|
|
l.userTrigger = false
|
|
l.running <- constants.Running
|
|
} else { // crash
|
|
l.backoffTime = defaultBackoffTime
|
|
l.statusManager.SetStatus(constants.Running)
|
|
}
|
|
|
|
stayHere := true
|
|
for stayHere {
|
|
select {
|
|
case <-ctx.Done():
|
|
getCancel()
|
|
close(errorCh)
|
|
filepath := *l.state.GetSettings().IPFilepath
|
|
l.logger.Info("Removing ip file " + filepath)
|
|
if err := os.Remove(filepath); err != nil {
|
|
l.logger.Error(err.Error())
|
|
}
|
|
return
|
|
case <-l.start:
|
|
l.userTrigger = true
|
|
getCancel()
|
|
stayHere = false
|
|
case <-l.stop:
|
|
l.userTrigger = true
|
|
l.logger.Info("stopping")
|
|
getCancel()
|
|
<-errorCh
|
|
l.stopped <- struct{}{}
|
|
case result := <-resultCh:
|
|
getCancel()
|
|
|
|
message := "Public IP address is " + result.IP.String()
|
|
message += " (" + result.Country + ", " + result.Region + ", " + result.City + ")"
|
|
l.logger.Info(message)
|
|
|
|
l.state.SetData(result)
|
|
|
|
filepath := *l.state.GetSettings().IPFilepath
|
|
err := persistPublicIP(filepath, result.IP.String(), l.puid, l.pgid)
|
|
if err != nil {
|
|
l.logger.Error(err.Error())
|
|
}
|
|
l.statusManager.SetStatus(constants.Completed)
|
|
case err := <-errorCh:
|
|
getCancel()
|
|
close(resultCh)
|
|
l.statusManager.SetStatus(constants.Crashed)
|
|
l.logAndWait(ctx, err)
|
|
stayHere = false
|
|
}
|
|
}
|
|
close(errorCh)
|
|
}
|
|
}
|