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,
  }
 }