|
|
|
@ -5,10 +5,16 @@
|
|
|
|
|
package cmd |
|
|
|
|
|
|
|
|
|
import ( |
|
|
|
|
"bufio" |
|
|
|
|
"fmt" |
|
|
|
|
"io" |
|
|
|
|
"io/ioutil" |
|
|
|
|
"os" |
|
|
|
|
"path" |
|
|
|
|
"strings" |
|
|
|
|
|
|
|
|
|
"github.com/codegangsta/cli" |
|
|
|
|
|
|
|
|
|
"github.com/gogits/gogs/models" |
|
|
|
|
"github.com/gogits/gogs/modules/setting" |
|
|
|
|
) |
|
|
|
@ -16,28 +22,152 @@ import (
|
|
|
|
|
var CmdFix = cli.Command{ |
|
|
|
|
Name: "fix", |
|
|
|
|
Usage: "This command for upgrade from old version", |
|
|
|
|
Description: `Fix provide upgrade from old version`, |
|
|
|
|
Action: runFix, |
|
|
|
|
Subcommands: fixCommands, |
|
|
|
|
Flags: []cli.Flag{}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func runFix(k *cli.Context) { |
|
|
|
|
workDir, _ := setting.WorkDir() |
|
|
|
|
newLogger(workDir) |
|
|
|
|
func runFix(ctx *cli.Context) { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
setting.NewConfigContext() |
|
|
|
|
models.LoadModelsConfig() |
|
|
|
|
var fixCommands = []cli.Command{ |
|
|
|
|
{ |
|
|
|
|
Name: "location", |
|
|
|
|
Usage: "Change Gogs app location", |
|
|
|
|
Description: `Command location fixes location change of Gogs |
|
|
|
|
|
|
|
|
|
gogs fix location <old Gogs path> |
|
|
|
|
`, |
|
|
|
|
Action: runFixLocation, |
|
|
|
|
}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// rewriteAuthorizedKeys replaces old Gogs path to the new one.
|
|
|
|
|
func rewriteAuthorizedKeys(sshPath, oldPath, newPath string) error { |
|
|
|
|
fr, err := os.Open(sshPath) |
|
|
|
|
if err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
defer fr.Close() |
|
|
|
|
|
|
|
|
|
tmpPath := sshPath + ".tmp" |
|
|
|
|
fw, err := os.Create(tmpPath) |
|
|
|
|
if err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
defer fw.Close() |
|
|
|
|
|
|
|
|
|
oldPath = "command=\"" + oldPath + " serv" |
|
|
|
|
newPath = "command=\"" + newPath + " serv" |
|
|
|
|
buf := bufio.NewReader(fr) |
|
|
|
|
for { |
|
|
|
|
line, errRead := buf.ReadString('\n') |
|
|
|
|
line = strings.TrimSpace(line) |
|
|
|
|
|
|
|
|
|
if errRead != nil { |
|
|
|
|
if errRead != io.EOF { |
|
|
|
|
return errRead |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Reached end of file, if nothing to read then break,
|
|
|
|
|
// otherwise handle the last line.
|
|
|
|
|
if len(line) == 0 { |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Still finding the line, copy the line that currently read.
|
|
|
|
|
if _, err = fw.WriteString(strings.Replace(line, oldPath, newPath, 1) + "\n"); err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if models.UseSQLite3 { |
|
|
|
|
os.Chdir(workDir) |
|
|
|
|
if errRead == io.EOF { |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
models.SetEngine() |
|
|
|
|
if err = os.Remove(sshPath); err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
return os.Rename(tmpPath, sshPath) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func rewriteUpdateHook(path, appPath string) error { |
|
|
|
|
rp := strings.NewReplacer("\\", "/", " ", "\\ ") |
|
|
|
|
if err := ioutil.WriteFile(path, []byte(fmt.Sprintf(models.TPL_UPDATE_HOOK, |
|
|
|
|
setting.ScriptType, rp.Replace(appPath))), os.ModePerm); err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
return nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
err := models.Fix() |
|
|
|
|
func walkDir(rootPath, recPath, appPath string, depth int) error { |
|
|
|
|
depth++ |
|
|
|
|
if depth > 3 { |
|
|
|
|
return nil |
|
|
|
|
} else if depth == 3 { |
|
|
|
|
if err := rewriteUpdateHook(path.Join(rootPath, "hooks/update"), appPath); err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
dir, err := os.Open(rootPath) |
|
|
|
|
if err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
defer dir.Close() |
|
|
|
|
|
|
|
|
|
fis, err := dir.Readdir(0) |
|
|
|
|
if err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for _, fi := range fis { |
|
|
|
|
if strings.Contains(fi.Name(), ".DS_Store") { |
|
|
|
|
continue |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
relPath := path.Join(recPath, fi.Name()) |
|
|
|
|
curPath := path.Join(rootPath, fi.Name()) |
|
|
|
|
if fi.IsDir() { |
|
|
|
|
if err = walkDir(curPath, relPath, appPath, depth); err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func runFixLocation(ctx *cli.Context) { |
|
|
|
|
if len(ctx.Args()) != 1 { |
|
|
|
|
fmt.Println("Incorrect arguments number, expect 1") |
|
|
|
|
os.Exit(2) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
execPath, _ := setting.ExecPath() |
|
|
|
|
|
|
|
|
|
oldPath := ctx.Args().First() |
|
|
|
|
fmt.Printf("Old location: %s\n", oldPath) |
|
|
|
|
fmt.Println("This command should be executed in the new Gogs path") |
|
|
|
|
fmt.Printf("Do you want to change Gogs app path from old location to:\n") |
|
|
|
|
fmt.Printf("-> %s?\n", execPath) |
|
|
|
|
fmt.Print("Press <enter> to continue, use <Ctrl+c> to exit.") |
|
|
|
|
fmt.Scanln() |
|
|
|
|
|
|
|
|
|
// Fix in authorized_keys file.
|
|
|
|
|
sshPath := path.Join(models.SshPath, "authorized_keys") |
|
|
|
|
fmt.Printf("Fixing pathes in file: %s\n", sshPath) |
|
|
|
|
if err := rewriteAuthorizedKeys(sshPath, oldPath, execPath); err != nil { |
|
|
|
|
fmt.Println(err) |
|
|
|
|
os.Exit(1) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Fix position in gogs-repositories.
|
|
|
|
|
setting.NewConfigContext() |
|
|
|
|
fmt.Printf("Fixing pathes in repositories: %s\n", setting.RepoRootPath) |
|
|
|
|
if err := walkDir(setting.RepoRootPath, "", execPath, 0); err != nil { |
|
|
|
|
fmt.Println(err) |
|
|
|
|
} else { |
|
|
|
|
fmt.Println("Fix successfully!") |
|
|
|
|
os.Exit(1) |
|
|
|
|
} |
|
|
|
|
fmt.Println("Fix position finished!") |
|
|
|
|
} |
|
|
|
|