Grafana api

README.md

@@ -29,3 +29,7 @@ Test your setup http://asquera.de/blog/2013-07-10/an-elasticsearch-workflow/
 ## How to setup MYSQL locally
 
 `docker run -e MYSQL_ROOT_PASSWORD=my-secret-pw -e MYSQL_DATABASE=flowupdb -e MYSQL_USER=flowupUser -e MYSQL_PASSWORD=flowupPassword -p 127.0.0.1:3306:3306 -d mysql:5.6`
+
+## How to setup grafana locally
+
+`docker run -p 3000:3000 -d grafana/grafana`

app/Module.java

@@ -1,6 +1,6 @@
 import com.google.inject.AbstractModule;
 import com.google.inject.name.Names;
-import datasources.ElasticSearchDatasource;
+import datasources.elasticsearch.ElasticSearchDatasource;
 import play.Configuration;
 import play.Environment;
 import usecases.MetricsDatasource;
@@ -27,5 +27,10 @@ protected void configure() {
         bind(MetricsDatasource.class)
                 .to(ElasticSearchDatasource.class)
                 .asEagerSingleton();
+
+        Configuration grafanaConf = configuration.getConfig("grafana");
+        bind(Configuration.class)
+                .annotatedWith(Names.named("grafana"))
+                .toInstance(grafanaConf);
     }
 }

app/datasources/ActionWriteResponse.java → app/datasources/elasticsearch/ActionWriteResponse.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import lombok.Data;
 import org.jetbrains.annotations.Nullable;

app/datasources/BulkError.java → app/datasources/elasticsearch/BulkError.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 
 import com.fasterxml.jackson.annotation.JsonProperty;

app/datasources/BulkItemResponse.java → app/datasources/elasticsearch/BulkItemResponse.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonProperty;

app/datasources/BulkResponse.java → app/datasources/elasticsearch/BulkResponse.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import lombok.Data;
 

app/datasources/ElasticSearchDatasource.java → app/datasources/elasticsearch/ElasticSearchDatasource.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import com.fasterxml.jackson.databind.node.ObjectNode;
 import play.libs.F;

app/datasources/ElasticsearchClient.java → app/datasources/elasticsearch/ElasticsearchClient.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import com.fasterxml.jackson.databind.JsonNode;
 import org.apache.commons.lang3.StringUtils;

app/datasources/IndexAction.java → app/datasources/elasticsearch/IndexAction.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
 import lombok.Data;

app/datasources/IndexRequest.java → app/datasources/elasticsearch/IndexRequest.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import com.fasterxml.jackson.databind.JsonNode;
 import lombok.Data;

app/datasources/IndexResponse.java → app/datasources/elasticsearch/IndexResponse.java

@@ -1,4 +1,4 @@
-package datasources;
+package datasources.elasticsearch;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonProperty;

app/datasources/grafana/GrafanaClient.java

