func downloadFile(
wri http.ResponseWriter,
req *http.Request,
) {
var gae = appengine.NewContext(req)
var bucket = util.GetBucketName()
var vars = mux.Vars(req)
log.Infof(gae, "download file %s from bucket %s", vars["key"], bucket)
var cloudCtx = getCloudContext(gae)
rc, err := storage.NewReader(cloudCtx, bucket, vars["key"])
if err != nil {
log.Infof(gae, "unable to open file %q from bucket %q", vars["key"], bucket)
wri.WriteHeader(http.StatusNotFound)
return
}
defer rc.Close()
// Return an signed URL from which the client can download a file
accessIdFile := os.Getenv("GOOGLE_ACCESS_ID")
if accessIdFile == "" {
log.Errorf(gae, "Missing GOOGLE_ACCESS_ID from app.yaml")
wri.WriteHeader(http.StatusInternalServerError)
return
}
accessId, err := ioutil.ReadFile(accessIdFile)
if err != nil {
log.Errorf(gae, "Cannot read service account access id from %s", accessIdFile)
wri.WriteHeader(http.StatusInternalServerError)
}
privateKeyFile := os.Getenv("GOOGLE_PRIVATE_KEY")
if privateKeyFile == "" {
log.Errorf(gae, "Missing GOOGLE_PRIVATE_KEY from app.yaml")
wri.WriteHeader(http.StatusInternalServerError)
return
}
log.Infof(gae, "Private key file %s", privateKeyFile)
privateKey, err := ioutil.ReadFile(privateKeyFile)
if err != nil {
log.Errorf(gae, "Cannot read service account private key from %s", privateKeyFile)
wri.WriteHeader(http.StatusInternalServerError)
return
}
var opts = storage.SignedURLOptions{
GoogleAccessID: string(accessId),
PrivateKey: privateKey,
Method: "GET",
Expires: time.Now().UTC().Add(300 * time.Second),
}
signedURL, err := storage.SignedURL(bucket, vars["key"], &opts)
if err != nil {
log.Errorf(gae, "Unable to generate a signedURL: %s", err)
wri.WriteHeader(http.StatusInternalServerError)
}
http.Redirect(wri, req, signedURL, http.StatusTemporaryRedirect)
}
func uploadFile(
wri http.ResponseWriter,
req *http.Request,
) {
var gae = appengine.NewContext(req)
var bucket = util.GetBucketName()
log.Infof(gae, "uploading file to bucket %s", bucket)
cloudCtx := getCloudContext(gae)
var numOfBytes = 10
var randBytes = make([]byte, numOfBytes)
if _, err := rand.Read(randBytes); err != nil {
log.Errorf(gae, "crypto/rand.Read() error: %s", err)
wri.WriteHeader(http.StatusInternalServerError)
return
}
var filename = req.Header.Get("X-Gaefile-Filename")
log.Infof(gae, "before filename = %s", string(filename))
var randStr = base64.URLEncoding.EncodeToString(randBytes)
filename = randStr + "_" + filename
log.Infof(gae, "after filename = %s", string(filename))
cc := cloudBucketContext{
ctx: cloudCtx,
bucket: bucket,
filename: filename,
req: req,
}
if err := createFile(cc, gae); err != nil {
wri.WriteHeader(http.StatusInternalServerError)
log.Errorf(gae, "createFile error: %s", err)
return
} else {
var resp = createFileResponse{
Key: filename,
}
log.Infof(gae, "createFileResponse = %+v", resp)
var buf bytes.Buffer
if err := json.NewEncoder(&buf).Encode(&resp); err != nil {
wri.WriteHeader(http.StatusInternalServerError)
log.Errorf(gae, "createFileResponse error: %s", err)
return
}
if _, err := io.Copy(wri, &buf); err != nil {
wri.WriteHeader(http.StatusInternalServerError)
log.Errorf(gae, "createFileResponse io.Copy error: %s", err)
return
}
}
}