79 lines
2 KiB
Go
79 lines
2 KiB
Go
|
// Copyright 2016 Jimmy Zelinskie
|
||
|
|
//
|
||
|
|
// 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 bittorrent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ErrUnknownEvent is returned when New fails to return an event.
|
||
|
|
var ErrUnknownEvent = errors.New("unknown event")
|
||
|
|
|
||
|
|
// Event represents an event done by a BitTorrent client.
|
||
|
|
type Event uint8
|
||
|
|
|
||
|
|
const (
|
||
|
|
// None is the event when a BitTorrent client announces due to time lapsed
|
||
|
|
// since the previous announce.
|
||
|
|
None Event = iota
|
||
|
|
|
||
|
|
// Started is the event sent by a BitTorrent client when it joins a swarm.
|
||
|
|
Started
|
||
|
|
|
||
|
|
// Stopped is the event sent by a BitTorrent client when it leaves a swarm.
|
||
|
|
Stopped
|
||
|
|
|
||
|
|
// Completed is the event sent by a BitTorrent client when it finishes
|
||
|
|
// downloading all of the required chunks.
|
||
|
|
Completed
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
eventToString = make(map[Event]string)
|
||
|
|
stringToEvent = make(map[string]Event)
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
eventToString[None] = "none"
|
||
|
|
eventToString[Started] = "started"
|
||
|
|
eventToString[Stopped] = "stopped"
|
||
|
|
eventToString[Completed] = "completed"
|
||
|
|
|
||
|
|
stringToEvent[""] = None
|
||
|
|
|
||
|
|
for k, v := range eventToString {
|
||
|
|
stringToEvent[v] = k
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewEvent returns the proper Event given a string.
|
||
|
|
func NewEvent(eventStr string) (Event, error) {
|
||
|
|
if e, ok := stringToEvent[strings.ToLower(eventStr)]; ok {
|
||
|
|
return e, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return None, ErrUnknownEvent
|
||
|
|
}
|
||
|
|
|
||
|
|
// String implements Stringer for an event.
|
||
|
|
func (e Event) String() string {
|
||
|
|
if name, ok := eventToString[e]; ok {
|
||
|
|
return name
|
||
|
|
}
|
||
|
|
|
||
|
|
panic("bittorrent: event has no associated name")
|
||
|
|
}
|