Initial import
This commit is contained in:
183
cmd/import-letsmesh-nodes/main.go
Normal file
183
cmd/import-letsmesh-nodes/main.go
Normal file
@@ -0,0 +1,183 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/sirupsen/logrus"
|
||||
"github.com/urfave/cli/v3"
|
||||
|
||||
"git.maze.io/go/ham/protocol/meshcore"
|
||||
|
||||
"git.maze.io/ham/hamview"
|
||||
"git.maze.io/ham/hamview/internal/cmd"
|
||||
|
||||
_ "github.com/cridenour/go-postgis" // PostGIS support
|
||||
_ "github.com/lib/pq" // PostgreSQL support
|
||||
)
|
||||
|
||||
var logger = logrus.New()
|
||||
|
||||
/*
|
||||
{
|
||||
"public_key": "E119666239EE254E8E7B2937A99FE9DB7CBB58040B5D0E995B719C598CD261F6",
|
||||
"name": "~ Jonzy Heltec Repeater",
|
||||
"device_role": 2,
|
||||
"regions": [
|
||||
"OMA"
|
||||
],
|
||||
"first_seen": "2026-01-18T04:31:21.694Z",
|
||||
"last_seen": "2026-02-20T10:30:16.144Z",
|
||||
"is_mqtt_connected": true,
|
||||
"decoded_payload": {
|
||||
"lat": 41.28516,
|
||||
"lon": -96.13876,
|
||||
"mode": "Repeater",
|
||||
"name": "~ Jonzy Heltec Repeater",
|
||||
"flags": 146,
|
||||
"is_valid": true,
|
||||
"signature": "F08599E4D7357E9276B5F78246C698BFFCF14EC83D8A70CAB6F8E63EDF3FEB5CB692A60C3072593ABE0261B164709F9E012AC526B5EF08407B3520C13719900E",
|
||||
"timestamp": 1771583405,
|
||||
"public_key": "E119666239EE254E8E7B2937A99FE9DB7CBB58040B5D0E995B719C598CD261F6"
|
||||
},
|
||||
"location": {
|
||||
"latitude": 41.28516,
|
||||
"longitude": -96.13876
|
||||
},
|
||||
"node_settings": {
|
||||
"show_neighbors": true,
|
||||
"show_adverts": true
|
||||
}
|
||||
},
|
||||
*/
|
||||
type node struct {
|
||||
PublicKey string `json:"public_key"`
|
||||
Name string `json:"name"`
|
||||
Type int `json:"device_role"`
|
||||
FirstHeard time.Time `json:"first_seen"`
|
||||
LastHeard time.Time `json:"last_seen"`
|
||||
Position *meshcore.Position `json:"location"`
|
||||
Payload payload `json:"decoded_payload`
|
||||
}
|
||||
|
||||
type payload struct {
|
||||
Timestamp int64 `json:"timestamp"`
|
||||
}
|
||||
|
||||
func main() {
|
||||
cmd := &cli.Command{
|
||||
Name: "import-letsmesh-nodes",
|
||||
Action: run,
|
||||
Before: cmd.ConfigureLogging(&logger),
|
||||
Flags: append([]cli.Flag{
|
||||
&cli.StringFlag{
|
||||
Name: "dump",
|
||||
Usage: "letsmesh node json",
|
||||
Value: "letsmeshnodes.json",
|
||||
},
|
||||
}, cmd.AllFlags("hamview-collector.yaml")...),
|
||||
}
|
||||
|
||||
if err := cmd.Run(context.Background(), os.Args); err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
type collectorConfig struct {
|
||||
hamview.CollectorConfig `yaml:",inline"`
|
||||
Broker map[string]any `yaml:"broker"`
|
||||
Meshcore map[string]any `yaml:"meshcore"`
|
||||
Include []string
|
||||
}
|
||||
|
||||
func (config *collectorConfig) Includes() []string {
|
||||
includes := config.Include
|
||||
config.Include = nil
|
||||
return includes
|
||||
}
|
||||
|
||||
func run(ctx context.Context, command *cli.Command) error {
|
||||
var config collectorConfig
|
||||
if err := cmd.Load(logger, command.String(cmd.FlagConfig), &config); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
db, err := sql.Open(config.Database.Type, config.Database.Conf)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer db.Close()
|
||||
|
||||
b, err := os.ReadFile(command.String("dump"))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var nodes struct {
|
||||
Nodes []*node `json:"nodes"`
|
||||
}
|
||||
if err = json.Unmarshal(b, &nodes); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
logger.Infof("found %d nodes", len(nodes.Nodes))
|
||||
for _, node := range nodes.Nodes {
|
||||
k, err := hex.DecodeString(node.PublicKey)
|
||||
if err != nil {
|
||||
logger.Warnf("node %s has incorrect public key: %v", node.Name, err)
|
||||
continue
|
||||
}
|
||||
logger.Infof("node %s at %s", node.Name, node.Position)
|
||||
var latitude, longitude *float64
|
||||
if node.Position != nil {
|
||||
latitude = &node.Position.Latitude
|
||||
longitude = &node.Position.Longitude
|
||||
}
|
||||
if _, err = db.Exec(
|
||||
`INSERT INTO meshcore_node (
|
||||
node_type,
|
||||
public_key,
|
||||
name,
|
||||
local_time,
|
||||
first_heard,
|
||||
last_heard,
|
||||
last_latitude,
|
||||
last_longitude
|
||||
) VALUES (
|
||||
$1,
|
||||
$2,
|
||||
$3,
|
||||
$4,
|
||||
$5,
|
||||
$6,
|
||||
$7,
|
||||
$8
|
||||
)
|
||||
ON CONFLICT (public_key)
|
||||
DO UPDATE
|
||||
SET
|
||||
name = $3,
|
||||
local_time = $4,
|
||||
last_heard = $6,
|
||||
last_latitude = $7,
|
||||
last_longitude = $8
|
||||
`,
|
||||
node.Type,
|
||||
k,
|
||||
node.Name,
|
||||
time.Unix(node.Payload.Timestamp, 0),
|
||||
node.FirstHeard,
|
||||
node.LastHeard,
|
||||
latitude,
|
||||
longitude,
|
||||
); err != nil {
|
||||
logger.Fatalf("node %s insert failed: %v", node.Name, err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user