mirror of https://github.com/gogits/gogs.git
Unknown
12 years ago
5 changed files with 328 additions and 133 deletions
@ -0,0 +1,140 @@
|
||||
// Copyright 2013 gopm authors.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||
// not use this file except in compliance with the License. You may obtain
|
||||
// a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||
// License for the specific language governing permissions and limitations
|
||||
// under the License.
|
||||
|
||||
package doc |
||||
|
||||
import ( |
||||
"archive/tar" |
||||
"bytes" |
||||
"compress/gzip" |
||||
"io" |
||||
"net/http" |
||||
"os" |
||||
//"path"
|
||||
"regexp" |
||||
"strings" |
||||
) |
||||
|
||||
var LaunchpadPattern = regexp.MustCompile(`^launchpad\.net/(?P<repo>(?P<project>[a-z0-9A-Z_.\-]+)(?P<series>/[a-z0-9A-Z_.\-]+)?|~[a-z0-9A-Z_.\-]+/(\+junk|[a-z0-9A-Z_.\-]+)/[a-z0-9A-Z_.\-]+)(?P<dir>/[a-z0-9A-Z_.\-/]+)*$`) |
||||
|
||||
// GetLaunchpadDoc downloads tarball from launchpad.net.
|
||||
func GetLaunchpadDoc(client *http.Client, match map[string]string, installRepoPath string, nod *Node, cmdFlags map[string]bool) ([]string, error) { |
||||
|
||||
if match["project"] != "" && match["series"] != "" { |
||||
rc, err := httpGet(client, expand("https://code.launchpad.net/{project}{series}/.bzr/branch-format", match), nil) |
||||
switch { |
||||
case err == nil: |
||||
rc.Close() |
||||
// The structure of the import path is launchpad.net/{root}/{dir}.
|
||||
case isNotFound(err): |
||||
// The structure of the import path is is launchpad.net/{project}/{dir}.
|
||||
match["repo"] = match["project"] |
||||
match["dir"] = expand("{series}{dir}", match) |
||||
default: |
||||
return nil, err |
||||
} |
||||
} |
||||
|
||||
var downloadPath string |
||||
// Check if download with specific revision.
|
||||
if len(nod.Value) == 0 { |
||||
downloadPath = expand("https://bazaar.launchpad.net/+branch/{repo}/tarball", match) |
||||
} else { |
||||
downloadPath = expand("https://bazaar.launchpad.net/+branch/{repo}/tarball/"+nod.Value, match) |
||||
} |
||||
|
||||
// Scrape the repo browser to find the project revision and individual Go files.
|
||||
p, err := HttpGetBytes(client, downloadPath, nil) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
installPath := installRepoPath + "/" + nod.ImportPath |
||||
|
||||
// Remove old files.
|
||||
os.RemoveAll(installPath + "/") |
||||
os.MkdirAll(installPath+"/", os.ModePerm) |
||||
|
||||
gzr, err := gzip.NewReader(bytes.NewReader(p)) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
defer gzr.Close() |
||||
|
||||
tr := tar.NewReader(gzr) |
||||
|
||||
var autoPath string // Auto path is the root path that generated by bitbucket.org.
|
||||
// Get source file data.
|
||||
dirs := make([]string, 0, 5) |
||||
for { |
||||
h, err := tr.Next() |
||||
if err == io.EOF { |
||||
break |
||||
} else if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
fn := h.FileInfo().Name() |
||||
// Check root path.
|
||||
if len(autoPath) == 0 { |
||||
autoPath = fn[:strings.Index(fn, match["repo"])+len(match["repo"])] |
||||
} |
||||
absPath := strings.Replace(fn, autoPath, installPath, 1) |
||||
|
||||
switch { |
||||
case h.FileInfo().IsDir(): // Directory.
|
||||
// Create diretory before create file.
|
||||
os.MkdirAll(absPath+"/", os.ModePerm) |
||||
// Check if current directory is example.
|
||||
if !(!cmdFlags["-e"] && strings.Contains(absPath, "example")) { |
||||
dirs = append(dirs, absPath) |
||||
} |
||||
case !strings.HasPrefix(fn, "."): |
||||
//os.MkdirAll(path.Dir(absPath)+"/", os.ModePerm)
|
||||
|
||||
// Get data from archive.
|
||||
fbytes := make([]byte, h.Size) |
||||
if _, err := io.ReadFull(tr, fbytes); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
// Write data to file
|
||||
fw, err := os.Create(absPath) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
_, err = fw.Write(fbytes) |
||||
fw.Close() |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
} |
||||
|
||||
var imports []string |
||||
|
||||
// Check if need to check imports.
|
||||
if nod.IsGetDeps { |
||||
for _, d := range dirs { |
||||
importPkgs, err := CheckImports(d+"/", match["importPath"]) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
imports = append(imports, importPkgs...) |
||||
} |
||||
} |
||||
|
||||
return imports, err |
||||
} |
Loading…
Reference in new issue