cli.go

package main

import (
	"context"
	"flag"
	"fmt"
	"io"
	"net/http"
	"os"
	"path"
	"strings"

	"github.com/hashicorp/go-version"

	"github.com/google/go-github/github"
)

var (
	basePath string
)

func fileExists(filename string) bool {
	info, err := os.Stat(filename)
	if os.IsNotExist(err) {
		return false
	}
	return !info.IsDir()
}

func init() {
	flag.StringVar(&basePath, "base", "/var/www/pojt-get", "-base /var/www/pojt-get")
}

func main() {
	flag.Parse()

	baseVersion := "v0.0.0-test.0"
	lastVersion, err := version.NewVersion(baseVersion)
	if err != nil {
		panic(err)
	}

	client := github.NewClient(nil)
	opt := &github.ListOptions{}

	releases, _, err := client.Repositories.ListReleases(context.Background(), "Pigeon-Developer", "pigeon-oj-tool", opt)

	if err != nil {
		panic(err)
	}

	downloadDir := path.Join(basePath, "download")

	for _, release := range releases {
		releasePath := path.Join(downloadDir, *release.TagName)
		os.MkdirAll(releasePath, os.ModePerm)

		fmt.Println("sync release ", *release.TagName)

		// isStable := !*release.Prerelease && !*release.Draft
		isStable := true
		if isStable {
			v, err := version.NewVersion(*release.TagName)
			if err != nil {
				continue
			}

			if v.GreaterThan(lastVersion) {
				lastVersion = v
			}
		}

		for _, asset := range release.Assets {
			filePath := path.Join(releasePath, *asset.Name)
			if fileExists(filePath) {
				fmt.Println("sync file ", filePath, " already exists")
				continue
			}
			if !strings.HasPrefix(*asset.Name, "pojt-") {
				continue
			}

			fmt.Println("sync file ", *asset.Name)

			rc, url, err := client.Repositories.DownloadReleaseAsset(context.Background(), "Pigeon-Developer", "pigeon-oj-tool", *asset.ID)
			if err != nil {
				fmt.Println("sync file ", filePath, " error: ", err)
				continue
			}

			f, err := os.OpenFile(filePath, os.O_CREATE|os.O_WRONLY, os.ModePerm)
			if err != nil {
				fmt.Println("sync file ", filePath, " error: ", err)
				continue
			}
			defer f.Close()

			if rc != nil {
				defer rc.Close()
				_, err := io.Copy(f, rc)
				if err != nil {
					fmt.Println("sync file ", filePath, " error: ", err)
					continue
				}
			} else {
				resp, err := http.Get(url)
				if err != nil {
					fmt.Println("sync file ", filePath, " error: ", err)
					continue
				}
				defer resp.Body.Close()

				_, err = io.Copy(f, resp.Body)

				if err != nil {
					fmt.Println("sync file ", filePath, " error: ", err)
					continue
				}
			}
		}
	}

	if lastVersion.Original() != baseVersion {
		versionPath := path.Join(basePath, "latest-version")
		versionFile, err := os.OpenFile(versionPath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, os.ModePerm)
		if err != nil {
			panic(err)
		}

		defer versionFile.Close()

		versionFile.WriteString(lastVersion.String())
	}
}