简单实现AI音乐suno-api

前言

在科技与艺术的交汇处，AI音乐创作正以其独特的魅力，引领着音乐产业的一次革命。不久前，AI音乐的浪潮席卷了整个创意领域，激发了无数音乐爱好者和技术开发者的无限想象。在这场音乐与科技的盛宴中，主流的AI音乐平台suno无疑成为了焦点，尽管它尚未对外开放API服务，但这并未阻止我们探索的脚步。

今天，我们将踏上一段奇妙的旅程，用Go语言这把精准而强大的工具，尝试构建一个简易的suno-api。

开发前准备

1. golang开发环境。
2. golang版本：1.21.0
3. 可登录suno的环境
4. 获取suno平台cookie

开发过程

1.安装gin框架

go get -u github.com/gin-gonic/gin

2.封装suno请求

1.相关结构体

printf("hello world!");package internal



import "time"



// GenerateReq generate-Req

type GenerateReq struct {

    GptDescriptionPrompt string `json:"gpt_description_prompt"`

    Prompt               string `json:"prompt"`

    Mv                   string `json:"mv"`

    Title                string `json:"title"`

    Tags                 string `json:"tags"`

}



// GenerateResp generate-Resp

type GenerateResp struct {

    BatchSize         int       `json:"batch_size"`

    Clips             []Clips   `json:"clips"`

    CreatedAt         time.Time `json:"created_at"`

    ID                string    `json:"id"`

    Status            string    `json:"status"`

    Metadata          `json:"metadata"`

    MajorModelVersion string `json:"major_model_version"`

}



// TokenResp token-Response

type TokenResp struct {

    Jwt    string

    Object string

}



// SidResp session-resp

type SidResp struct {

    Response struct {

       Object              string      `json:"object"`

       ID                  string      `json:"id"`

       Sessions            []Session   `json:"sessions"`

       SignIn              interface{} `json:"sign_in"`

       SignUp              interface{} `json:"sign_up"`

       LastActiveSessionID string      `json:"last_active_session_id"`

       CreatedAt           time.Time   `json:"created_at"`

       UpdatedAt           time.Time   `json:"updated_at"`

    } `json:"response"`

    Client interface{} `json:"client"`

}



// Clips clips

type Clips struct {

    Detail            string      `json:"detail"`

    Id                string      `json:"id"`

    VideoUrl          string      `json:"video_url"`

    AudioUrl          string      `json:"audio_url"`

    ImageUrl          string      `json:"image_url"`

    ImageLargeUrl     string      `json:"image_large_url"`

    MajorModelVersion string      `json:"major_model_version"`

    ModelName         string      `json:"model_name"`

    Metadata          *Metadata   `json:"metadata"`

    IsLiked           bool        `json:"is_liked"`

    UserId            string      `json:"user_id"`

    IsTrashed         bool        `json:"is_trashed"`

    Reaction          interface{} `json:"reaction"`

    CreatedAt         time.Time   `json:"created_at"`

    Status            string      `json:"status"`

    Title             string      `json:"title"`

    PlayCount         int         `json:"play_count"`

    UpvoteCount       int         `json:"upvote_count"`

    IsPublic          bool        `json:"is_public"`

}



// Metadata Metadata

type Metadata struct {

    Tags                 string      `json:"tags"`

    Prompt               string      `json:"prompt"`

    GptDescriptionPrompt string      `json:"gpt_description_prompt"`

    AudioPromptId        interface{} `json:"audio_prompt_id"`

    History              interface{} `json:"history"`

    ConcatHistory        interface{} `json:"concat_history"`

    Type                 string      `json:"type"`

    Duration             float64     `json:"duration"`

    RefundCredits        bool        `json:"refund_credits"`

    Stream               bool        `json:"stream"`

    ErrorType            interface{} `json:"error_type"`

    ErrorMessage         interface{} `json:"error_message"`

}



// Session sessionId

type Session struct {

    Object string `json:"object"`

    ID     string `json:"id"`

}

2.获取suno的token

实现逻辑为通过suno的cookie获取到sessionid，后通过sessionid获取token.

// GetToken getToken

func (s *service) GetToken(cookieString string) (token string, err error) {

   var cookies []*http.Cookie

   cookies = s.parseCookieString(cookieString)



   jar, _ := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})

   sidUrl, _ := url.Parse(GetSidUrl)

   jar.SetCookies(sidUrl, cookies)



   client := &http.Client{

      Jar: jar,

   }



   sidResp, err := client.Get(sidUrl.String())

   if err != nil {

      return "", err

   }

   defer sidResp.Body.Close()



   var sidResponse SidResp

   _ = json.NewDecoder(sidResp.Body).Decode(&sidResponse)



   sid := ""

   if len(sidResponse.Response.Sessions) > 0 {

      sid = sidResponse.Response.Sessions[0].ID

      fmt.Println(sidResponse.Response.Sessions[0].ID)

   } else {

      err = errors.New("获取sessionId失败:Response.Sessions为空")

      return "", err

   }



   getTokenUrl := fmt.Sprintf("%s/%s/tokens?_clerk_js_version=4.72.0-snapshot.vc141245", GetTokenUrl, sid)

   tokenResp, err := client.Post(getTokenUrl, ContentType, nil)

   if err != nil {

      return "", err

   }

   defer tokenResp.Body.Close()



   var tokenResponse TokenResp

   bodyBytes, err := io.ReadAll(tokenResp.Body)

   if err != nil {

      return "", fmt.Errorf("error reading response body: %v", err)

   }



   if tokenResp.StatusCode != http.StatusOK {

      fmt.Printf("Error decoding JSON response: %v\nResponse body: %s\n", err, bodyBytes)

      fmt.Printf("请求失败，状态码: %d, 响应内容: %s\n", tokenResp.StatusCode, bodyBytes)

      return "", errors.New(fmt.Sprintf("请求失败，状态码: %d, 响应内容: %s\n", tokenResp.StatusCode, bodyBytes))

   }



   err = json.Unmarshal(bodyBytes, &tokenResponse)

   if err != nil {

      return "", err

   }



   if tokenResponse.Jwt == "" {

      return "", errors.New("获取token失败:token为空")

   }



   return tokenResponse.Jwt, nil

}



