mirror of
https://github.com/go-gitea/gitea.git
synced 2025-07-04 22:57:34 -04:00
56e42be36d
Adds the `flat-square` style to action badges. Styles can be selected by adding `?style=<style>` to the badge endpoint. If no style query is given, or if the query is invalid, the style defaults to `flat`. --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
60 lines
1.8 KiB
Go
60 lines
1.8 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package actions
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
actions_model "code.gitea.io/gitea/models/actions"
|
|
"code.gitea.io/gitea/modules/badge"
|
|
"code.gitea.io/gitea/modules/git"
|
|
"code.gitea.io/gitea/modules/util"
|
|
"code.gitea.io/gitea/services/context"
|
|
)
|
|
|
|
func GetWorkflowBadge(ctx *context.Context) {
|
|
workflowFile := ctx.PathParam("workflow_name")
|
|
branch := ctx.FormString("branch", ctx.Repo.Repository.DefaultBranch)
|
|
event := ctx.FormString("event")
|
|
style := ctx.FormString("style")
|
|
|
|
branchRef := git.RefNameFromBranch(branch)
|
|
b, err := getWorkflowBadge(ctx, workflowFile, branchRef.String(), event)
|
|
if err != nil {
|
|
ctx.ServerError("GetWorkflowBadge", err)
|
|
return
|
|
}
|
|
|
|
ctx.Data["Badge"] = b
|
|
ctx.RespHeader().Set("Content-Type", "image/svg+xml")
|
|
switch style {
|
|
case badge.StyleFlatSquare:
|
|
ctx.HTML(http.StatusOK, "shared/actions/runner_badge_flat-square")
|
|
default: // defaults to badge.StyleFlat
|
|
ctx.HTML(http.StatusOK, "shared/actions/runner_badge_flat")
|
|
}
|
|
}
|
|
|
|
func getWorkflowBadge(ctx *context.Context, workflowFile, branchName, event string) (badge.Badge, error) {
|
|
extension := filepath.Ext(workflowFile)
|
|
workflowName := strings.TrimSuffix(workflowFile, extension)
|
|
|
|
run, err := actions_model.GetWorkflowLatestRun(ctx, ctx.Repo.Repository.ID, workflowFile, branchName, event)
|
|
if err != nil {
|
|
if errors.Is(err, util.ErrNotExist) {
|
|
return badge.GenerateBadge(workflowName, "no status", badge.DefaultColor), nil
|
|
}
|
|
return badge.Badge{}, err
|
|
}
|
|
|
|
color, ok := badge.GlobalVars().StatusColorMap[run.Status]
|
|
if !ok {
|
|
return badge.GenerateBadge(workflowName, "unknown status", badge.DefaultColor), nil
|
|
}
|
|
return badge.GenerateBadge(workflowName, run.Status.String(), color), nil
|
|
}
|