资源上传

package main

import (
  "encoding/json"
  "fmt"
  "io"
  "mime/multipart"
  "net/http"
  "os"
  "path/filepath"
)

type asset struct { URL string `json:"url"` }

func main() {
  filePath := "./reference.png"
  file, err := os.Open(filePath)
  if err != nil { panic(err) }
  defer file.Close()

  bodyReader, bodyWriter := io.Pipe()
  form := multipart.NewWriter(bodyWriter)
  go func() {
    part, err := form.CreateFormFile("file", filepath.Base(filePath))
    if err == nil { _, err = io.Copy(part, file) }
    if closeErr := form.Close(); err == nil { err = closeErr }
    _ = bodyWriter.CloseWithError(err)
  }()

  request, err := http.NewRequest(http.MethodPost, "https://api.fireflyfusion.ai/api/customer/v1/assets", bodyReader)
  if err != nil { panic(err) }
  request.Header.Set("Authorization", "Bearer "+os.Getenv("FIREFLY_API_KEY"))
  request.Header.Set("Content-Type", form.FormDataContentType())

  response, err := http.DefaultClient.Do(request)
  if err != nil { panic(err) }
  defer response.Body.Close()
  if response.StatusCode < 200 || response.StatusCode >= 300 {
    message, _ := io.ReadAll(response.Body)
    panic(string(message))
  }
  var result asset
  if err := json.NewDecoder(response.Body).Decode(&result); err != nil { panic(err) }
  fmt.Println(result.URL)
}