// parseCookieString 格式化cookie

func (s *service) parseCookieString(cookieString string) []*http.Cookie {

   var cookies []*http.Cookie

   for _, cookiePair := range strings.Split(cookieString, "; ") {

      parts := strings.Split(cookiePair, "=")

      if len(parts) == 2 {

         cookies = append(cookies, &http.Cookie{Name: parts[0], Value: parts[1]})

      }

   }

   return cookies

}

3.生成歌曲提交

// Generate 发起生成请求

func (s *service) Generate(prompt string) (response GenerateResp, err error) {

   // token校验

   if err = s.checkToken(); err != nil {

      return

   }



   // 携带token发起请求

   genReq := GenerateReq{

      GptDescriptionPrompt: prompt,

      Mv:                   "chirp-v3-0",

   }

   reqData, _ := json.Marshal(genReq)

   genMusicUrl := BaseUrl + "/api/generate/v2/"

   req, _ := http.NewRequest("POST", genMusicUrl, bytes.NewBuffer(reqData))

   req.Header.Set("Authorization", "Bearer "+Token)

   req.Header.Set("Content-Type", "application/json")



   client := &http.Client{}

   resp, err := client.Do(req)

   if err != nil {

      return

   }



   defer resp.Body.Close()



   var genResponse GenerateResp

   err = json.NewDecoder(resp.Body).Decode(&genResponse)

   if err != nil {

      return

   }



   return genResponse, nil

}

4.查询歌曲详情

// GetFeed 查询详情

func (s *service) GetFeed(ids []string) (result []Clips, err error) {

   // token校验

   if err = s.checkToken(); err != nil {

      return

   }



   idsFormat := url.PathEscape(strings.Join(ids, ","))

   feedUrl := fmt.Sprintf("%s/api/feed/?ids=%s", BaseUrl, idsFormat)

   req, _ := http.NewRequest("GET", feedUrl, nil)

   req.Header.Set("Authorization", "Bearer "+Token)

   req.Header.Set("Content-Type", "application/json")



   client := &http.Client{

      Timeout: 10 * time.Second,

   }



   resp, err := client.Do(req)

   if err != nil {

      return nil, err

   }

   defer resp.Body.Close()



   bodyBytes, err := io.ReadAll(resp.Body)

   if err != nil {

      return nil, fmt.Errorf("error reading response body: %v", err)

   }



   err = json.Unmarshal(bodyBytes, &result)

   if err != nil {

      fmt.Printf("Error decoding JSON response: %v\nResponse body: %s\n", err, bodyBytes)

      return nil, err

   }

   return result, nil

}

3.控制器

// GenMusic 提交音乐请求

func (h *Handler) GenMusic(c *gin.Context) {

   var genReq genReq

   if err := c.ShouldBindJSON(&genReq); err != nil {

      errResponse(c, err.Error(), "Params Fail")

      return

   }



   token, err := SunoService.GetToken(cookieString)

   if err != nil {

      errResponse(c, err.Error(), "Request Failed")

      return

   }



   SunoService.SetToken(token)



   res, err := SunoService.Generate(genReq.Prompt)

   if err != nil {

      errResponse(c, err.Error(), "Request Failed")

      return

   }



   okResponse(c, res)

   return

}



// GetFeed 查询详情

func (h *Handler) GetFeed(c *gin.Context) {

   var feedReq feedReq

   if err := c.ShouldBindJSON(&feedReq); err != nil {

      errResponse(c, err.Error(), "Params Fail")

      return

   }



   token, err := SunoService.GetToken(cookieString)

   if err != nil {

      errResponse(c, err.Error(), "Request Failed")

      return

   }

   SunoService.SetToken(token)



   data, err := SunoService.GetFeed(feedReq.ClipsIds)

   if err != nil {

      errResponse(c, err.Error(), "Request Failed")

      return

   }

   okResponse(c, data)

   return

}

postman测试

项目启动

cd cmd

go run main.go

生成音乐

查询音乐详情

结语

虽然目前这个suno-api仅揭开了其神秘的面纱，实现了基础的生成及详情查询功能，但它所蕴含的潜力与未来的发展空间却是无限的。如果我们能够为这个API增添更多的功能和优化，它将变得更加强大和完善。比如，通过配置文件来管理cookie，让它们更加安全和持久；增加cookie的保活机制，让用户体验更加流畅；记录每一次的请求日志，让问题追踪和系统监控变得更加简单；实施接口限流策略，保障系统在高并发情况下的稳定性等等。希望后续能够激发更多的创意火花，创造出更多动人心弦的旋律。

感谢这段旅程有你的陪伴，让我们一起期待下一次更加精彩的启程。

文章转自微信公众号@AI创业之路