Merge pull request #1090 from wakatime/bugfix/url-type

Add missing entity type url and event

cmd/root.go

@@ -100,7 +100,7 @@ func setFlags(cmd *cobra.Command, v *viper.Viper) {
 	flags.String(
 		"entity-type",
 		"",
-		"Entity type for this heartbeat. Can be \"file\", \"domain\" or \"app\". Defaults to \"file\".",
+		"Entity type for this heartbeat. Can be \"file\", \"domain\", \"url\", or \"app\". Defaults to \"file\".",
 	)
 	flags.StringSlice(
 		"exclude",

pkg/heartbeat/entity.go

@@ -13,13 +13,19 @@ const (
 	FileType EntityType = iota
 	// DomainType represents a domain entity.
 	DomainType
+	// UrlType represents a url entity without the url params.
+	URLType
+	// EventType represents a meeting or calendar event.
+	EventType
 	// AppType represents an app entity.
 	AppType
 )
 
 const (
 	fileTypeString   = "file"
 	domainTypeString = "domain"
+	urlTypeString    = "url"
+	eventTypeString  = "event"
 	appTypeString    = "app"
 )
 
@@ -30,6 +36,10 @@ func ParseEntityType(s string) (EntityType, error) {
 		return FileType, nil
 	case domainTypeString:
 		return DomainType, nil
+	case urlTypeString:
+		return URLType, nil
+	case eventTypeString:
+		return EventType, nil
 	case appTypeString:
 		return AppType, nil
 	default:
@@ -68,6 +78,10 @@ func (t EntityType) String() string {
 		return fileTypeString
 	case DomainType:
 		return domainTypeString
+	case URLType:
+		return urlTypeString
+	case EventType:
+		return eventTypeString
 	case AppType:
 		return appTypeString
 	default:

pkg/heartbeat/entity_test.go

@@ -14,6 +14,8 @@ func typeTests() map[string]heartbeat.EntityType {
 	return map[string]heartbeat.EntityType{
 		"file":   heartbeat.FileType,
 		"domain": heartbeat.DomainType,
+		"url":    heartbeat.URLType,
+		"event":  heartbeat.EventType,
 		"app":    heartbeat.AppType,
 	}
 }