main.go

package main

import (
	"bytes"
	"fmt"
	"io"
	"log"
	"time"

	"github.com/priyangshupal/distributed-file-system/crypto"
	"github.com/priyangshupal/distributed-file-system/p2p"
	"github.com/priyangshupal/distributed-file-system/store"
)

func makeServer (listenAddr string, nodes ...string) *FileServer {
	tcpTransportOpts := p2p.TCPTransportOpts {
		ListenAddr: listenAddr,
		HandshakeFunc: p2p.NOPHandshakeFunc,
		Decoder: p2p.DefaultDecoder{},
	}
	tcpTransport := p2p.NewTCPTransport(tcpTransportOpts)
	fileServerOpts := FileServerOpts {
		EncKey: crypto.NewEncryptionKey(),
		StorageRoot: listenAddr + "_network",
		PathTransformFunc: store.CASPathTransformFunc,
		Transport: tcpTransport,
		BootstrapNodes: nodes,
	}
	s := NewFileServer(fileServerOpts)
	tcpTransport.OnPeer = s.onPeer
	return s
}

func main () {
	s1 := makeServer(":3000", "")
	s2 := makeServer(":4000", ":3000")
	s3 := makeServer(":5000", ":3000", ":4000")
	
	go func () {log.Fatal(s1.Start())}()
	
	time.Sleep(time.Millisecond * 500)
	
	go func () {log.Fatal(s2.Start())}()
	
	time.Sleep(time.Millisecond * 500)

	go s3.Start()

	time.Sleep(time.Second * 1)

	for i := 0; i < 5; i++ {
		key := fmt.Sprintf("picture_%d", i)
		data := bytes.NewReader([]byte("my big data file here!"))
		s2.Store(key, data)
		time.Sleep(time.Millisecond * 5)

		if err := s2.store.Delete(s2.ID, key); err != nil {
			log.Fatal(err)
		}
		
		r, err := s2.Get(key)
		if err != nil {
			log.Fatal(err)
		}
		b, err := io.ReadAll(r)
		if err != nil {
			log.Fatal(err)
		}
		log.Println("read bytes from the file system:", string (b))		
	}
}