コード例 #1
0
ファイル: main.go プロジェクト: apokalyptik/fof
func init() {
	if db, err := bolt.Open("my_bolt.db", 0600, nil); err != nil {
		log.Fatalf("Error opening bolt database: %s", err.Error())
	} else {
		boltDatabase = db
		state = stow.NewJSONStore(db, []byte("state"))
		destinyUsers = stow.NewJSONStore(db, []byte("destiny-users"))
		destinyAccounts = stow.NewJSONStore(db, []byte("destiny-accounts"))
		destinyCharacters = stow.NewJSONStore(db, []byte("destiny-characters"))
	}

	if fp, err := os.Open("creds.json"); err != nil {
		log.Fatalf("Unable to open creds.json: %s", err.Error())
	} else {
		dec := json.NewDecoder(fp)
		if err := dec.Decode(&creds); err != nil {
			log.Fatalf("Unable to decode creds.json: %s", err.Error())
		}
	}
	destinyClient = destiny.New(
		creds["bungieAPI"],
		"github.com/apokalyptik/fof/micro/services/new/destiny-db",
	).Platform(destiny.PlatformXBL)
}
コード例 #2
0
ファイル: main.go プロジェクト: Xe/duke
func main() {
	flag.Parse()

	conf = new(config.Config)

	err := gcfg.ReadFileInto(conf, *cfgFile)
	if err != nil {
		panic(err)
	}

	bdb, err := bolt.Open(conf.Server.DataLocation, 0600, nil)
	if err != nil {
		panic(err)
	}
	defer bdb.Close()

	switch conf.Server.Format {
	case "xml":
		db = stow.NewXMLStore(bdb, []byte("duke-xml"))
	case "json":
		db = stow.NewJSONStore(bdb, []byte("duke-json"))
	case "gob":
		db = stow.NewStore(bdb, []byte("duke-gob"))
	default:
		panic("no set data store in " + *cfgFile)
	}

	db.Put([]byte("foo"), "bar")

	mux := http.NewServeMux()

	mux.HandleFunc("/lookup", func(rw http.ResponseWriter, r *http.Request) {
		if r.Method != "GET" {
			rw.WriteHeader(http.StatusNotAcceptable)
			fmt.Fprintf(rw, "Wrong method")
			return
		}

		user := r.URL.Query().Get("user")

		if len(user) == 0 {
			rw.WriteHeader(http.StatusNotAcceptable)
			fmt.Fprintf(rw, "Need user parameter")
			return
		}

		rw.WriteHeader(http.StatusOK)

		db.ForEach(func(key *duke.Key) {
			if key.Username == user {
				rw.Write([]byte(fmt.Sprintf("# for %s with fp %s\n", key.Username, key.Fingerprint)))
				rw.Write([]byte(key.String()))
				rw.Write([]byte("\n"))
			}
		})
	})

	mux.HandleFunc("/add", func(rw http.ResponseWriter, r *http.Request) {
		if r.Method != "POST" {
			rw.WriteHeader(http.StatusNotAcceptable)
			fmt.Fprintf(rw, "Wrong method")
			return
		}

		user := r.URL.Query().Get("user")

		if len(user) == 0 {
			rw.WriteHeader(http.StatusNotAcceptable)
			fmt.Fprintf(rw, "Need user parameter")
			return
		}

		body, err := ioutil.ReadAll(r.Body)
		if err != nil {
			panic(err) // should never happen
		}

		k := duke.NewKey(user, string(body))

		err = db.Put([]byte(k.Fingerprint), k)
		if err != nil {
			panic(err)
		}

		rw.WriteHeader(http.StatusOK)
	})

	mux.HandleFunc("/remove", func(rw http.ResponseWriter, r *http.Request) {
		if r.Method != "DELETE" {
			rw.WriteHeader(http.StatusNotAcceptable)
			fmt.Fprintf(rw, "Wrong method")
			return
		}

		fingerprint := r.URL.Query().Get("fingerprint")

		if len(fingerprint) == 0 {
			rw.WriteHeader(http.StatusNotAcceptable)
			fmt.Fprintf(rw, "Need fingerprint parameter")
			return
		}

		var k = new(duke.Key)

		err := db.Pull([]byte(fingerprint), k)
		if err != nil {
			panic(err)
		}

		rw.WriteHeader(http.StatusOK)
		fmt.Fprintf(rw, "Key with FP %s for user %s deleted.", k.Fingerprint, k.Username)
	})

	n := negroni.Classic()
	n.UseHandler(mux)
	n.Run(conf.Server.Listen)
}