// mesh provides implementation agnostic logic for managing // the mesh package mesh import ( "net" "github.com/tim-beatham/wgmesh/pkg/conf" "golang.zx2c4.com/wireguard/wgctrl" "golang.zx2c4.com/wireguard/wgctrl/wgtypes" ) // MeshNode represents an implementation of a node in a mesh type MeshNode interface { // GetHostEndpoint: gets the gRPC endpoint of the node GetHostEndpoint() string // GetPublicKey: gets the public key of the node GetPublicKey() (wgtypes.Key, error) // GetWgEndpoint(): get IP and port of the wireguard endpoint GetWgEndpoint() string // GetWgHost: get the IP address of the WireGuard node GetWgHost() *net.IPNet // GetTimestamp: get the UNIX time stamp of the ndoe GetTimeStamp() int64 // GetRoutes: returns the routes that the nodes provides GetRoutes() []string // GetIdentifier: returns the identifier of the node GetIdentifier() string // GetDescription: returns the description for this node GetDescription() string } type MeshSnapshot interface { // GetNodes() returns the nodes in the mesh GetNodes() map[string]MeshNode } // MeshSyncer syncs two meshes type MeshSyncer interface { GenerateMessage() ([]byte, bool) RecvMessage(mesg []byte) error Complete() } // Mesh: Represents an implementation of a mesh type MeshProvider interface { // AddNode() adds a node to the mesh AddNode(node MeshNode) // GetMesh() returns a snapshot of the mesh provided by the mesh provider GetMesh() (MeshSnapshot, error) // GetMeshId() returns the ID of the mesh network GetMeshId() string // Save() saves the mesh network Save() []byte // Load() loads a mesh network Load([]byte) error // GetDevice() get the device corresponding with the mesh GetDevice() (*wgtypes.Device, error) // HasChanges returns true if we have changes since last time we synced HasChanges() bool // Record that we have changes and save the corresponding changes SaveChanges() // UpdateTimeStamp: update the timestamp of the given node UpdateTimeStamp(nodeId string) error // AddRoutes: adds routes to the given node AddRoutes(nodeId string, route ...string) error // DeleteRoutes: deletes the routes from the node RemoveRoutes(nodeId string, route ...string) error // GetSyncer: returns the automerge syncer for sync GetSyncer() MeshSyncer // SetDescription: sets the description of this automerge data type SetDescription(nodeId string, description string) error // Prune: prunes all nodes that have not updated their timestamp in // pruneAmount seconds Prune(pruneAmount int) error } // HostParameters contains the IDs of a node type HostParameters struct { HostEndpoint string } // MeshProviderFactoryParams parameters required to build a mesh provider type MeshProviderFactoryParams struct { DevName string MeshId string Port int Conf *conf.WgMeshConfiguration Client *wgctrl.Client } // MeshProviderFactory creates an instance of a mesh provider type MeshProviderFactory interface { CreateMesh(params *MeshProviderFactoryParams) (MeshProvider, error) } // MeshNodeFactoryParams are the parameters required to construct // a mesh node type MeshNodeFactoryParams struct { PublicKey *wgtypes.Key NodeIP net.IP WgPort int Endpoint string } // MeshBuilder build the hosts mesh node for it to be added to the mesh type MeshNodeFactory interface { Build(params *MeshNodeFactoryParams) MeshNode }