mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-12-24 13:09:23 -05:00
175 lines
5.2 KiB
Go
175 lines
5.2 KiB
Go
|
// Copyright 2021 The Gitea Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a MIT-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package pypi
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"io"
|
||
|
"net/http"
|
||
|
"regexp"
|
||
|
"strings"
|
||
|
|
||
|
packages_model "code.gitea.io/gitea/models/packages"
|
||
|
"code.gitea.io/gitea/modules/context"
|
||
|
packages_module "code.gitea.io/gitea/modules/packages"
|
||
|
pypi_module "code.gitea.io/gitea/modules/packages/pypi"
|
||
|
"code.gitea.io/gitea/modules/setting"
|
||
|
"code.gitea.io/gitea/modules/templates"
|
||
|
"code.gitea.io/gitea/modules/validation"
|
||
|
"code.gitea.io/gitea/routers/api/packages/helper"
|
||
|
packages_service "code.gitea.io/gitea/services/packages"
|
||
|
)
|
||
|
|
||
|
// https://www.python.org/dev/peps/pep-0503/#normalized-names
|
||
|
var normalizer = strings.NewReplacer(".", "-", "_", "-")
|
||
|
var nameMatcher = regexp.MustCompile(`\A[a-z0-9\.\-_]+\z`)
|
||
|
|
||
|
// https://www.python.org/dev/peps/pep-0440/#appendix-b-parsing-version-strings-with-regular-expressions
|
||
|
var versionMatcher = regexp.MustCompile(`^([1-9][0-9]*!)?(0|[1-9][0-9]*)(\.(0|[1-9][0-9]*))*((a|b|rc)(0|[1-9][0-9]*))?(\.post(0|[1-9][0-9]*))?(\.dev(0|[1-9][0-9]*))?$`)
|
||
|
|
||
|
func apiError(ctx *context.Context, status int, obj interface{}) {
|
||
|
helper.LogAndProcessError(ctx, status, obj, func(message string) {
|
||
|
ctx.PlainText(status, message)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// PackageMetadata returns the metadata for a single package
|
||
|
func PackageMetadata(ctx *context.Context) {
|
||
|
packageName := normalizer.Replace(ctx.Params("id"))
|
||
|
|
||
|
pvs, err := packages_model.GetVersionsByPackageName(ctx, ctx.Package.Owner.ID, packages_model.TypePyPI, packageName)
|
||
|
if err != nil {
|
||
|
apiError(ctx, http.StatusInternalServerError, err)
|
||
|
return
|
||
|
}
|
||
|
if len(pvs) == 0 {
|
||
|
apiError(ctx, http.StatusNotFound, err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
pds, err := packages_model.GetPackageDescriptors(ctx, pvs)
|
||
|
if err != nil {
|
||
|
apiError(ctx, http.StatusInternalServerError, err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
ctx.Data["RegistryURL"] = setting.AppURL + "api/packages/" + ctx.Package.Owner.Name + "/pypi"
|
||
|
ctx.Data["PackageDescriptor"] = pds[0]
|
||
|
ctx.Data["PackageDescriptors"] = pds
|
||
|
ctx.Render = templates.HTMLRenderer()
|
||
|
ctx.HTML(http.StatusOK, "api/packages/pypi/simple")
|
||
|
}
|
||
|
|
||
|
// DownloadPackageFile serves the content of a package
|
||
|
func DownloadPackageFile(ctx *context.Context) {
|
||
|
packageName := normalizer.Replace(ctx.Params("id"))
|
||
|
packageVersion := ctx.Params("version")
|
||
|
filename := ctx.Params("filename")
|
||
|
|
||
|
s, pf, err := packages_service.GetFileStreamByPackageNameAndVersion(
|
||
|
ctx,
|
||
|
&packages_service.PackageInfo{
|
||
|
Owner: ctx.Package.Owner,
|
||
|
PackageType: packages_model.TypePyPI,
|
||
|
Name: packageName,
|
||
|
Version: packageVersion,
|
||
|
},
|
||
|
&packages_service.PackageFileInfo{
|
||
|
Filename: filename,
|
||
|
},
|
||
|
)
|
||
|
if err != nil {
|
||
|
if err == packages_model.ErrPackageNotExist || err == packages_model.ErrPackageFileNotExist {
|
||
|
apiError(ctx, http.StatusNotFound, err)
|
||
|
return
|
||
|
}
|
||
|
apiError(ctx, http.StatusInternalServerError, err)
|
||
|
return
|
||
|
}
|
||
|
defer s.Close()
|
||
|
|
||
|
ctx.ServeStream(s, pf.Name)
|
||
|
}
|
||
|
|
||
|
// UploadPackageFile adds a file to the package. If the package does not exist, it gets created.
|
||
|
func UploadPackageFile(ctx *context.Context) {
|
||
|
file, fileHeader, err := ctx.Req.FormFile("content")
|
||
|
if err != nil {
|
||
|
apiError(ctx, http.StatusBadRequest, err)
|
||
|
return
|
||
|
}
|
||
|
defer file.Close()
|
||
|
|
||
|
buf, err := packages_module.CreateHashedBufferFromReader(file, 32*1024*1024)
|
||
|
if err != nil {
|
||
|
apiError(ctx, http.StatusInternalServerError, err)
|
||
|
return
|
||
|
}
|
||
|
defer buf.Close()
|
||
|
|
||
|
_, _, hashSHA256, _ := buf.Sums()
|
||
|
|
||
|
if !strings.EqualFold(ctx.Req.FormValue("sha256_digest"), fmt.Sprintf("%x", hashSHA256)) {
|
||
|
apiError(ctx, http.StatusBadRequest, "hash mismatch")
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if _, err := buf.Seek(0, io.SeekStart); err != nil {
|
||
|
apiError(ctx, http.StatusInternalServerError, err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
packageName := normalizer.Replace(ctx.Req.FormValue("name"))
|
||
|
packageVersion := ctx.Req.FormValue("version")
|
||
|
if !nameMatcher.MatchString(packageName) || !versionMatcher.MatchString(packageVersion) {
|
||
|
apiError(ctx, http.StatusBadRequest, "invalid name or version")
|
||
|
return
|
||
|
}
|
||
|
|
||
|
projectURL := ctx.Req.FormValue("home_page")
|
||
|
if !validation.IsValidURL(projectURL) {
|
||
|
projectURL = ""
|
||
|
}
|
||
|
|
||
|
_, _, err = packages_service.CreatePackageOrAddFileToExisting(
|
||
|
&packages_service.PackageCreationInfo{
|
||
|
PackageInfo: packages_service.PackageInfo{
|
||
|
Owner: ctx.Package.Owner,
|
||
|
PackageType: packages_model.TypePyPI,
|
||
|
Name: packageName,
|
||
|
Version: packageVersion,
|
||
|
},
|
||
|
SemverCompatible: true,
|
||
|
Creator: ctx.Doer,
|
||
|
Metadata: &pypi_module.Metadata{
|
||
|
Author: ctx.Req.FormValue("author"),
|
||
|
Description: ctx.Req.FormValue("description"),
|
||
|
LongDescription: ctx.Req.FormValue("long_description"),
|
||
|
Summary: ctx.Req.FormValue("summary"),
|
||
|
ProjectURL: projectURL,
|
||
|
License: ctx.Req.FormValue("license"),
|
||
|
RequiresPython: ctx.Req.FormValue("requires_python"),
|
||
|
},
|
||
|
},
|
||
|
&packages_service.PackageFileCreationInfo{
|
||
|
PackageFileInfo: packages_service.PackageFileInfo{
|
||
|
Filename: fileHeader.Filename,
|
||
|
},
|
||
|
Data: buf,
|
||
|
IsLead: true,
|
||
|
},
|
||
|
)
|
||
|
if err != nil {
|
||
|
if err == packages_model.ErrDuplicatePackageFile {
|
||
|
apiError(ctx, http.StatusBadRequest, err)
|
||
|
return
|
||
|
}
|
||
|
apiError(ctx, http.StatusInternalServerError, err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
ctx.Status(http.StatusCreated)
|
||
|
}
|