Integrate audit log (#18)

Author: edmundoa

src/main/java/org/graylog/plugins/usagestatistics/UsageStatsModule.java

@@ -17,6 +17,7 @@
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import okhttp3.OkHttpClient;
+import org.graylog.plugins.usagestatistics.audit.UsageStatsAuditEventTypes;
 import org.graylog.plugins.usagestatistics.collectors.ClusterCollector;
 import org.graylog.plugins.usagestatistics.collectors.CollectorCollector;
 import org.graylog.plugins.usagestatistics.collectors.ElasticsearchCollector;
@@ -58,5 +59,7 @@ protected void configure() {
         addRestResource(UsageStatsOptOutResource.class);
 
         addConfigBeans();
+
+        addAuditEventTypes(UsageStatsAuditEventTypes.class);
     }
 }

src/main/java/org/graylog/plugins/usagestatistics/UsageStatsOptOutResource.java

@@ -21,6 +21,8 @@
 import io.swagger.annotations.ApiResponse;
 import io.swagger.annotations.ApiResponses;
 import org.apache.shiro.authz.annotation.RequiresAuthentication;
+import org.graylog.plugins.usagestatistics.audit.UsageStatsAuditEventTypes;
+import org.graylog2.audit.jersey.AuditEvent;
 import org.graylog2.plugin.rest.PluginRestResource;
 import org.graylog2.shared.rest.resources.RestResource;
 
@@ -80,6 +82,7 @@ public UsageStatsOptOutState getOptOutState() {
             @ApiResponse(code = 400, message = "Missing or invalid opt-out state"),
             @ApiResponse(code = 500, message = "Internal Server Error")
     })
+    @AuditEvent(type = UsageStatsAuditEventTypes.OPT_OUT_UPDATE)
     public void setOptOutState(@Valid @NotNull UsageStatsOptOutState optOutState) {
         checkPermission(CLUSTER_CONFIG_ENTRY_CREATE, CLUSTER_CONFIG_INSTANCE);
 

src/main/java/org/graylog/plugins/usagestatistics/audit/UsageStatsAuditEventTypes.java

@@ -0,0 +1,36 @@
+/**
+ * Copyright (C) 2015 Graylog, Inc. ([email protected])
+ *
+ * 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 org.graylog.plugins.usagestatistics.audit;
+
+import com.google.common.collect.ImmutableSet;
+import org.graylog2.audit.PluginAuditEventTypes;
+
+import java.util.Set;
+
+public class UsageStatsAuditEventTypes implements PluginAuditEventTypes {
+    private static final String NAMESPACE = "usage_statistics:";
+
+    public static final String OPT_OUT_UPDATE = NAMESPACE + "opt_out:update";
+
+    private static final Set<String> EVENT_TYPES = ImmutableSet.<String>builder()
+            .add(OPT_OUT_UPDATE)
+            .build();
+
+    @Override
+    public Set<String> auditEventTypes() {
+        return EVENT_TYPES;
+    }
+}