You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
142 lines
2.6 KiB
142 lines
2.6 KiB
package Seeder
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"io"
|
|
"os"
|
|
"time"
|
|
|
|
"git.tovijaeschke.xyz/tovi/Capsule/Backend/Database"
|
|
"git.tovijaeschke.xyz/tovi/Capsule/Backend/Models"
|
|
)
|
|
|
|
func seedFriend(userRequestTo, userRequestFrom Models.User, accepted bool) error {
|
|
var (
|
|
friendRequest Models.FriendRequest
|
|
symKey AesKey
|
|
encPublicKey []byte
|
|
err error
|
|
)
|
|
|
|
symKey, err = GenerateAesKey()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
encPublicKey, err = symKey.AesEncrypt([]byte(PublicKey))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
friendRequest = Models.FriendRequest{
|
|
UserID: userRequestTo.ID,
|
|
FriendID: base64.StdEncoding.EncodeToString(
|
|
EncryptWithPublicKey(
|
|
[]byte(userRequestFrom.ID.String()),
|
|
decodedPublicKey,
|
|
),
|
|
),
|
|
FriendUsername: base64.StdEncoding.EncodeToString(
|
|
EncryptWithPublicKey(
|
|
[]byte(userRequestFrom.Username),
|
|
decodedPublicKey,
|
|
),
|
|
),
|
|
FriendPublicAsymmetricKey: base64.StdEncoding.EncodeToString(
|
|
encPublicKey,
|
|
),
|
|
SymmetricKey: base64.StdEncoding.EncodeToString(
|
|
EncryptWithPublicKey(symKey.Key, decodedPublicKey),
|
|
),
|
|
}
|
|
|
|
if accepted {
|
|
friendRequest.AcceptedAt.Time = time.Now()
|
|
friendRequest.AcceptedAt.Valid = true
|
|
}
|
|
|
|
return Database.CreateFriendRequest(&friendRequest)
|
|
}
|
|
|
|
func copyProfileImage() error {
|
|
var (
|
|
srcFile *os.File
|
|
dstFile *os.File
|
|
err error
|
|
)
|
|
|
|
srcFile, err = os.Open("./Database/Seeder/profile_image_enc.dat")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dstFile, err = os.Create("./attachments/profile_image")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer dstFile.Close()
|
|
_, err = io.Copy(dstFile, srcFile)
|
|
return err
|
|
}
|
|
|
|
// SeedFriends creates dummy friends for testing/development
|
|
func SeedFriends() {
|
|
var (
|
|
primaryUser Models.User
|
|
secondaryUser Models.User
|
|
accepted bool
|
|
i int
|
|
err error
|
|
)
|
|
|
|
err = copyProfileImage()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
primaryUser, err = Database.GetUserByUsername("testUser")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
secondaryUser, err = Database.GetUserByUsername("ATestUser2")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = seedFriend(primaryUser, secondaryUser, true)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = seedFriend(secondaryUser, primaryUser, true)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
accepted = false
|
|
|
|
for i = 0; i <= 5; i++ {
|
|
secondaryUser, err = Database.GetUserByUsername(userNames[i])
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if i > 3 {
|
|
accepted = true
|
|
}
|
|
|
|
err = seedFriend(primaryUser, secondaryUser, accepted)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if accepted {
|
|
err = seedFriend(secondaryUser, primaryUser, accepted)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
}
|