4c557eff5d
* Add github api token option to generate-license & generate-gitignore Without api toke, Will face rate limit sometimes. Signed-off-by: a1012112796 <1012112796@qq.com> * Use Basic authentication with tokens Signed-off-by: Andrew Thornton <art27@cantab.net> Co-authored-by: a1012112796 <1012112796@qq.com> Co-authored-by: techknowlogick <techknowlogick@gitea.io>
131 lines
2.8 KiB
Go
131 lines
2.8 KiB
Go
// +build ignore
|
|
|
|
package main
|
|
|
|
import (
|
|
"archive/tar"
|
|
"compress/gzip"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"code.gitea.io/gitea/modules/util"
|
|
)
|
|
|
|
func main() {
|
|
var (
|
|
prefix = "gitea-gitignore"
|
|
url = "https://api.github.com/repos/github/gitignore/tarball"
|
|
githubApiToken = ""
|
|
githubUsername = ""
|
|
destination = ""
|
|
)
|
|
|
|
flag.StringVar(&destination, "dest", "options/gitignore/", "destination for the gitignores")
|
|
flag.StringVar(&githubUsername, "username", "", "github username")
|
|
flag.StringVar(&githubApiToken, "token", "", "github api token")
|
|
flag.Parse()
|
|
|
|
file, err := ioutil.TempFile(os.TempDir(), prefix)
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to create temp file. %s", err)
|
|
}
|
|
|
|
defer util.Remove(file.Name())
|
|
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
log.Fatalf("Failed to download archive. %s", err)
|
|
}
|
|
|
|
if len(githubApiToken) > 0 && len(githubUsername) > 0 {
|
|
req.SetBasicAuth(githubUsername, githubApiToken)
|
|
}
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
log.Fatalf("Failed to download archive. %s", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if _, err := io.Copy(file, resp.Body); err != nil {
|
|
log.Fatalf("Failed to copy archive to file. %s", err)
|
|
}
|
|
|
|
if _, err := file.Seek(0, 0); err != nil {
|
|
log.Fatalf("Failed to reset seek on archive. %s", err)
|
|
}
|
|
|
|
gz, err := gzip.NewReader(file)
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to gunzip the archive. %s", err)
|
|
}
|
|
|
|
tr := tar.NewReader(gz)
|
|
|
|
filesToCopy := make(map[string]string, 0)
|
|
|
|
for {
|
|
hdr, err := tr.Next()
|
|
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to iterate archive. %s", err)
|
|
}
|
|
|
|
if filepath.Ext(hdr.Name) != ".gitignore" {
|
|
continue
|
|
}
|
|
|
|
if hdr.Typeflag == tar.TypeSymlink {
|
|
fmt.Printf("Found symlink %s -> %s\n", hdr.Name, hdr.Linkname)
|
|
filesToCopy[strings.TrimSuffix(filepath.Base(hdr.Name), ".gitignore")] = strings.TrimSuffix(filepath.Base(hdr.Linkname), ".gitignore")
|
|
continue
|
|
}
|
|
|
|
out, err := os.Create(path.Join(destination, strings.TrimSuffix(filepath.Base(hdr.Name), ".gitignore")))
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to create new file. %s", err)
|
|
}
|
|
|
|
defer out.Close()
|
|
|
|
if _, err := io.Copy(out, tr); err != nil {
|
|
log.Fatalf("Failed to write new file. %s", err)
|
|
} else {
|
|
fmt.Printf("Written %s\n", out.Name())
|
|
}
|
|
}
|
|
|
|
for dst, src := range filesToCopy {
|
|
// Read all content of src to data
|
|
src = path.Join(destination, src)
|
|
data, err := ioutil.ReadFile(src)
|
|
if err != nil {
|
|
log.Fatalf("Failed to read src file. %s", err)
|
|
}
|
|
// Write data to dst
|
|
dst = path.Join(destination, dst)
|
|
err = ioutil.WriteFile(dst, data, 0644)
|
|
if err != nil {
|
|
log.Fatalf("Failed to write new file. %s", err)
|
|
}
|
|
fmt.Printf("Written (copy of %s) %s\n", src, dst)
|
|
}
|
|
|
|
fmt.Println("Done")
|
|
}
|