61 lines
1.9 KiB
Go
61 lines
1.9 KiB
Go
|
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)
|
||
|
|
}
|