Move wiki related funtions from models to services/wiki (#9355)
* Move wiki related funtions from models to services/wiki
This commit is contained in:
		
							parent
							
								
									e5d8e2d10c
								
							
						
					
					
						commit
						b9309e52f0
					
				
					 7 changed files with 558 additions and 529 deletions
				
			
		
							
								
								
									
										313
									
								
								models/wiki.go
									
									
									
									
									
								
							
							
						
						
									
										313
									
								
								models/wiki.go
									
									
									
									
									
								
							|  | @ -5,53 +5,12 @@ | ||||||
| package models | package models | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
| 	"fmt" |  | ||||||
| 	"net/url" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" | 	"path/filepath" | ||||||
| 	"strings" | 	"strings" | ||||||
| 
 | 
 | ||||||
| 	"code.gitea.io/gitea/modules/git" |  | ||||||
| 	"code.gitea.io/gitea/modules/log" |  | ||||||
| 	"code.gitea.io/gitea/modules/sync" |  | ||||||
| 
 |  | ||||||
| 	"github.com/unknwon/com" | 	"github.com/unknwon/com" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| var ( |  | ||||||
| 	reservedWikiNames = []string{"_pages", "_new", "_edit", "raw"} |  | ||||||
| 	wikiWorkingPool   = sync.NewExclusivePool() |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| // NormalizeWikiName normalizes a wiki name
 |  | ||||||
| func NormalizeWikiName(name string) string { |  | ||||||
| 	return strings.Replace(name, "-", " ", -1) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // WikiNameToSubURL converts a wiki name to its corresponding sub-URL.
 |  | ||||||
| func WikiNameToSubURL(name string) string { |  | ||||||
| 	return url.QueryEscape(strings.Replace(name, " ", "-", -1)) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // WikiNameToFilename converts a wiki name to its corresponding filename.
 |  | ||||||
| func WikiNameToFilename(name string) string { |  | ||||||
| 	name = strings.Replace(name, " ", "-", -1) |  | ||||||
| 	return url.QueryEscape(name) + ".md" |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // WikiFilenameToName converts a wiki filename to its corresponding page name.
 |  | ||||||
| func WikiFilenameToName(filename string) (string, error) { |  | ||||||
| 	if !strings.HasSuffix(filename, ".md") { |  | ||||||
| 		return "", ErrWikiInvalidFileName{filename} |  | ||||||
| 	} |  | ||||||
| 	basename := filename[:len(filename)-3] |  | ||||||
| 	unescaped, err := url.QueryUnescape(basename) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return "", err |  | ||||||
| 	} |  | ||||||
| 	return NormalizeWikiName(unescaped), nil |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // WikiCloneLink returns clone URLs of repository wiki.
 | // WikiCloneLink returns clone URLs of repository wiki.
 | ||||||
| func (repo *Repository) WikiCloneLink() *CloneLink { | func (repo *Repository) WikiCloneLink() *CloneLink { | ||||||
| 	return repo.cloneLink(x, true) | 	return repo.cloneLink(x, true) | ||||||
|  | @ -71,275 +30,3 @@ func (repo *Repository) WikiPath() string { | ||||||
| func (repo *Repository) HasWiki() bool { | func (repo *Repository) HasWiki() bool { | ||||||
| 	return com.IsDir(repo.WikiPath()) | 	return com.IsDir(repo.WikiPath()) | ||||||
| } | } | ||||||
| 
 |  | ||||||
| // InitWiki initializes a wiki for repository,
 |  | ||||||
| // it does nothing when repository already has wiki.
 |  | ||||||
| func (repo *Repository) InitWiki() error { |  | ||||||
| 	if repo.HasWiki() { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	if err := git.InitRepository(repo.WikiPath(), true); err != nil { |  | ||||||
| 		return fmt.Errorf("InitRepository: %v", err) |  | ||||||
| 	} else if err = createDelegateHooks(repo.WikiPath()); err != nil { |  | ||||||
| 		return fmt.Errorf("createDelegateHooks: %v", err) |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // nameAllowed checks if a wiki name is allowed
 |  | ||||||
| func nameAllowed(name string) error { |  | ||||||
| 	for _, reservedName := range reservedWikiNames { |  | ||||||
| 		if name == reservedName { |  | ||||||
| 			return ErrWikiReservedName{name} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // updateWikiPage adds a new page to the repository wiki.
 |  | ||||||
| func (repo *Repository) updateWikiPage(doer *User, oldWikiName, newWikiName, content, message string, isNew bool) (err error) { |  | ||||||
| 	if err = nameAllowed(newWikiName); err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	wikiWorkingPool.CheckIn(com.ToStr(repo.ID)) |  | ||||||
| 	defer wikiWorkingPool.CheckOut(com.ToStr(repo.ID)) |  | ||||||
| 
 |  | ||||||
| 	if err = repo.InitWiki(); err != nil { |  | ||||||
| 		return fmt.Errorf("InitWiki: %v", err) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	hasMasterBranch := git.IsBranchExist(repo.WikiPath(), "master") |  | ||||||
| 
 |  | ||||||
| 	basePath, err := CreateTemporaryPath("update-wiki") |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	defer func() { |  | ||||||
| 		if err := RemoveTemporaryPath(basePath); err != nil { |  | ||||||
| 			log.Error("Merge: RemoveTemporaryPath: %s", err) |  | ||||||
| 		} |  | ||||||
| 	}() |  | ||||||
| 
 |  | ||||||
| 	cloneOpts := git.CloneRepoOptions{ |  | ||||||
| 		Bare:   true, |  | ||||||
| 		Shared: true, |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	if hasMasterBranch { |  | ||||||
| 		cloneOpts.Branch = "master" |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	if err := git.Clone(repo.WikiPath(), basePath, cloneOpts); err != nil { |  | ||||||
| 		log.Error("Failed to clone repository: %s (%v)", repo.FullName(), err) |  | ||||||
| 		return fmt.Errorf("Failed to clone repository: %s (%v)", repo.FullName(), err) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	gitRepo, err := git.OpenRepository(basePath) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Error("Unable to open temporary repository: %s (%v)", basePath, err) |  | ||||||
| 		return fmt.Errorf("Failed to open new temporary repository in: %s %v", basePath, err) |  | ||||||
| 	} |  | ||||||
| 	defer gitRepo.Close() |  | ||||||
| 
 |  | ||||||
| 	if hasMasterBranch { |  | ||||||
| 		if err := gitRepo.ReadTreeToIndex("HEAD"); err != nil { |  | ||||||
| 			log.Error("Unable to read HEAD tree to index in: %s %v", basePath, err) |  | ||||||
| 			return fmt.Errorf("Unable to read HEAD tree to index in: %s %v", basePath, err) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	newWikiPath := WikiNameToFilename(newWikiName) |  | ||||||
| 	if isNew { |  | ||||||
| 		filesInIndex, err := gitRepo.LsFiles(newWikiPath) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Error("%v", err) |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 		for _, file := range filesInIndex { |  | ||||||
| 			if file == newWikiPath { |  | ||||||
| 				return ErrWikiAlreadyExist{newWikiPath} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} else { |  | ||||||
| 		oldWikiPath := WikiNameToFilename(oldWikiName) |  | ||||||
| 		filesInIndex, err := gitRepo.LsFiles(oldWikiPath) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Error("%v", err) |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 		found := false |  | ||||||
| 		for _, file := range filesInIndex { |  | ||||||
| 			if file == oldWikiPath { |  | ||||||
| 				found = true |  | ||||||
| 				break |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if found { |  | ||||||
| 			err := gitRepo.RemoveFilesFromIndex(oldWikiPath) |  | ||||||
| 			if err != nil { |  | ||||||
| 				log.Error("%v", err) |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// FIXME: The wiki doesn't have lfs support at present - if this changes need to check attributes here
 |  | ||||||
| 
 |  | ||||||
| 	objectHash, err := gitRepo.HashObject(strings.NewReader(content)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Error("%v", err) |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	if err := gitRepo.AddObjectToIndex("100644", objectHash, newWikiPath); err != nil { |  | ||||||
| 		log.Error("%v", err) |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	tree, err := gitRepo.WriteTree() |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Error("%v", err) |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	commitTreeOpts := git.CommitTreeOpts{ |  | ||||||
| 		Message: message, |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	sign, signingKey := repo.SignWikiCommit(doer) |  | ||||||
| 	if sign { |  | ||||||
| 		commitTreeOpts.KeyID = signingKey |  | ||||||
| 	} else { |  | ||||||
| 		commitTreeOpts.NoGPGSign = true |  | ||||||
| 	} |  | ||||||
| 	if hasMasterBranch { |  | ||||||
| 		commitTreeOpts.Parents = []string{"HEAD"} |  | ||||||
| 	} |  | ||||||
| 	commitHash, err := gitRepo.CommitTree(doer.NewGitSig(), tree, commitTreeOpts) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Error("%v", err) |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	if err := git.Push(basePath, git.PushOptions{ |  | ||||||
| 		Remote: "origin", |  | ||||||
| 		Branch: fmt.Sprintf("%s:%s%s", commitHash.String(), git.BranchPrefix, "master"), |  | ||||||
| 		Env: FullPushingEnvironment( |  | ||||||
| 			doer, |  | ||||||
| 			doer, |  | ||||||
| 			repo, |  | ||||||
| 			repo.Name+".wiki", |  | ||||||
| 			0, |  | ||||||
| 		), |  | ||||||
| 	}); err != nil { |  | ||||||
| 		log.Error("%v", err) |  | ||||||
| 		return fmt.Errorf("Push: %v", err) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // AddWikiPage adds a new wiki page with a given wikiPath.
 |  | ||||||
| func (repo *Repository) AddWikiPage(doer *User, wikiName, content, message string) error { |  | ||||||
| 	return repo.updateWikiPage(doer, "", wikiName, content, message, true) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // EditWikiPage updates a wiki page identified by its wikiPath,
 |  | ||||||
| // optionally also changing wikiPath.
 |  | ||||||
| func (repo *Repository) EditWikiPage(doer *User, oldWikiName, newWikiName, content, message string) error { |  | ||||||
| 	return repo.updateWikiPage(doer, oldWikiName, newWikiName, content, message, false) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // DeleteWikiPage deletes a wiki page identified by its path.
 |  | ||||||
| func (repo *Repository) DeleteWikiPage(doer *User, wikiName string) (err error) { |  | ||||||
| 	wikiWorkingPool.CheckIn(com.ToStr(repo.ID)) |  | ||||||
| 	defer wikiWorkingPool.CheckOut(com.ToStr(repo.ID)) |  | ||||||
| 
 |  | ||||||
| 	if err = repo.InitWiki(); err != nil { |  | ||||||
| 		return fmt.Errorf("InitWiki: %v", err) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	basePath, err := CreateTemporaryPath("update-wiki") |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	defer func() { |  | ||||||
| 		if err := RemoveTemporaryPath(basePath); err != nil { |  | ||||||
| 			log.Error("Merge: RemoveTemporaryPath: %s", err) |  | ||||||
| 		} |  | ||||||
| 	}() |  | ||||||
| 
 |  | ||||||
| 	if err := git.Clone(repo.WikiPath(), basePath, git.CloneRepoOptions{ |  | ||||||
| 		Bare:   true, |  | ||||||
| 		Shared: true, |  | ||||||
| 		Branch: "master", |  | ||||||
| 	}); err != nil { |  | ||||||
| 		log.Error("Failed to clone repository: %s (%v)", repo.FullName(), err) |  | ||||||
| 		return fmt.Errorf("Failed to clone repository: %s (%v)", repo.FullName(), err) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	gitRepo, err := git.OpenRepository(basePath) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Error("Unable to open temporary repository: %s (%v)", basePath, err) |  | ||||||
| 		return fmt.Errorf("Failed to open new temporary repository in: %s %v", basePath, err) |  | ||||||
| 	} |  | ||||||
| 	defer gitRepo.Close() |  | ||||||
| 
 |  | ||||||
| 	if err := gitRepo.ReadTreeToIndex("HEAD"); err != nil { |  | ||||||
| 		log.Error("Unable to read HEAD tree to index in: %s %v", basePath, err) |  | ||||||
| 		return fmt.Errorf("Unable to read HEAD tree to index in: %s %v", basePath, err) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	wikiPath := WikiNameToFilename(wikiName) |  | ||||||
| 	filesInIndex, err := gitRepo.LsFiles(wikiPath) |  | ||||||
| 	found := false |  | ||||||
| 	for _, file := range filesInIndex { |  | ||||||
| 		if file == wikiPath { |  | ||||||
| 			found = true |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	if found { |  | ||||||
| 		err := gitRepo.RemoveFilesFromIndex(wikiPath) |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} else { |  | ||||||
| 		return os.ErrNotExist |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// FIXME: The wiki doesn't have lfs support at present - if this changes need to check attributes here
 |  | ||||||
| 
 |  | ||||||
| 	tree, err := gitRepo.WriteTree() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	message := "Delete page '" + wikiName + "'" |  | ||||||
| 	commitTreeOpts := git.CommitTreeOpts{ |  | ||||||
| 		Message: message, |  | ||||||
| 		Parents: []string{"HEAD"}, |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	sign, signingKey := repo.SignWikiCommit(doer) |  | ||||||
| 	if sign { |  | ||||||
| 		commitTreeOpts.KeyID = signingKey |  | ||||||
| 	} else { |  | ||||||
| 		commitTreeOpts.NoGPGSign = true |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	commitHash, err := gitRepo.CommitTree(doer.NewGitSig(), tree, commitTreeOpts) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	if err := git.Push(basePath, git.PushOptions{ |  | ||||||
| 		Remote: "origin", |  | ||||||
| 		Branch: fmt.Sprintf("%s:%s%s", commitHash.String(), git.BranchPrefix, "master"), |  | ||||||
| 		Env:    PushingEnvironment(doer, repo), |  | ||||||
| 	}); err != nil { |  | ||||||
| 		return fmt.Errorf("Push: %v", err) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
|  | @ -8,100 +8,11 @@ import ( | ||||||
| 	"path/filepath" | 	"path/filepath" | ||||||
| 	"testing" | 	"testing" | ||||||
| 
 | 
 | ||||||
| 	"code.gitea.io/gitea/modules/git" |  | ||||||
| 	"code.gitea.io/gitea/modules/setting" | 	"code.gitea.io/gitea/modules/setting" | ||||||
| 
 | 
 | ||||||
| 	"github.com/stretchr/testify/assert" | 	"github.com/stretchr/testify/assert" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| func TestNormalizeWikiName(t *testing.T) { |  | ||||||
| 	type test struct { |  | ||||||
| 		Expected string |  | ||||||
| 		WikiName string |  | ||||||
| 	} |  | ||||||
| 	for _, test := range []test{ |  | ||||||
| 		{"wiki name", "wiki name"}, |  | ||||||
| 		{"wiki name", "wiki-name"}, |  | ||||||
| 		{"name with/slash", "name with/slash"}, |  | ||||||
| 		{"name with%percent", "name-with%percent"}, |  | ||||||
| 		{"%2F", "%2F"}, |  | ||||||
| 	} { |  | ||||||
| 		assert.Equal(t, test.Expected, NormalizeWikiName(test.WikiName)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestWikiNameToFilename(t *testing.T) { |  | ||||||
| 	type test struct { |  | ||||||
| 		Expected string |  | ||||||
| 		WikiName string |  | ||||||
| 	} |  | ||||||
| 	for _, test := range []test{ |  | ||||||
| 		{"wiki-name.md", "wiki name"}, |  | ||||||
| 		{"wiki-name.md", "wiki-name"}, |  | ||||||
| 		{"name-with%2Fslash.md", "name with/slash"}, |  | ||||||
| 		{"name-with%25percent.md", "name with%percent"}, |  | ||||||
| 	} { |  | ||||||
| 		assert.Equal(t, test.Expected, WikiNameToFilename(test.WikiName)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestWikiNameToSubURL(t *testing.T) { |  | ||||||
| 	type test struct { |  | ||||||
| 		Expected string |  | ||||||
| 		WikiName string |  | ||||||
| 	} |  | ||||||
| 	for _, test := range []test{ |  | ||||||
| 		{"wiki-name", "wiki name"}, |  | ||||||
| 		{"wiki-name", "wiki-name"}, |  | ||||||
| 		{"name-with%2Fslash", "name with/slash"}, |  | ||||||
| 		{"name-with%25percent", "name with%percent"}, |  | ||||||
| 	} { |  | ||||||
| 		assert.Equal(t, test.Expected, WikiNameToSubURL(test.WikiName)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestWikiFilenameToName(t *testing.T) { |  | ||||||
| 	type test struct { |  | ||||||
| 		Expected string |  | ||||||
| 		Filename string |  | ||||||
| 	} |  | ||||||
| 	for _, test := range []test{ |  | ||||||
| 		{"hello world", "hello-world.md"}, |  | ||||||
| 		{"symbols/?*", "symbols%2F%3F%2A.md"}, |  | ||||||
| 	} { |  | ||||||
| 		name, err := WikiFilenameToName(test.Filename) |  | ||||||
| 		assert.NoError(t, err) |  | ||||||
| 		assert.Equal(t, test.Expected, name) |  | ||||||
| 	} |  | ||||||
| 	for _, badFilename := range []string{ |  | ||||||
| 		"nofileextension", |  | ||||||
| 		"wrongfileextension.txt", |  | ||||||
| 	} { |  | ||||||
| 		_, err := WikiFilenameToName(badFilename) |  | ||||||
| 		assert.Error(t, err) |  | ||||||
| 		assert.True(t, IsErrWikiInvalidFileName(err)) |  | ||||||
| 	} |  | ||||||
| 	_, err := WikiFilenameToName("badescaping%%.md") |  | ||||||
| 	assert.Error(t, err) |  | ||||||
| 	assert.False(t, IsErrWikiInvalidFileName(err)) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestWikiNameToFilenameToName(t *testing.T) { |  | ||||||
| 	// converting from wiki name to filename, then back to wiki name should
 |  | ||||||
| 	// return the original (normalized) name
 |  | ||||||
| 	for _, name := range []string{ |  | ||||||
| 		"wiki-name", |  | ||||||
| 		"wiki name", |  | ||||||
| 		"wiki name with/slash", |  | ||||||
| 		"$$$%%%^^&&!@#$(),.<>", |  | ||||||
| 	} { |  | ||||||
| 		filename := WikiNameToFilename(name) |  | ||||||
| 		resultName, err := WikiFilenameToName(filename) |  | ||||||
| 		assert.NoError(t, err) |  | ||||||
| 		assert.Equal(t, NormalizeWikiName(name), resultName) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestRepository_WikiCloneLink(t *testing.T) { | func TestRepository_WikiCloneLink(t *testing.T) { | ||||||
| 	assert.NoError(t, PrepareTestDatabase()) | 	assert.NoError(t, PrepareTestDatabase()) | ||||||
| 
 | 
 | ||||||
|  | @ -131,107 +42,3 @@ func TestRepository_HasWiki(t *testing.T) { | ||||||
| 	repo2 := AssertExistsAndLoadBean(t, &Repository{ID: 2}).(*Repository) | 	repo2 := AssertExistsAndLoadBean(t, &Repository{ID: 2}).(*Repository) | ||||||
| 	assert.False(t, repo2.HasWiki()) | 	assert.False(t, repo2.HasWiki()) | ||||||
| } | } | ||||||
| 
 |  | ||||||
| func TestRepository_InitWiki(t *testing.T) { |  | ||||||
| 	PrepareTestEnv(t) |  | ||||||
| 	// repo1 already has a wiki
 |  | ||||||
| 	repo1 := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |  | ||||||
| 	assert.NoError(t, repo1.InitWiki()) |  | ||||||
| 
 |  | ||||||
| 	// repo2 does not already have a wiki
 |  | ||||||
| 	repo2 := AssertExistsAndLoadBean(t, &Repository{ID: 2}).(*Repository) |  | ||||||
| 	assert.NoError(t, repo2.InitWiki()) |  | ||||||
| 	assert.True(t, repo2.HasWiki()) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestRepository_AddWikiPage(t *testing.T) { |  | ||||||
| 	assert.NoError(t, PrepareTestDatabase()) |  | ||||||
| 	const wikiContent = "This is the wiki content" |  | ||||||
| 	const commitMsg = "Commit message" |  | ||||||
| 	repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |  | ||||||
| 	doer := AssertExistsAndLoadBean(t, &User{ID: 2}).(*User) |  | ||||||
| 	for _, wikiName := range []string{ |  | ||||||
| 		"Another page", |  | ||||||
| 		"Here's a <tag> and a/slash", |  | ||||||
| 	} { |  | ||||||
| 		wikiName := wikiName |  | ||||||
| 		t.Run("test wiki exist: "+wikiName, func(t *testing.T) { |  | ||||||
| 			t.Parallel() |  | ||||||
| 			assert.NoError(t, repo.AddWikiPage(doer, wikiName, wikiContent, commitMsg)) |  | ||||||
| 			// Now need to show that the page has been added:
 |  | ||||||
| 			gitRepo, err := git.OpenRepository(repo.WikiPath()) |  | ||||||
| 			assert.NoError(t, err) |  | ||||||
| 			defer gitRepo.Close() |  | ||||||
| 			masterTree, err := gitRepo.GetTree("master") |  | ||||||
| 			assert.NoError(t, err) |  | ||||||
| 			wikiPath := WikiNameToFilename(wikiName) |  | ||||||
| 			entry, err := masterTree.GetTreeEntryByPath(wikiPath) |  | ||||||
| 			assert.NoError(t, err) |  | ||||||
| 			assert.Equal(t, wikiPath, entry.Name(), "%s not addded correctly", wikiName) |  | ||||||
| 		}) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	t.Run("check wiki already exist", func(t *testing.T) { |  | ||||||
| 		t.Parallel() |  | ||||||
| 		// test for already-existing wiki name
 |  | ||||||
| 		err := repo.AddWikiPage(doer, "Home", wikiContent, commitMsg) |  | ||||||
| 		assert.Error(t, err) |  | ||||||
| 		assert.True(t, IsErrWikiAlreadyExist(err)) |  | ||||||
| 	}) |  | ||||||
| 
 |  | ||||||
| 	t.Run("check wiki reserved name", func(t *testing.T) { |  | ||||||
| 		t.Parallel() |  | ||||||
| 		// test for reserved wiki name
 |  | ||||||
| 		err := repo.AddWikiPage(doer, "_edit", wikiContent, commitMsg) |  | ||||||
| 		assert.Error(t, err) |  | ||||||
| 		assert.True(t, IsErrWikiReservedName(err)) |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestRepository_EditWikiPage(t *testing.T) { |  | ||||||
| 	const newWikiContent = "This is the new content" |  | ||||||
| 	const commitMsg = "Commit message" |  | ||||||
| 	repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |  | ||||||
| 	doer := AssertExistsAndLoadBean(t, &User{ID: 2}).(*User) |  | ||||||
| 	for _, newWikiName := range []string{ |  | ||||||
| 		"Home", // same name as before
 |  | ||||||
| 		"New home", |  | ||||||
| 		"New/name/with/slashes", |  | ||||||
| 	} { |  | ||||||
| 		PrepareTestEnv(t) |  | ||||||
| 		assert.NoError(t, repo.EditWikiPage(doer, "Home", newWikiName, newWikiContent, commitMsg)) |  | ||||||
| 
 |  | ||||||
| 		// Now need to show that the page has been added:
 |  | ||||||
| 		gitRepo, err := git.OpenRepository(repo.WikiPath()) |  | ||||||
| 		assert.NoError(t, err) |  | ||||||
| 		masterTree, err := gitRepo.GetTree("master") |  | ||||||
| 		assert.NoError(t, err) |  | ||||||
| 		wikiPath := WikiNameToFilename(newWikiName) |  | ||||||
| 		entry, err := masterTree.GetTreeEntryByPath(wikiPath) |  | ||||||
| 		assert.NoError(t, err) |  | ||||||
| 		assert.Equal(t, wikiPath, entry.Name(), "%s not editted correctly", newWikiName) |  | ||||||
| 
 |  | ||||||
| 		if newWikiName != "Home" { |  | ||||||
| 			_, err := masterTree.GetTreeEntryByPath("Home.md") |  | ||||||
| 			assert.Error(t, err) |  | ||||||
| 		} |  | ||||||
| 		gitRepo.Close() |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| func TestRepository_DeleteWikiPage(t *testing.T) { |  | ||||||
| 	PrepareTestEnv(t) |  | ||||||
| 	repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |  | ||||||
| 	doer := AssertExistsAndLoadBean(t, &User{ID: 2}).(*User) |  | ||||||
| 	assert.NoError(t, repo.DeleteWikiPage(doer, "Home")) |  | ||||||
| 
 |  | ||||||
| 	// Now need to show that the page has been added:
 |  | ||||||
| 	gitRepo, err := git.OpenRepository(repo.WikiPath()) |  | ||||||
| 	assert.NoError(t, err) |  | ||||||
| 	defer gitRepo.Close() |  | ||||||
| 	masterTree, err := gitRepo.GetTree("master") |  | ||||||
| 	assert.NoError(t, err) |  | ||||||
| 	wikiPath := WikiNameToFilename("Home") |  | ||||||
| 	_, err = masterTree.GetTreeEntryByPath(wikiPath) |  | ||||||
| 	assert.Error(t, err) |  | ||||||
| } |  | ||||||
|  |  | ||||||
|  | @ -15,6 +15,7 @@ import ( | ||||||
| 	"code.gitea.io/gitea/modules/private" | 	"code.gitea.io/gitea/modules/private" | ||||||
| 	"code.gitea.io/gitea/modules/setting" | 	"code.gitea.io/gitea/modules/setting" | ||||||
| 	repo_service "code.gitea.io/gitea/services/repository" | 	repo_service "code.gitea.io/gitea/services/repository" | ||||||
|  | 	wiki_service "code.gitea.io/gitea/services/wiki" | ||||||
| 
 | 
 | ||||||
| 	"gitea.com/macaron/macaron" | 	"gitea.com/macaron/macaron" | ||||||
| ) | ) | ||||||
|  | @ -320,7 +321,7 @@ func ServCommand(ctx *macaron.Context) { | ||||||
| 
 | 
 | ||||||
| 	// Finally if we're trying to touch the wiki we should init it
 | 	// Finally if we're trying to touch the wiki we should init it
 | ||||||
| 	if results.IsWiki { | 	if results.IsWiki { | ||||||
| 		if err = repo.InitWiki(); err != nil { | 		if err = wiki_service.InitWiki(repo); err != nil { | ||||||
| 			log.Error("Failed to initialize the wiki in %-v Error: %v", repo, err) | 			log.Error("Failed to initialize the wiki in %-v Error: %v", repo, err) | ||||||
| 			ctx.JSON(http.StatusInternalServerError, map[string]interface{}{ | 			ctx.JSON(http.StatusInternalServerError, map[string]interface{}{ | ||||||
| 				"results": results, | 				"results": results, | ||||||
|  |  | ||||||
|  | @ -22,6 +22,7 @@ import ( | ||||||
| 	"code.gitea.io/gitea/modules/markup/markdown" | 	"code.gitea.io/gitea/modules/markup/markdown" | ||||||
| 	"code.gitea.io/gitea/modules/timeutil" | 	"code.gitea.io/gitea/modules/timeutil" | ||||||
| 	"code.gitea.io/gitea/modules/util" | 	"code.gitea.io/gitea/modules/util" | ||||||
|  | 	wiki_service "code.gitea.io/gitea/services/wiki" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| const ( | const ( | ||||||
|  | @ -124,7 +125,7 @@ func wikiContentsByEntry(ctx *context.Context, entry *git.TreeEntry) []byte { | ||||||
| func wikiContentsByName(ctx *context.Context, commit *git.Commit, wikiName string) ([]byte, *git.TreeEntry, string, bool) { | func wikiContentsByName(ctx *context.Context, commit *git.Commit, wikiName string) ([]byte, *git.TreeEntry, string, bool) { | ||||||
| 	var entry *git.TreeEntry | 	var entry *git.TreeEntry | ||||||
| 	var err error | 	var err error | ||||||
| 	pageFilename := models.WikiNameToFilename(wikiName) | 	pageFilename := wiki_service.NameToFilename(wikiName) | ||||||
| 	if entry, err = findEntryForFile(commit, pageFilename); err != nil { | 	if entry, err = findEntryForFile(commit, pageFilename); err != nil { | ||||||
| 		ctx.ServerError("findEntryForFile", err) | 		ctx.ServerError("findEntryForFile", err) | ||||||
| 		return nil, nil, "", false | 		return nil, nil, "", false | ||||||
|  | @ -157,7 +158,7 @@ func renderViewPage(ctx *context.Context) (*git.Repository, *git.TreeEntry) { | ||||||
| 		if !entry.IsRegular() { | 		if !entry.IsRegular() { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		wikiName, err := models.WikiFilenameToName(entry.Name()) | 		wikiName, err := wiki_service.FilenameToName(entry.Name()) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			if models.IsErrWikiInvalidFileName(err) { | 			if models.IsErrWikiInvalidFileName(err) { | ||||||
| 				continue | 				continue | ||||||
|  | @ -172,17 +173,17 @@ func renderViewPage(ctx *context.Context) (*git.Repository, *git.TreeEntry) { | ||||||
| 		} | 		} | ||||||
| 		pages = append(pages, PageMeta{ | 		pages = append(pages, PageMeta{ | ||||||
| 			Name:   wikiName, | 			Name:   wikiName, | ||||||
| 			SubURL: models.WikiNameToSubURL(wikiName), | 			SubURL: wiki_service.NameToSubURL(wikiName), | ||||||
| 		}) | 		}) | ||||||
| 	} | 	} | ||||||
| 	ctx.Data["Pages"] = pages | 	ctx.Data["Pages"] = pages | ||||||
| 
 | 
 | ||||||
| 	// get requested pagename
 | 	// get requested pagename
 | ||||||
| 	pageName := models.NormalizeWikiName(ctx.Params(":page")) | 	pageName := wiki_service.NormalizeWikiName(ctx.Params(":page")) | ||||||
| 	if len(pageName) == 0 { | 	if len(pageName) == 0 { | ||||||
| 		pageName = "Home" | 		pageName = "Home" | ||||||
| 	} | 	} | ||||||
| 	ctx.Data["PageURL"] = models.WikiNameToSubURL(pageName) | 	ctx.Data["PageURL"] = wiki_service.NameToSubURL(pageName) | ||||||
| 	ctx.Data["old_title"] = pageName | 	ctx.Data["old_title"] = pageName | ||||||
| 	ctx.Data["Title"] = pageName | 	ctx.Data["Title"] = pageName | ||||||
| 	ctx.Data["title"] = pageName | 	ctx.Data["title"] = pageName | ||||||
|  | @ -243,11 +244,11 @@ func renderRevisionPage(ctx *context.Context) (*git.Repository, *git.TreeEntry) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	// get requested pagename
 | 	// get requested pagename
 | ||||||
| 	pageName := models.NormalizeWikiName(ctx.Params(":page")) | 	pageName := wiki_service.NormalizeWikiName(ctx.Params(":page")) | ||||||
| 	if len(pageName) == 0 { | 	if len(pageName) == 0 { | ||||||
| 		pageName = "Home" | 		pageName = "Home" | ||||||
| 	} | 	} | ||||||
| 	ctx.Data["PageURL"] = models.WikiNameToSubURL(pageName) | 	ctx.Data["PageURL"] = wiki_service.NameToSubURL(pageName) | ||||||
| 	ctx.Data["old_title"] = pageName | 	ctx.Data["old_title"] = pageName | ||||||
| 	ctx.Data["Title"] = pageName | 	ctx.Data["Title"] = pageName | ||||||
| 	ctx.Data["title"] = pageName | 	ctx.Data["title"] = pageName | ||||||
|  | @ -320,11 +321,11 @@ func renderEditPage(ctx *context.Context) { | ||||||
| 	}() | 	}() | ||||||
| 
 | 
 | ||||||
| 	// get requested pagename
 | 	// get requested pagename
 | ||||||
| 	pageName := models.NormalizeWikiName(ctx.Params(":page")) | 	pageName := wiki_service.NormalizeWikiName(ctx.Params(":page")) | ||||||
| 	if len(pageName) == 0 { | 	if len(pageName) == 0 { | ||||||
| 		pageName = "Home" | 		pageName = "Home" | ||||||
| 	} | 	} | ||||||
| 	ctx.Data["PageURL"] = models.WikiNameToSubURL(pageName) | 	ctx.Data["PageURL"] = wiki_service.NameToSubURL(pageName) | ||||||
| 	ctx.Data["old_title"] = pageName | 	ctx.Data["old_title"] = pageName | ||||||
| 	ctx.Data["Title"] = pageName | 	ctx.Data["Title"] = pageName | ||||||
| 	ctx.Data["title"] = pageName | 	ctx.Data["title"] = pageName | ||||||
|  | @ -474,7 +475,7 @@ func WikiPages(ctx *context.Context) { | ||||||
| 			ctx.ServerError("GetCommit", err) | 			ctx.ServerError("GetCommit", err) | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 		wikiName, err := models.WikiFilenameToName(entry.Name()) | 		wikiName, err := wiki_service.FilenameToName(entry.Name()) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			if models.IsErrWikiInvalidFileName(err) { | 			if models.IsErrWikiInvalidFileName(err) { | ||||||
| 				continue | 				continue | ||||||
|  | @ -488,7 +489,7 @@ func WikiPages(ctx *context.Context) { | ||||||
| 		} | 		} | ||||||
| 		pages = append(pages, PageMeta{ | 		pages = append(pages, PageMeta{ | ||||||
| 			Name:        wikiName, | 			Name:        wikiName, | ||||||
| 			SubURL:      models.WikiNameToSubURL(wikiName), | 			SubURL:      wiki_service.NameToSubURL(wikiName), | ||||||
| 			UpdatedUnix: timeutil.TimeStamp(c.Author.When.Unix()), | 			UpdatedUnix: timeutil.TimeStamp(c.Author.When.Unix()), | ||||||
| 		}) | 		}) | ||||||
| 	} | 	} | ||||||
|  | @ -528,7 +529,7 @@ func WikiRaw(ctx *context.Context) { | ||||||
| 				providedPath = providedPath[:len(providedPath)-3] | 				providedPath = providedPath[:len(providedPath)-3] | ||||||
| 			} | 			} | ||||||
| 
 | 
 | ||||||
| 			wikiPath := models.WikiNameToFilename(providedPath) | 			wikiPath := wiki_service.NameToFilename(providedPath) | ||||||
| 			entry, err = findEntryForFile(commit, wikiPath) | 			entry, err = findEntryForFile(commit, wikiPath) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				ctx.ServerError("findFile", err) | 				ctx.ServerError("findFile", err) | ||||||
|  | @ -576,8 +577,8 @@ func NewWikiPost(ctx *context.Context, form auth.NewWikiForm) { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	wikiName := models.NormalizeWikiName(form.Title) | 	wikiName := wiki_service.NormalizeWikiName(form.Title) | ||||||
| 	if err := ctx.Repo.Repository.AddWikiPage(ctx.User, wikiName, form.Content, form.Message); err != nil { | 	if err := wiki_service.AddWikiPage(ctx.User, ctx.Repo.Repository, wikiName, form.Content, form.Message); err != nil { | ||||||
| 		if models.IsErrWikiReservedName(err) { | 		if models.IsErrWikiReservedName(err) { | ||||||
| 			ctx.Data["Err_Title"] = true | 			ctx.Data["Err_Title"] = true | ||||||
| 			ctx.RenderWithErr(ctx.Tr("repo.wiki.reserved_page", wikiName), tplWikiNew, &form) | 			ctx.RenderWithErr(ctx.Tr("repo.wiki.reserved_page", wikiName), tplWikiNew, &form) | ||||||
|  | @ -590,7 +591,7 @@ func NewWikiPost(ctx *context.Context, form auth.NewWikiForm) { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	ctx.Redirect(ctx.Repo.RepoLink + "/wiki/" + models.WikiNameToSubURL(wikiName)) | 	ctx.Redirect(ctx.Repo.RepoLink + "/wiki/" + wiki_service.NameToSubURL(wikiName)) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // EditWiki render wiki modify page
 | // EditWiki render wiki modify page
 | ||||||
|  | @ -623,25 +624,25 @@ func EditWikiPost(ctx *context.Context, form auth.NewWikiForm) { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	oldWikiName := models.NormalizeWikiName(ctx.Params(":page")) | 	oldWikiName := wiki_service.NormalizeWikiName(ctx.Params(":page")) | ||||||
| 	newWikiName := models.NormalizeWikiName(form.Title) | 	newWikiName := wiki_service.NormalizeWikiName(form.Title) | ||||||
| 
 | 
 | ||||||
| 	if err := ctx.Repo.Repository.EditWikiPage(ctx.User, oldWikiName, newWikiName, form.Content, form.Message); err != nil { | 	if err := wiki_service.EditWikiPage(ctx.User, ctx.Repo.Repository, oldWikiName, newWikiName, form.Content, form.Message); err != nil { | ||||||
| 		ctx.ServerError("EditWikiPage", err) | 		ctx.ServerError("EditWikiPage", err) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	ctx.Redirect(ctx.Repo.RepoLink + "/wiki/" + models.WikiNameToSubURL(newWikiName)) | 	ctx.Redirect(ctx.Repo.RepoLink + "/wiki/" + wiki_service.NameToSubURL(newWikiName)) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // DeleteWikiPagePost delete wiki page
 | // DeleteWikiPagePost delete wiki page
 | ||||||
| func DeleteWikiPagePost(ctx *context.Context) { | func DeleteWikiPagePost(ctx *context.Context) { | ||||||
| 	wikiName := models.NormalizeWikiName(ctx.Params(":page")) | 	wikiName := wiki_service.NormalizeWikiName(ctx.Params(":page")) | ||||||
| 	if len(wikiName) == 0 { | 	if len(wikiName) == 0 { | ||||||
| 		wikiName = "Home" | 		wikiName = "Home" | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	if err := ctx.Repo.Repository.DeleteWikiPage(ctx.User, wikiName); err != nil { | 	if err := wiki_service.DeleteWikiPage(ctx.User, ctx.Repo.Repository, wikiName); err != nil { | ||||||
| 		ctx.ServerError("DeleteWikiPage", err) | 		ctx.ServerError("DeleteWikiPage", err) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | @ -13,6 +13,7 @@ import ( | ||||||
| 	"code.gitea.io/gitea/modules/auth" | 	"code.gitea.io/gitea/modules/auth" | ||||||
| 	"code.gitea.io/gitea/modules/git" | 	"code.gitea.io/gitea/modules/git" | ||||||
| 	"code.gitea.io/gitea/modules/test" | 	"code.gitea.io/gitea/modules/test" | ||||||
|  | 	wiki_service "code.gitea.io/gitea/services/wiki" | ||||||
| 
 | 
 | ||||||
| 	"github.com/stretchr/testify/assert" | 	"github.com/stretchr/testify/assert" | ||||||
| ) | ) | ||||||
|  | @ -29,7 +30,7 @@ func wikiEntry(t *testing.T, repo *models.Repository, wikiName string) *git.Tree | ||||||
| 	entries, err := commit.ListEntries() | 	entries, err := commit.ListEntries() | ||||||
| 	assert.NoError(t, err) | 	assert.NoError(t, err) | ||||||
| 	for _, entry := range entries { | 	for _, entry := range entries { | ||||||
| 		if entry.Name() == models.WikiNameToFilename(wikiName) { | 		if entry.Name() == wiki_service.NameToFilename(wikiName) { | ||||||
| 			return entry | 			return entry | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
							
								
								
									
										322
									
								
								services/wiki/wiki.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										322
									
								
								services/wiki/wiki.go
									
									
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,322 @@ | ||||||
|  | // Copyright 2015 The Gogs Authors. All rights reserved.
 | ||||||
|  | // Copyright 2019 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 wiki | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"net/url" | ||||||
|  | 	"os" | ||||||
|  | 	"strings" | ||||||
|  | 
 | ||||||
|  | 	"code.gitea.io/gitea/models" | ||||||
|  | 	"code.gitea.io/gitea/modules/git" | ||||||
|  | 	"code.gitea.io/gitea/modules/log" | ||||||
|  | 	"code.gitea.io/gitea/modules/sync" | ||||||
|  | 	"code.gitea.io/gitea/modules/util" | ||||||
|  | 
 | ||||||
|  | 	"github.com/unknwon/com" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | var ( | ||||||
|  | 	reservedWikiNames = []string{"_pages", "_new", "_edit", "raw"} | ||||||
|  | 	wikiWorkingPool   = sync.NewExclusivePool() | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | func nameAllowed(name string) error { | ||||||
|  | 	if util.IsStringInSlice(name, reservedWikiNames) { | ||||||
|  | 		return models.ErrWikiReservedName{ | ||||||
|  | 			Title: name, | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // NameToSubURL converts a wiki name to its corresponding sub-URL.
 | ||||||
|  | func NameToSubURL(name string) string { | ||||||
|  | 	return url.QueryEscape(strings.Replace(name, " ", "-", -1)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // NormalizeWikiName normalizes a wiki name
 | ||||||
|  | func NormalizeWikiName(name string) string { | ||||||
|  | 	return strings.Replace(name, "-", " ", -1) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // NameToFilename converts a wiki name to its corresponding filename.
 | ||||||
|  | func NameToFilename(name string) string { | ||||||
|  | 	name = strings.Replace(name, " ", "-", -1) | ||||||
|  | 	return url.QueryEscape(name) + ".md" | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // FilenameToName converts a wiki filename to its corresponding page name.
 | ||||||
|  | func FilenameToName(filename string) (string, error) { | ||||||
|  | 	if !strings.HasSuffix(filename, ".md") { | ||||||
|  | 		return "", models.ErrWikiInvalidFileName{ | ||||||
|  | 			FileName: filename, | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	basename := filename[:len(filename)-3] | ||||||
|  | 	unescaped, err := url.QueryUnescape(basename) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return NormalizeWikiName(unescaped), nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // InitWiki initializes a wiki for repository,
 | ||||||
|  | // it does nothing when repository already has wiki.
 | ||||||
|  | func InitWiki(repo *models.Repository) error { | ||||||
|  | 	if repo.HasWiki() { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if err := git.InitRepository(repo.WikiPath(), true); err != nil { | ||||||
|  | 		return fmt.Errorf("InitRepository: %v", err) | ||||||
|  | 	} else if err = models.CreateDelegateHooks(repo.WikiPath()); err != nil { | ||||||
|  | 		return fmt.Errorf("createDelegateHooks: %v", err) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // updateWikiPage adds a new page to the repository wiki.
 | ||||||
|  | func updateWikiPage(doer *models.User, repo *models.Repository, oldWikiName, newWikiName, content, message string, isNew bool) (err error) { | ||||||
|  | 	if err = nameAllowed(newWikiName); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	wikiWorkingPool.CheckIn(com.ToStr(repo.ID)) | ||||||
|  | 	defer wikiWorkingPool.CheckOut(com.ToStr(repo.ID)) | ||||||
|  | 
 | ||||||
|  | 	if err = InitWiki(repo); err != nil { | ||||||
|  | 		return fmt.Errorf("InitWiki: %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	hasMasterBranch := git.IsBranchExist(repo.WikiPath(), "master") | ||||||
|  | 
 | ||||||
|  | 	basePath, err := models.CreateTemporaryPath("update-wiki") | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	defer func() { | ||||||
|  | 		if err := models.RemoveTemporaryPath(basePath); err != nil { | ||||||
|  | 			log.Error("Merge: RemoveTemporaryPath: %s", err) | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 
 | ||||||
|  | 	cloneOpts := git.CloneRepoOptions{ | ||||||
|  | 		Bare:   true, | ||||||
|  | 		Shared: true, | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if hasMasterBranch { | ||||||
|  | 		cloneOpts.Branch = "master" | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if err := git.Clone(repo.WikiPath(), basePath, cloneOpts); err != nil { | ||||||
|  | 		log.Error("Failed to clone repository: %s (%v)", repo.FullName(), err) | ||||||
|  | 		return fmt.Errorf("Failed to clone repository: %s (%v)", repo.FullName(), err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	gitRepo, err := git.OpenRepository(basePath) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Error("Unable to open temporary repository: %s (%v)", basePath, err) | ||||||
|  | 		return fmt.Errorf("Failed to open new temporary repository in: %s %v", basePath, err) | ||||||
|  | 	} | ||||||
|  | 	defer gitRepo.Close() | ||||||
|  | 
 | ||||||
|  | 	if hasMasterBranch { | ||||||
|  | 		if err := gitRepo.ReadTreeToIndex("HEAD"); err != nil { | ||||||
|  | 			log.Error("Unable to read HEAD tree to index in: %s %v", basePath, err) | ||||||
|  | 			return fmt.Errorf("Unable to read HEAD tree to index in: %s %v", basePath, err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	newWikiPath := NameToFilename(newWikiName) | ||||||
|  | 	if isNew { | ||||||
|  | 		filesInIndex, err := gitRepo.LsFiles(newWikiPath) | ||||||
|  | 		if err != nil { | ||||||
|  | 			log.Error("%v", err) | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		if util.IsStringInSlice(newWikiPath, filesInIndex) { | ||||||
|  | 			return models.ErrWikiAlreadyExist{ | ||||||
|  | 				Title: newWikiPath, | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		oldWikiPath := NameToFilename(oldWikiName) | ||||||
|  | 		filesInIndex, err := gitRepo.LsFiles(oldWikiPath) | ||||||
|  | 		if err != nil { | ||||||
|  | 			log.Error("%v", err) | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		if util.IsStringInSlice(oldWikiPath, filesInIndex) { | ||||||
|  | 			err := gitRepo.RemoveFilesFromIndex(oldWikiPath) | ||||||
|  | 			if err != nil { | ||||||
|  | 				log.Error("%v", err) | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// FIXME: The wiki doesn't have lfs support at present - if this changes need to check attributes here
 | ||||||
|  | 
 | ||||||
|  | 	objectHash, err := gitRepo.HashObject(strings.NewReader(content)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Error("%v", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if err := gitRepo.AddObjectToIndex("100644", objectHash, newWikiPath); err != nil { | ||||||
|  | 		log.Error("%v", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	tree, err := gitRepo.WriteTree() | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Error("%v", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	commitTreeOpts := git.CommitTreeOpts{ | ||||||
|  | 		Message: message, | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	sign, signingKey := repo.SignWikiCommit(doer) | ||||||
|  | 	if sign { | ||||||
|  | 		commitTreeOpts.KeyID = signingKey | ||||||
|  | 	} else { | ||||||
|  | 		commitTreeOpts.NoGPGSign = true | ||||||
|  | 	} | ||||||
|  | 	if hasMasterBranch { | ||||||
|  | 		commitTreeOpts.Parents = []string{"HEAD"} | ||||||
|  | 	} | ||||||
|  | 	commitHash, err := gitRepo.CommitTree(doer.NewGitSig(), tree, commitTreeOpts) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Error("%v", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if err := git.Push(basePath, git.PushOptions{ | ||||||
|  | 		Remote: "origin", | ||||||
|  | 		Branch: fmt.Sprintf("%s:%s%s", commitHash.String(), git.BranchPrefix, "master"), | ||||||
|  | 		Env: models.FullPushingEnvironment( | ||||||
|  | 			doer, | ||||||
|  | 			doer, | ||||||
|  | 			repo, | ||||||
|  | 			repo.Name+".wiki", | ||||||
|  | 			0, | ||||||
|  | 		), | ||||||
|  | 	}); err != nil { | ||||||
|  | 		log.Error("%v", err) | ||||||
|  | 		return fmt.Errorf("Push: %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // AddWikiPage adds a new wiki page with a given wikiPath.
 | ||||||
|  | func AddWikiPage(doer *models.User, repo *models.Repository, wikiName, content, message string) error { | ||||||
|  | 	return updateWikiPage(doer, repo, "", wikiName, content, message, true) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // EditWikiPage updates a wiki page identified by its wikiPath,
 | ||||||
|  | // optionally also changing wikiPath.
 | ||||||
|  | func EditWikiPage(doer *models.User, repo *models.Repository, oldWikiName, newWikiName, content, message string) error { | ||||||
|  | 	return updateWikiPage(doer, repo, oldWikiName, newWikiName, content, message, false) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // DeleteWikiPage deletes a wiki page identified by its path.
 | ||||||
|  | func DeleteWikiPage(doer *models.User, repo *models.Repository, wikiName string) (err error) { | ||||||
|  | 	wikiWorkingPool.CheckIn(com.ToStr(repo.ID)) | ||||||
|  | 	defer wikiWorkingPool.CheckOut(com.ToStr(repo.ID)) | ||||||
|  | 
 | ||||||
|  | 	if err = InitWiki(repo); err != nil { | ||||||
|  | 		return fmt.Errorf("InitWiki: %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	basePath, err := models.CreateTemporaryPath("update-wiki") | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	defer func() { | ||||||
|  | 		if err := models.RemoveTemporaryPath(basePath); err != nil { | ||||||
|  | 			log.Error("Merge: RemoveTemporaryPath: %s", err) | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 
 | ||||||
|  | 	if err := git.Clone(repo.WikiPath(), basePath, git.CloneRepoOptions{ | ||||||
|  | 		Bare:   true, | ||||||
|  | 		Shared: true, | ||||||
|  | 		Branch: "master", | ||||||
|  | 	}); err != nil { | ||||||
|  | 		log.Error("Failed to clone repository: %s (%v)", repo.FullName(), err) | ||||||
|  | 		return fmt.Errorf("Failed to clone repository: %s (%v)", repo.FullName(), err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	gitRepo, err := git.OpenRepository(basePath) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Error("Unable to open temporary repository: %s (%v)", basePath, err) | ||||||
|  | 		return fmt.Errorf("Failed to open new temporary repository in: %s %v", basePath, err) | ||||||
|  | 	} | ||||||
|  | 	defer gitRepo.Close() | ||||||
|  | 
 | ||||||
|  | 	if err := gitRepo.ReadTreeToIndex("HEAD"); err != nil { | ||||||
|  | 		log.Error("Unable to read HEAD tree to index in: %s %v", basePath, err) | ||||||
|  | 		return fmt.Errorf("Unable to read HEAD tree to index in: %s %v", basePath, err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	wikiPath := NameToFilename(wikiName) | ||||||
|  | 	filesInIndex, err := gitRepo.LsFiles(wikiPath) | ||||||
|  | 	found := false | ||||||
|  | 	for _, file := range filesInIndex { | ||||||
|  | 		if file == wikiPath { | ||||||
|  | 			found = true | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if found { | ||||||
|  | 		err := gitRepo.RemoveFilesFromIndex(wikiPath) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		return os.ErrNotExist | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// FIXME: The wiki doesn't have lfs support at present - if this changes need to check attributes here
 | ||||||
|  | 
 | ||||||
|  | 	tree, err := gitRepo.WriteTree() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	message := "Delete page '" + wikiName + "'" | ||||||
|  | 	commitTreeOpts := git.CommitTreeOpts{ | ||||||
|  | 		Message: message, | ||||||
|  | 		Parents: []string{"HEAD"}, | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	sign, signingKey := repo.SignWikiCommit(doer) | ||||||
|  | 	if sign { | ||||||
|  | 		commitTreeOpts.KeyID = signingKey | ||||||
|  | 	} else { | ||||||
|  | 		commitTreeOpts.NoGPGSign = true | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	commitHash, err := gitRepo.CommitTree(doer.NewGitSig(), tree, commitTreeOpts) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if err := git.Push(basePath, git.PushOptions{ | ||||||
|  | 		Remote: "origin", | ||||||
|  | 		Branch: fmt.Sprintf("%s:%s%s", commitHash.String(), git.BranchPrefix, "master"), | ||||||
|  | 		Env:    models.PushingEnvironment(doer, repo), | ||||||
|  | 	}); err != nil { | ||||||
|  | 		return fmt.Errorf("Push: %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										210
									
								
								services/wiki/wiki_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										210
									
								
								services/wiki/wiki_test.go
									
									
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,210 @@ | ||||||
|  | // Copyright 2019 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 wiki | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"testing" | ||||||
|  | 
 | ||||||
|  | 	"code.gitea.io/gitea/models" | ||||||
|  | 	"code.gitea.io/gitea/modules/git" | ||||||
|  | 	"github.com/stretchr/testify/assert" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | func TestMain(m *testing.M) { | ||||||
|  | 	models.MainTest(m, filepath.Join("..", "..")) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestWikiNameToSubURL(t *testing.T) { | ||||||
|  | 	type test struct { | ||||||
|  | 		Expected string | ||||||
|  | 		WikiName string | ||||||
|  | 	} | ||||||
|  | 	for _, test := range []test{ | ||||||
|  | 		{"wiki-name", "wiki name"}, | ||||||
|  | 		{"wiki-name", "wiki-name"}, | ||||||
|  | 		{"name-with%2Fslash", "name with/slash"}, | ||||||
|  | 		{"name-with%25percent", "name with%percent"}, | ||||||
|  | 	} { | ||||||
|  | 		assert.Equal(t, test.Expected, NameToSubURL(test.WikiName)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestNormalizeWikiName(t *testing.T) { | ||||||
|  | 	type test struct { | ||||||
|  | 		Expected string | ||||||
|  | 		WikiName string | ||||||
|  | 	} | ||||||
|  | 	for _, test := range []test{ | ||||||
|  | 		{"wiki name", "wiki name"}, | ||||||
|  | 		{"wiki name", "wiki-name"}, | ||||||
|  | 		{"name with/slash", "name with/slash"}, | ||||||
|  | 		{"name with%percent", "name-with%percent"}, | ||||||
|  | 		{"%2F", "%2F"}, | ||||||
|  | 	} { | ||||||
|  | 		assert.Equal(t, test.Expected, NormalizeWikiName(test.WikiName)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestWikiNameToFilename(t *testing.T) { | ||||||
|  | 	type test struct { | ||||||
|  | 		Expected string | ||||||
|  | 		WikiName string | ||||||
|  | 	} | ||||||
|  | 	for _, test := range []test{ | ||||||
|  | 		{"wiki-name.md", "wiki name"}, | ||||||
|  | 		{"wiki-name.md", "wiki-name"}, | ||||||
|  | 		{"name-with%2Fslash.md", "name with/slash"}, | ||||||
|  | 		{"name-with%25percent.md", "name with%percent"}, | ||||||
|  | 	} { | ||||||
|  | 		assert.Equal(t, test.Expected, NameToFilename(test.WikiName)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestWikiFilenameToName(t *testing.T) { | ||||||
|  | 	type test struct { | ||||||
|  | 		Expected string | ||||||
|  | 		Filename string | ||||||
|  | 	} | ||||||
|  | 	for _, test := range []test{ | ||||||
|  | 		{"hello world", "hello-world.md"}, | ||||||
|  | 		{"symbols/?*", "symbols%2F%3F%2A.md"}, | ||||||
|  | 	} { | ||||||
|  | 		name, err := FilenameToName(test.Filename) | ||||||
|  | 		assert.NoError(t, err) | ||||||
|  | 		assert.Equal(t, test.Expected, name) | ||||||
|  | 	} | ||||||
|  | 	for _, badFilename := range []string{ | ||||||
|  | 		"nofileextension", | ||||||
|  | 		"wrongfileextension.txt", | ||||||
|  | 	} { | ||||||
|  | 		_, err := FilenameToName(badFilename) | ||||||
|  | 		assert.Error(t, err) | ||||||
|  | 		assert.True(t, models.IsErrWikiInvalidFileName(err)) | ||||||
|  | 	} | ||||||
|  | 	_, err := FilenameToName("badescaping%%.md") | ||||||
|  | 	assert.Error(t, err) | ||||||
|  | 	assert.False(t, models.IsErrWikiInvalidFileName(err)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestWikiNameToFilenameToName(t *testing.T) { | ||||||
|  | 	// converting from wiki name to filename, then back to wiki name should
 | ||||||
|  | 	// return the original (normalized) name
 | ||||||
|  | 	for _, name := range []string{ | ||||||
|  | 		"wiki-name", | ||||||
|  | 		"wiki name", | ||||||
|  | 		"wiki name with/slash", | ||||||
|  | 		"$$$%%%^^&&!@#$(),.<>", | ||||||
|  | 	} { | ||||||
|  | 		filename := NameToFilename(name) | ||||||
|  | 		resultName, err := FilenameToName(filename) | ||||||
|  | 		assert.NoError(t, err) | ||||||
|  | 		assert.Equal(t, NormalizeWikiName(name), resultName) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestRepository_InitWiki(t *testing.T) { | ||||||
|  | 	models.PrepareTestEnv(t) | ||||||
|  | 	// repo1 already has a wiki
 | ||||||
|  | 	repo1 := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 1}).(*models.Repository) | ||||||
|  | 	assert.NoError(t, InitWiki(repo1)) | ||||||
|  | 
 | ||||||
|  | 	// repo2 does not already have a wiki
 | ||||||
|  | 	repo2 := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 2}).(*models.Repository) | ||||||
|  | 	assert.NoError(t, InitWiki(repo2)) | ||||||
|  | 	assert.True(t, repo2.HasWiki()) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestRepository_AddWikiPage(t *testing.T) { | ||||||
|  | 	assert.NoError(t, models.PrepareTestDatabase()) | ||||||
|  | 	const wikiContent = "This is the wiki content" | ||||||
|  | 	const commitMsg = "Commit message" | ||||||
|  | 	repo := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 1}).(*models.Repository) | ||||||
|  | 	doer := models.AssertExistsAndLoadBean(t, &models.User{ID: 2}).(*models.User) | ||||||
|  | 	for _, wikiName := range []string{ | ||||||
|  | 		"Another page", | ||||||
|  | 		"Here's a <tag> and a/slash", | ||||||
|  | 	} { | ||||||
|  | 		wikiName := wikiName | ||||||
|  | 		t.Run("test wiki exist: "+wikiName, func(t *testing.T) { | ||||||
|  | 			t.Parallel() | ||||||
|  | 			assert.NoError(t, AddWikiPage(doer, repo, wikiName, wikiContent, commitMsg)) | ||||||
|  | 			// Now need to show that the page has been added:
 | ||||||
|  | 			gitRepo, err := git.OpenRepository(repo.WikiPath()) | ||||||
|  | 			assert.NoError(t, err) | ||||||
|  | 			defer gitRepo.Close() | ||||||
|  | 			masterTree, err := gitRepo.GetTree("master") | ||||||
|  | 			assert.NoError(t, err) | ||||||
|  | 			wikiPath := NameToFilename(wikiName) | ||||||
|  | 			entry, err := masterTree.GetTreeEntryByPath(wikiPath) | ||||||
|  | 			assert.NoError(t, err) | ||||||
|  | 			assert.Equal(t, wikiPath, entry.Name(), "%s not addded correctly", wikiName) | ||||||
|  | 		}) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	t.Run("check wiki already exist", func(t *testing.T) { | ||||||
|  | 		t.Parallel() | ||||||
|  | 		// test for already-existing wiki name
 | ||||||
|  | 		err := AddWikiPage(doer, repo, "Home", wikiContent, commitMsg) | ||||||
|  | 		assert.Error(t, err) | ||||||
|  | 		assert.True(t, models.IsErrWikiAlreadyExist(err)) | ||||||
|  | 	}) | ||||||
|  | 
 | ||||||
|  | 	t.Run("check wiki reserved name", func(t *testing.T) { | ||||||
|  | 		t.Parallel() | ||||||
|  | 		// test for reserved wiki name
 | ||||||
|  | 		err := AddWikiPage(doer, repo, "_edit", wikiContent, commitMsg) | ||||||
|  | 		assert.Error(t, err) | ||||||
|  | 		assert.True(t, models.IsErrWikiReservedName(err)) | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestRepository_EditWikiPage(t *testing.T) { | ||||||
|  | 	const newWikiContent = "This is the new content" | ||||||
|  | 	const commitMsg = "Commit message" | ||||||
|  | 	repo := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 1}).(*models.Repository) | ||||||
|  | 	doer := models.AssertExistsAndLoadBean(t, &models.User{ID: 2}).(*models.User) | ||||||
|  | 	for _, newWikiName := range []string{ | ||||||
|  | 		"Home", // same name as before
 | ||||||
|  | 		"New home", | ||||||
|  | 		"New/name/with/slashes", | ||||||
|  | 	} { | ||||||
|  | 		models.PrepareTestEnv(t) | ||||||
|  | 		assert.NoError(t, EditWikiPage(doer, repo, "Home", newWikiName, newWikiContent, commitMsg)) | ||||||
|  | 
 | ||||||
|  | 		// Now need to show that the page has been added:
 | ||||||
|  | 		gitRepo, err := git.OpenRepository(repo.WikiPath()) | ||||||
|  | 		assert.NoError(t, err) | ||||||
|  | 		masterTree, err := gitRepo.GetTree("master") | ||||||
|  | 		assert.NoError(t, err) | ||||||
|  | 		wikiPath := NameToFilename(newWikiName) | ||||||
|  | 		entry, err := masterTree.GetTreeEntryByPath(wikiPath) | ||||||
|  | 		assert.NoError(t, err) | ||||||
|  | 		assert.Equal(t, wikiPath, entry.Name(), "%s not editted correctly", newWikiName) | ||||||
|  | 
 | ||||||
|  | 		if newWikiName != "Home" { | ||||||
|  | 			_, err := masterTree.GetTreeEntryByPath("Home.md") | ||||||
|  | 			assert.Error(t, err) | ||||||
|  | 		} | ||||||
|  | 		gitRepo.Close() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestRepository_DeleteWikiPage(t *testing.T) { | ||||||
|  | 	models.PrepareTestEnv(t) | ||||||
|  | 	repo := models.AssertExistsAndLoadBean(t, &models.Repository{ID: 1}).(*models.Repository) | ||||||
|  | 	doer := models.AssertExistsAndLoadBean(t, &models.User{ID: 2}).(*models.User) | ||||||
|  | 	assert.NoError(t, DeleteWikiPage(doer, repo, "Home")) | ||||||
|  | 
 | ||||||
|  | 	// Now need to show that the page has been added:
 | ||||||
|  | 	gitRepo, err := git.OpenRepository(repo.WikiPath()) | ||||||
|  | 	assert.NoError(t, err) | ||||||
|  | 	defer gitRepo.Close() | ||||||
|  | 	masterTree, err := gitRepo.GetTree("master") | ||||||
|  | 	assert.NoError(t, err) | ||||||
|  | 	wikiPath := NameToFilename("Home") | ||||||
|  | 	_, err = masterTree.GetTreeEntryByPath(wikiPath) | ||||||
|  | 	assert.Error(t, err) | ||||||
|  | } | ||||||
		Loading…
	
		Reference in a new issue