82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/fiskerinc/cloud-services/services/attendant/services"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/cache"
|
|
"github.com/fiskerinc/cloud-services/pkg/common"
|
|
"github.com/fiskerinc/cloud-services/pkg/logger"
|
|
"github.com/fiskerinc/cloud-services/pkg/redis"
|
|
"github.com/fiskerinc/cloud-services/pkg/validator"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// Mega ducky repeat of /handlers/get_file_keys. Need a code re-org
|
|
func GetFileKeys(db *services.DB, device common.Device, id string, data []byte) error {
|
|
logger.Debug().Msgf("GetFileKeys %v %s", device, id)
|
|
var err error
|
|
var req *common.FileKeysRequest
|
|
|
|
client := services.RedisClientPool().GetFromPool()
|
|
defer client.Close()
|
|
|
|
req, err = parseGetFileKeysRequest(data)
|
|
if err != nil {
|
|
notifyFileKeysGeneralError(client, device, id, err)
|
|
return err
|
|
}
|
|
|
|
keys, err := cache.RetrieveFileEncryptionParams(client, db.GetFileKeys(), req.FileIDs)
|
|
if err != nil {
|
|
notifyFileKeysGeneralError(client, device, id, err)
|
|
return err
|
|
}
|
|
|
|
err = client.SafePublishMessage(device.Key(id), common.Message{
|
|
Handler: "filekeys",
|
|
Data: keys,
|
|
})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
logger.Debug().Msgf("GetFileKeys sent %v %s", device, id)
|
|
|
|
return nil
|
|
}
|
|
|
|
func parseGetFileKeysRequest(data []byte) (*common.FileKeysRequest, error) {
|
|
var status common.FileKeysRequest
|
|
|
|
err := json.Unmarshal(data, &status)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
err = validator.ValidateStruct(status)
|
|
if err != nil {
|
|
return &status, errors.WithStack(err)
|
|
}
|
|
|
|
return &status, nil
|
|
}
|
|
|
|
func notifyFileKeysGeneralError(client redis.Client, device common.Device, id string, err error) {
|
|
e := client.SafePublishMessage(device.Key(id), common.Message{
|
|
Handler: "filekeys",
|
|
Data: []common.FileKeyResponse{
|
|
{
|
|
FileID: "0",
|
|
Error: err.Error(),
|
|
},
|
|
},
|
|
})
|
|
if e != nil {
|
|
logger.Error().Err(errors.WithStack(e)).Send()
|
|
}
|
|
}
|