meowlib/client/storage.go

package client

import (
	"database/sql"
	"os"
	"path/filepath"

	"forge.redroom.link/yves/meowlib"
	"github.com/google/uuid"
	_ "github.com/mattn/go-sqlite3"
	"google.golang.org/protobuf/proto"
)

func StoreMessage(peer *Peer, usermessage *meowlib.UserMessage, password string) error {
	var dbid string
	// If no db/no ID create DB + Tablz
	if len(peer.DbIds) == 0 {
		dbid = uuid.NewString()
		file, err := os.Create(filepath.Join(GetConfig().StoragePath, dbid))
		if err != nil {
			return err
		}
		file.Close()
		peer.DbIds = append(peer.DbIds, dbid)
		sqliteDatabase, _ := sql.Open("sqlite3", filepath.Join(GetConfig().StoragePath, dbid)) // Open the created SQLite File
		err = createTable(sqliteDatabase)
		if err != nil {
			return err
		}
		sqliteDatabase.Close()
		GetConfig().me.Save()
	} else {
		dbid = peer.DbIds[len(peer.DbIds)-1]
	}
	// Open Db
	db, _ := sql.Open("sqlite3", filepath.Join(GetConfig().StoragePath, dbid)) // Open the created SQLite File
	defer db.Close()
	// Detach Files
	if len(usermessage.Files) > 0 {
		for _, f := range usermessage.Files {
			hiddenFilename := uuid.NewString()
			// Cypher file
			encData, err := meowlib.SymEncrypt(password, f.Data)
			if err != nil {
				return err
			}
			os.WriteFile(hiddenFilename, encData, 0600)
			// replace f.Data by uuid filename
			f.Data = []byte(hiddenFilename)
		}
	}
	// Encrypt message
	out, err := proto.Marshal(usermessage)
	if err != nil {
		return err
	}
	encData, err := meowlib.SymEncrypt(password, out)
	if err != nil {
		return err
	}
	// Insert message
	insertMessageSQL := `INSERT INTO message(m) VALUES (?)`
	statement, err := db.Prepare(insertMessageSQL) // Prepare statement.
	if err != nil {
		return err
	}
	_, err = statement.Exec(encData)
	if err != nil {
		return err
	}
	return nil
}

// Get last messages from a peer
func GetLastMessages(peer *Peer, lastIdx int, count int, gotUuid string) {
	// TODO Try to get count from last DB
	// TODO Get missing from previous DB
}

func createTable(db *sql.DB) error {
	createMessageTableSQL := `CREATE TABLE message (
		"id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,		
		"m" BLOB,	
	  );` // SQL Statement for Create Table
	statement, err := db.Prepare(createMessageTableSQL) // Prepare SQL Statement
	if err != nil {
		return err
	}
	statement.Exec() // Execute SQL Statements
	return nil
}