Ajout récupération de fichier (fingerprint à check)
This commit is contained in:
60
pkg/protocol/rules/readers/RetrieveFileRule.go
Normal file
60
pkg/protocol/rules/readers/RetrieveFileRule.go
Normal file
@@ -0,0 +1,60 @@
|
||||
package readers
|
||||
|
||||
import (
|
||||
"StoreBackEnd/pkg/protocol"
|
||||
"StoreBackEnd/pkg/protocol/repository"
|
||||
"StoreBackEnd/pkg/protocol/rules/writers"
|
||||
"StoreBackEnd/pkg/utils"
|
||||
"bufio"
|
||||
"fmt"
|
||||
"os"
|
||||
)
|
||||
|
||||
// RetrieveFileRulePrefix Identifiant de cette règle
|
||||
const RetrieveFileRulePrefix = "RETRIEVEFILE"
|
||||
|
||||
// RetrieveFileRule Demande de suppression d'un fichier
|
||||
type RetrieveFileRule struct {
|
||||
// Cmd Nom de la règle
|
||||
Cmd string
|
||||
// matcher Permet de vérifier le matching
|
||||
matcher *protocol.RegexMatcher
|
||||
// protocolRepo Instance de ProtocolRepository
|
||||
protocolRepo *repository.ProtocolRepository
|
||||
// storagePath Chemin de stockage du fichier
|
||||
storagePath string
|
||||
}
|
||||
|
||||
// CreateRetrieveFileRule Création d'une instance de RetrieveFileRule
|
||||
func CreateRetrieveFileRule(pattern string, protocolRepo *repository.ProtocolRepository, storagePath string) protocol.IProtocolReader {
|
||||
return &RetrieveFileRule{
|
||||
Cmd: RetrieveFileRulePrefix,
|
||||
matcher: protocol.CreateRegexMatcher(pattern),
|
||||
protocolRepo: protocolRepo,
|
||||
storagePath: storagePath,
|
||||
}
|
||||
}
|
||||
|
||||
func (rule RetrieveFileRule) GetCmd() string {
|
||||
return rule.Cmd
|
||||
}
|
||||
|
||||
func (rule RetrieveFileRule) Execute(data string) (*protocol.ProtocolWriterResult, func(reader *bufio.Reader) *protocol.ProtocolWriterResult) {
|
||||
|
||||
if rule.Match(data) {
|
||||
values := rule.matcher.Parse(data)
|
||||
hashFileName := values[1]
|
||||
|
||||
path := fmt.Sprintf("%s/%s", rule.storagePath, hashFileName)
|
||||
if fileinfo, err := os.Stat(path); err == nil {
|
||||
println("OKOKOKOEKOEKE : ", hashFileName, fmt.Sprintf("%d", fileinfo.Size()), fmt.Sprintf("%x", utils.GenFingerPrint(path)))
|
||||
|
||||
return rule.protocolRepo.ExecuteWriter(writers.RetrieveOkRulePrefix, hashFileName, fmt.Sprintf("%d", fileinfo.Size()), fmt.Sprintf("%x", utils.GenFingerPrint(path))), nil
|
||||
}
|
||||
}
|
||||
return rule.protocolRepo.ExecuteWriter(writers.RetrieveErrorRulePrefix), nil
|
||||
}
|
||||
|
||||
func (rule RetrieveFileRule) Match(data string) bool {
|
||||
return rule.matcher.Match(data)
|
||||
}
|
||||
Reference in New Issue
Block a user