@@ -0,0 +1,114 @@
+package datasources.grafana;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import models.Organization;
+import models.User;
+import org.apache.commons.lang3.RandomStringUtils;
+import org.jetbrains.annotations.NotNull;
+import play.Configuration;
+import play.Logger;
+import play.libs.Json;
+import play.libs.ws.WSClient;
+import play.libs.ws.WSRequest;
+import play.mvc.Http;
+
+import javax.inject.Inject;
+import javax.inject.Named;
+import java.security.SecureRandom;
+import java.util.UUID;
+import java.util.concurrent.CompletionStage;
+
+public class GrafanaClient {
+
+    private final WSClient ws;
+    private final String baseUrl;
+    private final String apiKey;
+    private final String adminUser;
+    private final String adminPassword;
+
+    @Inject
+    public GrafanaClient(WSClient ws, @Named("grafana") Configuration grafanaConf) {
+        this.ws = ws;
+
+        this.apiKey = grafanaConf.getString("api_key");
+        this.adminUser = grafanaConf.getString("admin_user");
+        this.adminPassword = grafanaConf.getString("admin_password");
+
+        String scheme = grafanaConf.getString("scheme");
+        String host = grafanaConf.getString("host");
+        String port = grafanaConf.getString("port");
+        this.baseUrl = scheme + "://" + host + ":" + port;
+    }
+
+    public  CompletionStage<GrafanaResponse> createUser(final User user) {
+        String adminUserEndpoint = "/api/admin/users";
+
+        String grafanaPassword = generatePassword();
+        ObjectNode userRequest = Json.newObject()
+                .put("name", user.getName())
+                .put("email", user.getEmail())
+                .put("password", grafanaPassword);
+
+        UUID userId = user.getId();
+
+        Logger.debug(userRequest.toString());
+
+        return getWsRequestForUserCreation(adminUserEndpoint).post(userRequest).thenApply(response -> {
+            Logger.debug(response.getBody());
+            if (response.getStatus() == Http.Status.OK) {
+                String id = response.asJson().get("id").toString();
+                User createdUser = User.find.byId(userId);
+                createdUser.setGrafanaUserId(id);
+                createdUser.setGrafanaPassword(grafanaPassword);
+                createdUser.save();
+            }
+            return Json.fromJson(response.asJson(), GrafanaResponse.class);
+        });
+    }
+
+    public CompletionStage<GrafanaResponse> addUserToOrganisation(User user, Organization organization) {
+        String adminUserEndpoint = "/api/orgs/:orgId/users".replaceFirst(":orgId", organization.grafanaId);
+
+        ObjectNode userRequest = Json.newObject()
+                .put("loginOrEmail", user.getEmail())
+                .put("role", "Viewer");
+
+        UUID userId = user.getId();
+
+        Logger.debug(userRequest.toString());
+
+        return getWsRequestForUserCreation(adminUserEndpoint).post(userRequest).thenApply(response -> {
+            Logger.debug(response.getBody());
+            return Json.fromJson(response.asJson(), GrafanaResponse.class);
+        });
+    }
+
+    public CompletionStage<GrafanaResponse> deleteUserInDefaultOrganisation(User user) {
+        String adminUserEndpoint = "/api/orgs/:orgId/users/:userId".replaceFirst(":orgId", "1").replaceFirst(":userId", user.getGrafanaUserId());
+
+        return getWsRequestForUserCreation(adminUserEndpoint).delete().thenApply(response -> {
+            Logger.debug(response.getBody());
+            return Json.fromJson(response.asJson(), GrafanaResponse.class);
+        });
+    }
+
+    private WSRequest getWsRequestForUserCreation(String adminUserEndpoint) {
+        WSRequest wsRequest = ws.url(baseUrl + adminUserEndpoint).setHeader("Accept", "application/json").setContentType("application/json")
+                .setAuth(this.adminUser, this.adminPassword);
+        Logger.debug(wsRequest.getHeaders().toString());
+        return wsRequest;
+    }
+
+    private WSRequest getWsRequest(String adminUserEndpoint) {
+        WSRequest wsRequest = ws.url(baseUrl + adminUserEndpoint).setHeader("Accept", "application/json").setContentType("application/json")
+                .setHeader("Authorization", "Bearer " + this.apiKey);
+        Logger.debug(wsRequest.getHeaders().toString());
+        return wsRequest;
+    }
+
+    @NotNull
+    private String generatePassword() {
+        String characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789~`!@#$%^&*()-_=+[{]}\\|;:\'\",<.>/?";
+        return RandomStringUtils.random(255, 0, 0, false, false, characters.toCharArray(), new SecureRandom());
+    }
+}

app/datasources/grafana/GrafanaResponse.java

@@ -0,0 +1,9 @@
+package datasources.grafana;
+
+
+import lombok.Data;
+
+@Data
+public class GrafanaResponse {
+    private String message;
+}

app/models/Organization.java

@@ -1,5 +1,6 @@
 package models;
 
+import com.avaje.ebean.ExpressionList;
 import com.avaje.ebean.Model;
 
 import javax.persistence.Entity;
@@ -22,5 +23,16 @@ public class Organization {
     @ManyToMany
     public List<User> members;
 
+    public String grafanaId;
+
     public static Model.Finder<UUID, Organization> find = new Model.Finder<>(Organization.class);
+
+    public static Organization findByGoogleAccount(String googleAccount) {
+        return getGoogleAccountUserFind(googleAccount).findUnique();
+    }
+
+    private static ExpressionList<Organization> getGoogleAccountUserFind(String googleAccount) {
+        return find.where().eq("google_account", googleAccount);
+    }
+
 }

app/models/User.java

@@ -18,24 +18,28 @@
 @Entity
 public class User extends Model {
     @Id
-    public UUID id;
+    private UUID id;
 
     @Constraints.Email
     @Column(unique = true)
-    public String email;
+    private String email;
 
     @Constraints.Required
-    public String name;
+    private String name;
 
-    public boolean active;
+    private boolean active;
 
-    public boolean emailValidated;
+    private boolean emailValidated;
+
+    private String grafanaUserId;
+
+    private String grafanaPassword;
 
     @OneToMany(cascade = CascadeType.ALL)
-    public List<LinkedAccount> linkedAccounts;
+    private List<LinkedAccount> linkedAccounts;
 
     @ManyToMany(mappedBy = "members")
-    public List<Organization> organizations;
+    private List<Organization> organizations;
 
     public static final Finder<UUID, User> find = new Finder<>(User.class);
 

app/module/AuthenticationModule.java

@@ -1,7 +1,6 @@
 package module;
 
 import com.feth.play.module.pa.Resolver;
-import com.feth.play.module.pa.providers.oauth2.github.GithubAuthProvider;
 import com.feth.play.module.pa.providers.oauth2.google.GoogleAuthProvider;
 import com.google.inject.AbstractModule;
 import service.AuthenticationResolver;
@@ -13,7 +12,6 @@ protected void configure() {
         // play-authenticate dependencies
         bind(Resolver.class).to(AuthenticationResolver.class);
         // Following class depend on PlayAuthenticate auth, and they self register to it.
-        bind(GithubAuthProvider.class).asEagerSingleton();
         bind(GoogleAuthProvider.class).asEagerSingleton();
         bind(FlowUpUserService.class).asEagerSingleton();
     }

app/service/FlowUpUserService.java

@@ -4,37 +4,76 @@
 import com.feth.play.module.pa.service.AbstractUserService;
 import com.feth.play.module.pa.user.AuthUser;
 import com.feth.play.module.pa.user.AuthUserIdentity;
+import datasources.grafana.GrafanaClient;
+import models.Organization;
 import models.User;
+import play.Logger;
 
 import javax.inject.Inject;
 import javax.inject.Singleton;
+import java.util.concurrent.ExecutionException;
 
 @Singleton
 public class FlowUpUserService extends AbstractUserService {
 
+    private final GrafanaClient grafanaClient;
+
     @Inject
-    public FlowUpUserService(final PlayAuthenticate auth) {
+    public FlowUpUserService(PlayAuthenticate auth, GrafanaClient grafanaClient) {
         super(auth);
+        this.grafanaClient = grafanaClient;
     }
 
     @Override
-    public Object save(final AuthUser authUser) {
+    public Object save(AuthUser authUser) {
         final boolean isLinked = User.existsByAuthUserIdentity(authUser);
         if (!isLinked) {
-            return User.create(authUser).id;
+            User user = User.create(authUser);
+            createGrafanaUser(user);
+            addUserToGrafanaOrg(user);
+            return user.getId();
         } else {
+            User user = User.findByAuthUserIdentity(authUser);
+            createGrafanaUser(user);
+            addUserToGrafanaOrg(user);
             // we have this user already, so return null
             return null;
         }
     }
 
+    private void createGrafanaUser(User user) {
+        try {
+            grafanaClient.createUser(user).toCompletableFuture().get();
+        } catch (InterruptedException e) {
+            Logger.debug(e.getMessage());
+        } catch (ExecutionException e) {
+            Logger.debug(e.getMessage());
+        }
+    }
+
+    private void addUserToGrafanaOrg(User user) {
+        String[] split = user.getEmail().split("@");
+        if (split.length == 2) {
+            Organization organization = Organization.findByGoogleAccount("@" + split[1]);
+            try {
+                grafanaClient.addUserToOrganisation(user, organization).toCompletableFuture().get();
+                user.refresh();
+                grafanaClient.deleteUserInDefaultOrganisation(user).toCompletableFuture().get();
+            } catch (InterruptedException e) {
+                Logger.debug(e.getMessage());
+            } catch (ExecutionException e) {
+                Logger.debug(e.getMessage());
+            }
+        }
+    }
+
     @Override
     public Object getLocalIdentity(final AuthUserIdentity identity) {
         // For production: Caching might be a good idea here...
         // ...and dont forget to sync the cache when users get deactivated/deleted
         final User u = User.findByAuthUserIdentity(identity);
         if(u != null) {
-            return u.id;
+            return u.getId();
         } else {
             return null;
         }

app/views/admin/user/list.scala.html

@@ -78,27 +78,19 @@ <h1 id="homeTitle">@Messages("users.list.title", currentPage.getTotalRowCount)</
 
                 @for(user <- currentPage.getList) {
                     <tr>
-                        <td><a href="@controllers.admin.routes.UserController.edit(user.id.toString)">@user.name</a></td>
+                        <td><a href="@controllers.admin.routes.UserController.edit(user.getId.toString)">@user.getName</a></td>
                         <td>
-                            @if(user.email == null) {
+                            @if(user.getEmail == null) {
                                 <em>-</em>
                             } else {
-                                @user.email
+                                @user.getEmail
                             }
                         </td>
                         <td>
-                            @if(user.active == null) {
-                                <em>-</em>
-                            } else {
-                                @user.active
-                            }
+                            @user.isActive
                         </td>
                         <td>
-                            @if(user.emailValidated == null) {
-                                <em>-</em>
-                            } else {
-                                @user.emailValidated
-                            }
+                            @user.isEmailValidated
                         </td>
                     </tr>
                 }