mirror of
https://codeberg.org/scip/ephemerup.git
synced 2025-12-17 20:50:56 +01:00
102 lines
2.3 KiB
Go
102 lines
2.3 KiB
Go
/*
|
|
Copyright © 2023 Thomas von Dein
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
bolt "go.etcd.io/bbolt"
|
|
)
|
|
|
|
const Bucket string = "uploads"
|
|
|
|
func DbInsert(db *bolt.DB, id string, entry *Upload) {
|
|
err := db.Update(func(tx *bolt.Tx) error {
|
|
bucket, err := tx.CreateBucketIfNotExists([]byte(Bucket))
|
|
if err != nil {
|
|
return fmt.Errorf("create bucket: %s", err)
|
|
}
|
|
|
|
jsonentry, err := json.Marshal(entry)
|
|
if err != nil {
|
|
return fmt.Errorf("json marshalling failure: %s", err)
|
|
}
|
|
|
|
err = bucket.Put([]byte(id), []byte(jsonentry))
|
|
if err != nil {
|
|
return fmt.Errorf("insert data: %s", err)
|
|
}
|
|
|
|
// results in:
|
|
// bbolt get /tmp/uploads.db uploads fb242922-86cb-43a8-92bc-b027c35f0586
|
|
// {"id":"fb242922-86cb-43a8-92bc-b027c35f0586","expire":"1d","file":"2023-02-17-13-09-data.zip"}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
Log("DB error: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
func DbLookupId(db *bolt.DB, id string) (Upload, error) {
|
|
var upload Upload
|
|
|
|
err := db.View(func(tx *bolt.Tx) error {
|
|
bucket := tx.Bucket([]byte(Bucket))
|
|
j := bucket.Get([]byte(id))
|
|
|
|
if len(j) == 0 {
|
|
return fmt.Errorf("id %s not found", id)
|
|
}
|
|
|
|
if err := json.Unmarshal(j, &upload); err != nil {
|
|
return fmt.Errorf("unable to unmarshal json: %s", err)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
Log("DB error: %s", err.Error())
|
|
return upload, err
|
|
}
|
|
|
|
return upload, nil
|
|
}
|
|
|
|
func DbDeleteId(db *bolt.DB, id string) error {
|
|
err := db.Update(func(tx *bolt.Tx) error {
|
|
bucket := tx.Bucket([]byte(Bucket))
|
|
|
|
j := bucket.Get([]byte(id))
|
|
|
|
if len(j) == 0 {
|
|
return fmt.Errorf("id %s not found", id)
|
|
}
|
|
|
|
err := bucket.Delete([]byte(id))
|
|
return err
|
|
})
|
|
|
|
if err != nil {
|
|
Log("DB error: %s", err.Error())
|
|
}
|
|
|
|
return err
|
|
}
|