86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/sha1"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
)
|
||
|
|
|
||
|
|
type EFile struct {
|
||
|
|
id int64
|
||
|
|
Path string
|
||
|
|
id_exercice int64
|
||
|
|
Name string
|
||
|
|
Checksum []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Exercice) GetFiles() ([]EFile, error) {
|
||
|
|
if rows, err := DBQuery("SELECT id_file, path, name, sha1 FROM exercice_files WHERE id_exercice = ?", e.id); err != nil {
|
||
|
|
return nil, err
|
||
|
|
} else {
|
||
|
|
defer rows.Close()
|
||
|
|
|
||
|
|
var files = make([]EFile, 0)
|
||
|
|
for rows.Next() {
|
||
|
|
var f EFile
|
||
|
|
f.id_exercice = e.id
|
||
|
|
if err := rows.Scan(&f.id, &f.Path, &f.Name, &f.Checksum); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
files = append(files, f)
|
||
|
|
}
|
||
|
|
if err := rows.Err(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return files, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Exercice) ImportFile(filePath string) (EFile, error) {
|
||
|
|
if fd, err := os.Open(filePath); err != nil {
|
||
|
|
return EFile{}, err
|
||
|
|
} else {
|
||
|
|
defer fd.Close()
|
||
|
|
|
||
|
|
hash := sha1.New()
|
||
|
|
if _, err := io.Copy(hash, fd); err != nil {
|
||
|
|
return EFile{}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
var result []byte
|
||
|
|
return e.AddFile(filePath, path.Base(filePath), hash.Sum(result))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e Exercice) AddFile(path string, name string, checksum []byte) (EFile, error) {
|
||
|
|
if res, err := DBExec("INSERT INTO exercice_files (id_exercice, path, name, sha1) VALUES (?, ?, ?, ?)", e.id, path, name, checksum); err != nil {
|
||
|
|
return EFile{}, err
|
||
|
|
} else if fid, err := res.LastInsertId(); err != nil {
|
||
|
|
return EFile{}, err
|
||
|
|
} else {
|
||
|
|
return EFile{fid, path, e.id, name, checksum}, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f EFile) Update() (int64, error) {
|
||
|
|
if res, err := DBExec("UPDATE exercice_files SET id_exercice = ?, path = ?, name = ?, sha1 = ? WHERE id_file = ?", f.id_exercice, f.Path, f.Name, f.Checksum, f.id); err != nil {
|
||
|
|
return 0, err
|
||
|
|
} else if nb, err := res.RowsAffected(); err != nil {
|
||
|
|
return 0, err
|
||
|
|
} else {
|
||
|
|
return nb, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f EFile) Delete() (int64, error) {
|
||
|
|
if res, err := DBExec("DELETE FROM exercice_files WHERE id_file = ?", f.id); err != nil {
|
||
|
|
return 0, err
|
||
|
|
} else if nb, err := res.RowsAffected(); err != nil {
|
||
|
|
return 0, err
|
||
|
|
} else {
|
||
|
|
return nb, err
|
||
|
|
}
|
||
|
|
}
|