Elgato_dark/zeppelin
1
0
Fork 0
mirror of https://github.com/apache/zeppelin synced 2026-05-24 09:38:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add new method to help to determinate if user is part of writer and/or owner for the given note

Browse source
This commit is contained in:
Anthony Corbacho 2016-10-28 16:22:13 +09:00
parent 4a43b07c05
commit da3415f3b3
1 changed files with 42 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
zeppelin-zengine/src/main/java/org/apache/zeppelin/notebook/NotebookAuthorization.java
View file

@ -17,18 +17,33 @@
package org.apache.zeppelin.notebook;
import com.google.common.base.Predicate;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.Sets;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.apache.commons.lang3.StringUtils;
import org.apache.zeppelin.conf.ZeppelinConfiguration;
import org.apache.zeppelin.user.AuthenticationInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.*;
import java.util.*;
import com.google.common.base.Predicate;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Sets;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
/**
* Contains authorization information for notes
@ -238,6 +253,26 @@ public class NotebookAuthorization {
intersection.retainAll(a);
return (b.isEmpty() || (intersection.size() > 0));
}
public boolean isUserOwnerOrWriter(String user, String noteId) {
if (StringUtils.isBlank(user)) {
LOG.error("Cannot check user permission with blank login");
return false;
}
Set<String> users = ImmutableSet.of(user);
return isUserOwnerOrWriter(users, noteId);
}
public boolean isUserOwnerOrWriter(Set<String> users, String noteId) {
if (conf.isAnonymousAllowed()) {
LOG.debug("Zeppelin runs in anonymous mode");
return true;
}
if (users == null) {
return false;
}
return isWriter(noteId, users);
}
public void removeNote(String noteId) {
authInfo.remove(noteId);