mirror of https://github.com/go-gitea/gitea
wxiaoguang
1 week ago
committed by
GitHub
commit
e07ddcb5d0
@ -0,0 +1,52 @@ |
|||||||
|
// Copyright 2024 The Gitea Authors. All rights reserved.
|
||||||
|
// SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
|
package v1_23 //nolint
|
||||||
|
|
||||||
|
import ( |
||||||
|
"code.gitea.io/gitea/modules/timeutil" |
||||||
|
|
||||||
|
"xorm.io/xorm" |
||||||
|
"xorm.io/xorm/schemas" |
||||||
|
) |
||||||
|
|
||||||
|
type improveActionTableIndicesAction struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
UserID int64 `xorm:"INDEX"` // Receiver user id.
|
||||||
|
OpType int |
||||||
|
ActUserID int64 // Action user id.
|
||||||
|
RepoID int64 |
||||||
|
CommentID int64 `xorm:"INDEX"` |
||||||
|
IsDeleted bool `xorm:"NOT NULL DEFAULT false"` |
||||||
|
RefName string |
||||||
|
IsPrivate bool `xorm:"NOT NULL DEFAULT false"` |
||||||
|
Content string `xorm:"TEXT"` |
||||||
|
CreatedUnix timeutil.TimeStamp `xorm:"created"` |
||||||
|
} |
||||||
|
|
||||||
|
// TableName sets the name of this table
|
||||||
|
func (*improveActionTableIndicesAction) TableName() string { |
||||||
|
return "action" |
||||||
|
} |
||||||
|
|
||||||
|
func (a *improveActionTableIndicesAction) TableIndices() []*schemas.Index { |
||||||
|
repoIndex := schemas.NewIndex("r_u_d", schemas.IndexType) |
||||||
|
repoIndex.AddColumn("repo_id", "user_id", "is_deleted") |
||||||
|
|
||||||
|
actUserIndex := schemas.NewIndex("au_r_c_u_d", schemas.IndexType) |
||||||
|
actUserIndex.AddColumn("act_user_id", "repo_id", "created_unix", "user_id", "is_deleted") |
||||||
|
|
||||||
|
cudIndex := schemas.NewIndex("c_u_d", schemas.IndexType) |
||||||
|
cudIndex.AddColumn("created_unix", "user_id", "is_deleted") |
||||||
|
|
||||||
|
cuIndex := schemas.NewIndex("c_u", schemas.IndexType) |
||||||
|
cuIndex.AddColumn("user_id", "is_deleted") |
||||||
|
|
||||||
|
indices := []*schemas.Index{actUserIndex, repoIndex, cudIndex, cuIndex} |
||||||
|
|
||||||
|
return indices |
||||||
|
} |
||||||
|
|
||||||
|
func AddNewIndexForUserDashboard(x *xorm.Engine) error { |
||||||
|
return x.Sync(new(improveActionTableIndicesAction)) |
||||||
|
} |
||||||
@ -1,48 +0,0 @@ |
|||||||
// Copyright 2022 The Gitea Authors. All rights reserved.
|
|
||||||
// SPDX-License-Identifier: MIT
|
|
||||||
|
|
||||||
package repository |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
|
|
||||||
"code.gitea.io/gitea/models/db" |
|
||||||
"code.gitea.io/gitea/models/perm" |
|
||||||
access_model "code.gitea.io/gitea/models/perm/access" |
|
||||||
repo_model "code.gitea.io/gitea/models/repo" |
|
||||||
user_model "code.gitea.io/gitea/models/user" |
|
||||||
|
|
||||||
"xorm.io/builder" |
|
||||||
) |
|
||||||
|
|
||||||
func AddCollaborator(ctx context.Context, repo *repo_model.Repository, u *user_model.User) error { |
|
||||||
if err := repo.LoadOwner(ctx); err != nil { |
|
||||||
return err |
|
||||||
} |
|
||||||
|
|
||||||
if user_model.IsUserBlockedBy(ctx, u, repo.OwnerID) || user_model.IsUserBlockedBy(ctx, repo.Owner, u.ID) { |
|
||||||
return user_model.ErrBlockedUser |
|
||||||
} |
|
||||||
|
|
||||||
return db.WithTx(ctx, func(ctx context.Context) error { |
|
||||||
has, err := db.Exist[repo_model.Collaboration](ctx, builder.Eq{ |
|
||||||
"repo_id": repo.ID, |
|
||||||
"user_id": u.ID, |
|
||||||
}) |
|
||||||
if err != nil { |
|
||||||
return err |
|
||||||
} else if has { |
|
||||||
return nil |
|
||||||
} |
|
||||||
|
|
||||||
if err = db.Insert(ctx, &repo_model.Collaboration{ |
|
||||||
RepoID: repo.ID, |
|
||||||
UserID: u.ID, |
|
||||||
Mode: perm.AccessModeWrite, |
|
||||||
}); err != nil { |
|
||||||
return err |
|
||||||
} |
|
||||||
|
|
||||||
return access_model.RecalculateUserAccess(ctx, repo, u.ID) |
|
||||||
}) |
|
||||||
} |
|
||||||
@ -1,280 +0,0 @@ |
|||||||
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
||||||
// SPDX-License-Identifier: MIT
|
|
||||||
|
|
||||||
package repository |
|
||||||
|
|
||||||
import ( |
|
||||||
"testing" |
|
||||||
|
|
||||||
"code.gitea.io/gitea/models/db" |
|
||||||
"code.gitea.io/gitea/models/organization" |
|
||||||
perm_model "code.gitea.io/gitea/models/perm" |
|
||||||
access_model "code.gitea.io/gitea/models/perm/access" |
|
||||||
repo_model "code.gitea.io/gitea/models/repo" |
|
||||||
"code.gitea.io/gitea/models/unit" |
|
||||||
"code.gitea.io/gitea/models/unittest" |
|
||||||
user_model "code.gitea.io/gitea/models/user" |
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert" |
|
||||||
) |
|
||||||
|
|
||||||
func TestRepository_AddCollaborator(t *testing.T) { |
|
||||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
||||||
|
|
||||||
testSuccess := func(repoID, userID int64) { |
|
||||||
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: repoID}) |
|
||||||
assert.NoError(t, repo.LoadOwner(db.DefaultContext)) |
|
||||||
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: userID}) |
|
||||||
assert.NoError(t, AddCollaborator(db.DefaultContext, repo, user)) |
|
||||||
unittest.CheckConsistencyFor(t, &repo_model.Repository{ID: repoID}, &user_model.User{ID: userID}) |
|
||||||
} |
|
||||||
testSuccess(1, 4) |
|
||||||
testSuccess(1, 4) |
|
||||||
testSuccess(3, 4) |
|
||||||
} |
|
||||||
|
|
||||||
func TestRepoPermissionPublicNonOrgRepo(t *testing.T) { |
|
||||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
||||||
|
|
||||||
// public non-organization repo
|
|
||||||
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 4}) |
|
||||||
assert.NoError(t, repo.LoadUnits(db.DefaultContext)) |
|
||||||
|
|
||||||
// plain user
|
|
||||||
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) |
|
||||||
perm, err := access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.False(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// change to collaborator
|
|
||||||
assert.NoError(t, AddCollaborator(db.DefaultContext, repo, user)) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// collaborator
|
|
||||||
collaborator := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 4}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, collaborator) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// owner
|
|
||||||
owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 5}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, owner) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// admin
|
|
||||||
admin := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, admin) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
func TestRepoPermissionPrivateNonOrgRepo(t *testing.T) { |
|
||||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
||||||
|
|
||||||
// private non-organization repo
|
|
||||||
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 2}) |
|
||||||
assert.NoError(t, repo.LoadUnits(db.DefaultContext)) |
|
||||||
|
|
||||||
// plain user
|
|
||||||
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 4}) |
|
||||||
perm, err := access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.False(t, perm.CanRead(unit.Type)) |
|
||||||
assert.False(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// change to collaborator to default write access
|
|
||||||
assert.NoError(t, AddCollaborator(db.DefaultContext, repo, user)) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
assert.NoError(t, repo_model.ChangeCollaborationAccessMode(db.DefaultContext, repo, user.ID, perm_model.AccessModeRead)) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.False(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// owner
|
|
||||||
owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, owner) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// admin
|
|
||||||
admin := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, admin) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
func TestRepoPermissionPublicOrgRepo(t *testing.T) { |
|
||||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
||||||
|
|
||||||
// public organization repo
|
|
||||||
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 32}) |
|
||||||
assert.NoError(t, repo.LoadUnits(db.DefaultContext)) |
|
||||||
|
|
||||||
// plain user
|
|
||||||
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 5}) |
|
||||||
perm, err := access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.False(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// change to collaborator to default write access
|
|
||||||
assert.NoError(t, AddCollaborator(db.DefaultContext, repo, user)) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
assert.NoError(t, repo_model.ChangeCollaborationAccessMode(db.DefaultContext, repo, user.ID, perm_model.AccessModeRead)) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.False(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// org member team owner
|
|
||||||
owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, owner) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// org member team tester
|
|
||||||
member := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 15}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, member) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
} |
|
||||||
assert.True(t, perm.CanWrite(unit.TypeIssues)) |
|
||||||
assert.False(t, perm.CanWrite(unit.TypeCode)) |
|
||||||
|
|
||||||
// admin
|
|
||||||
admin := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, admin) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
func TestRepoPermissionPrivateOrgRepo(t *testing.T) { |
|
||||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
||||||
|
|
||||||
// private organization repo
|
|
||||||
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 24}) |
|
||||||
assert.NoError(t, repo.LoadUnits(db.DefaultContext)) |
|
||||||
|
|
||||||
// plain user
|
|
||||||
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 5}) |
|
||||||
perm, err := access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.False(t, perm.CanRead(unit.Type)) |
|
||||||
assert.False(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// change to collaborator to default write access
|
|
||||||
assert.NoError(t, AddCollaborator(db.DefaultContext, repo, user)) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
assert.NoError(t, repo_model.ChangeCollaborationAccessMode(db.DefaultContext, repo, user.ID, perm_model.AccessModeRead)) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, user) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.False(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// org member team owner
|
|
||||||
owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 15}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, owner) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// update team information and then check permission
|
|
||||||
team := unittest.AssertExistsAndLoadBean(t, &organization.Team{ID: 5}) |
|
||||||
err = organization.UpdateTeamUnits(db.DefaultContext, team, nil) |
|
||||||
assert.NoError(t, err) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, owner) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
|
|
||||||
// org member team tester
|
|
||||||
tester := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, tester) |
|
||||||
assert.NoError(t, err) |
|
||||||
assert.True(t, perm.CanWrite(unit.TypeIssues)) |
|
||||||
assert.False(t, perm.CanWrite(unit.TypeCode)) |
|
||||||
assert.False(t, perm.CanRead(unit.TypeCode)) |
|
||||||
|
|
||||||
// org member team reviewer
|
|
||||||
reviewer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 20}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, reviewer) |
|
||||||
assert.NoError(t, err) |
|
||||||
assert.False(t, perm.CanRead(unit.TypeIssues)) |
|
||||||
assert.False(t, perm.CanWrite(unit.TypeCode)) |
|
||||||
assert.True(t, perm.CanRead(unit.TypeCode)) |
|
||||||
|
|
||||||
// admin
|
|
||||||
admin := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) |
|
||||||
perm, err = access_model.GetUserRepoPermission(db.DefaultContext, repo, admin) |
|
||||||
assert.NoError(t, err) |
|
||||||
for _, unit := range repo.Units { |
|
||||||
assert.True(t, perm.CanRead(unit.Type)) |
|
||||||
assert.True(t, perm.CanWrite(unit.Type)) |
|
||||||
} |
|
||||||
} |
|
||||||
Binary file not shown.
@ -1,2 +1,2 @@ |
|||||||
P pack-393dc29256bc27cb2ec73898507df710be7a3cf5.pack |
P pack-a7bef76cf6e2b46bc816936ab69306fb10aea571.pack |
||||||
|
|
||||||
|
|||||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@ -0,0 +1,40 @@ |
|||||||
|
// Copyright 2024 The Gitea Authors. All rights reserved.
|
||||||
|
// SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
|
package integration |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"net/http" |
||||||
|
"testing" |
||||||
|
|
||||||
|
auth_model "code.gitea.io/gitea/models/auth" |
||||||
|
repo_model "code.gitea.io/gitea/models/repo" |
||||||
|
"code.gitea.io/gitea/models/unittest" |
||||||
|
user_model "code.gitea.io/gitea/models/user" |
||||||
|
"code.gitea.io/gitea/modules/setting" |
||||||
|
"code.gitea.io/gitea/modules/test" |
||||||
|
"code.gitea.io/gitea/tests" |
||||||
|
) |
||||||
|
|
||||||
|
func TestAPIEditReleaseAttachmentWithUnallowedFile(t *testing.T) { |
||||||
|
// Limit the allowed release types (since by default there is no restriction)
|
||||||
|
defer test.MockVariableValue(&setting.Repository.Release.AllowedTypes, ".exe")() |
||||||
|
defer tests.PrepareTestEnv(t)() |
||||||
|
|
||||||
|
attachment := unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: 9}) |
||||||
|
release := unittest.AssertExistsAndLoadBean(t, &repo_model.Release{ID: attachment.ReleaseID}) |
||||||
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: attachment.RepoID}) |
||||||
|
repoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID}) |
||||||
|
|
||||||
|
session := loginUser(t, repoOwner.Name) |
||||||
|
token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteRepository) |
||||||
|
|
||||||
|
filename := "file.bad" |
||||||
|
urlStr := fmt.Sprintf("/api/v1/repos/%s/%s/releases/%d/assets/%d", repoOwner.Name, repo.Name, release.ID, attachment.ID) |
||||||
|
req := NewRequestWithValues(t, "PATCH", urlStr, map[string]string{ |
||||||
|
"name": filename, |
||||||
|
}).AddTokenAuth(token) |
||||||
|
|
||||||
|
session.MakeRequest(t, req, http.StatusUnprocessableEntity) |
||||||
|
} |
||||||
@ -0,0 +1,24 @@ |
|||||||
|
import {convertHtmlToMarkdown} from './html2markdown.ts'; |
||||||
|
import {createElementFromHTML} from '../utils/dom.ts'; |
||||||
|
|
||||||
|
const h = createElementFromHTML; |
||||||
|
|
||||||
|
test('convertHtmlToMarkdown', () => { |
||||||
|
expect(convertHtmlToMarkdown(h(`<h1>h</h1>`))).toBe('# h'); |
||||||
|
expect(convertHtmlToMarkdown(h(`<strong>txt</strong>`))).toBe('**txt**'); |
||||||
|
expect(convertHtmlToMarkdown(h(`<em>txt</em>`))).toBe('_txt_'); |
||||||
|
expect(convertHtmlToMarkdown(h(`<del>txt</del>`))).toBe('~~txt~~'); |
||||||
|
|
||||||
|
expect(convertHtmlToMarkdown(h(`<a href="link">txt</a>`))).toBe('[txt](link)'); |
||||||
|
expect(convertHtmlToMarkdown(h(`<a href="https://link">https://link</a>`))).toBe('https://link'); |
||||||
|
|
||||||
|
expect(convertHtmlToMarkdown(h(`<img src="link">`))).toBe(''); |
||||||
|
expect(convertHtmlToMarkdown(h(`<img src="link" alt="name">`))).toBe(''); |
||||||
|
expect(convertHtmlToMarkdown(h(`<img src="link" width="1" height="1">`))).toBe('<img alt="image" width="1" height="1" src="link">'); |
||||||
|
|
||||||
|
expect(convertHtmlToMarkdown(h(`<p>txt</p>`))).toBe('txt\n'); |
||||||
|
expect(convertHtmlToMarkdown(h(`<blockquote>a\nb</blockquote>`))).toBe('> a\n> b\n'); |
||||||
|
|
||||||
|
expect(convertHtmlToMarkdown(h(`<ol><li>a<ul><li>b</li></ul></li></ol>`))).toBe('1. a\n * b\n\n'); |
||||||
|
expect(convertHtmlToMarkdown(h(`<ol><li><input checked>a</li></ol>`))).toBe('1. [x] a\n'); |
||||||
|
}); |
||||||
@ -0,0 +1,119 @@ |
|||||||
|
import {htmlEscape} from 'escape-goat'; |
||||||
|
|
||||||
|
type Processors = { |
||||||
|
[tagName: string]: (el: HTMLElement) => string | HTMLElement | void; |
||||||
|
} |
||||||
|
|
||||||
|
type ProcessorContext = { |
||||||
|
elementIsFirst: boolean; |
||||||
|
elementIsLast: boolean; |
||||||
|
listNestingLevel: number; |
||||||
|
} |
||||||
|
|
||||||
|
function prepareProcessors(ctx:ProcessorContext): Processors { |
||||||
|
const processors = { |
||||||
|
H1(el) { |
||||||
|
const level = parseInt(el.tagName.slice(1)); |
||||||
|
el.textContent = `${'#'.repeat(level)} ${el.textContent.trim()}`; |
||||||
|
}, |
||||||
|
STRONG(el) { |
||||||
|
return `**${el.textContent}**`; |
||||||
|
}, |
||||||
|
EM(el) { |
||||||
|
return `_${el.textContent}_`; |
||||||
|
}, |
||||||
|
DEL(el) { |
||||||
|
return `~~${el.textContent}~~`; |
||||||
|
}, |
||||||
|
|
||||||
|
A(el) { |
||||||
|
const text = el.textContent || 'link'; |
||||||
|
const href = el.getAttribute('href'); |
||||||
|
if (/^https?:/.test(text) && text === href) { |
||||||
|
return text; |
||||||
|
} |
||||||
|
return href ? `[${text}](${href})` : text; |
||||||
|
}, |
||||||
|
IMG(el) { |
||||||
|
const alt = el.getAttribute('alt') || 'image'; |
||||||
|
const src = el.getAttribute('src'); |
||||||
|
const widthAttr = el.hasAttribute('width') ? ` width="${htmlEscape(el.getAttribute('width') || '')}"` : ''; |
||||||
|
const heightAttr = el.hasAttribute('height') ? ` height="${htmlEscape(el.getAttribute('height') || '')}"` : ''; |
||||||
|
if (widthAttr || heightAttr) { |
||||||
|
return `<img alt="${htmlEscape(alt)}"${widthAttr}${heightAttr} src="${htmlEscape(src)}">`; |
||||||
|
} |
||||||
|
return ``; |
||||||
|
}, |
||||||
|
|
||||||
|
P(el) { |
||||||
|
el.textContent = `${el.textContent}\n`; |
||||||
|
}, |
||||||
|
BLOCKQUOTE(el) { |
||||||
|
el.textContent = `${el.textContent.replace(/^/mg, '> ')}\n`; |
||||||
|
}, |
||||||
|
|
||||||
|
OL(el) { |
||||||
|
const preNewLine = ctx.listNestingLevel ? '\n' : ''; |
||||||
|
el.textContent = `${preNewLine}${el.textContent}\n`; |
||||||
|
}, |
||||||
|
LI(el) { |
||||||
|
const parent = el.parentNode; |
||||||
|
const bullet = parent.tagName === 'OL' ? `1. ` : '* '; |
||||||
|
const nestingIdentLevel = Math.max(0, ctx.listNestingLevel - 1); |
||||||
|
el.textContent = `${' '.repeat(nestingIdentLevel * 4)}${bullet}${el.textContent}${ctx.elementIsLast ? '' : '\n'}`; |
||||||
|
return el; |
||||||
|
}, |
||||||
|
INPUT(el) { |
||||||
|
return el.checked ? '[x] ' : '[ ] '; |
||||||
|
}, |
||||||
|
|
||||||
|
CODE(el) { |
||||||
|
const text = el.textContent; |
||||||
|
if (el.parentNode && el.parentNode.tagName === 'PRE') { |
||||||
|
el.textContent = `\`\`\`\n${text}\n\`\`\`\n`; |
||||||
|
return el; |
||||||
|
} |
||||||
|
if (text.includes('`')) { |
||||||
|
return `\`\` ${text} \`\``; |
||||||
|
} |
||||||
|
return `\`${text}\``; |
||||||
|
}, |
||||||
|
}; |
||||||
|
processors['UL'] = processors.OL; |
||||||
|
for (let level = 2; level <= 6; level++) { |
||||||
|
processors[`H${level}`] = processors.H1; |
||||||
|
} |
||||||
|
return processors; |
||||||
|
} |
||||||
|
|
||||||
|
function processElement(ctx :ProcessorContext, processors: Processors, el: HTMLElement) { |
||||||
|
if (el.hasAttribute('data-markdown-generated-content')) return el.textContent; |
||||||
|
if (el.tagName === 'A' && el.children.length === 1 && el.children[0].tagName === 'IMG') { |
||||||
|
return processElement(ctx, processors, el.children[0] as HTMLElement); |
||||||
|
} |
||||||
|
|
||||||
|
const isListContainer = el.tagName === 'OL' || el.tagName === 'UL'; |
||||||
|
if (isListContainer) ctx.listNestingLevel++; |
||||||
|
for (let i = 0; i < el.children.length; i++) { |
||||||
|
ctx.elementIsFirst = i === 0; |
||||||
|
ctx.elementIsLast = i === el.children.length - 1; |
||||||
|
processElement(ctx, processors, el.children[i] as HTMLElement); |
||||||
|
} |
||||||
|
if (isListContainer) ctx.listNestingLevel--; |
||||||
|
|
||||||
|
if (processors[el.tagName]) { |
||||||
|
const ret = processors[el.tagName](el); |
||||||
|
if (ret && ret !== el) { |
||||||
|
el.replaceWith(typeof ret === 'string' ? document.createTextNode(ret) : ret); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
export function convertHtmlToMarkdown(el: HTMLElement): string { |
||||||
|
const div = document.createElement('div'); |
||||||
|
div.append(el); |
||||||
|
const ctx = {} as ProcessorContext; |
||||||
|
ctx.listNestingLevel = 0; |
||||||
|
processElement(ctx, prepareProcessors(ctx), el); |
||||||
|
return div.textContent; |
||||||
|
} |
||||||
Loading…
Reference in new issue