1 files changed, 123 insertions, 0 deletions
diff --git a/jimfs/src/main/java/com/google/common/jimfs/OwnerAttributeProvider.java b/jimfs/src/main/java/com/google/common/jimfs/OwnerAttributeProvider.java
new file mode 100644
index 0000000..3408639
--- /dev/null
+++ b/jimfs/src/main/java/com/google/common/jimfs/OwnerAttributeProvider.java
@@ -0,0 +1,123 @@
+/*
+ * Copyright 2013 Google Inc.
+ *
+ * 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 com.google.common.jimfs;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+import static com.google.common.jimfs.UserLookupService.createUserPrincipal;
+
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableSet;
+import java.io.IOException;
+import java.nio.file.attribute.FileAttributeView;
+import java.nio.file.attribute.FileOwnerAttributeView;
+import java.nio.file.attribute.UserPrincipal;
+import java.util.Map;
+import org.checkerframework.checker.nullness.compatqual.NullableDecl;
+
+/**
+ * Attribute provider that provides the {@link FileOwnerAttributeView} ("owner").
+ *
+ * @author Colin Decker
+ */
+final class OwnerAttributeProvider extends AttributeProvider {
+
+  private static final ImmutableSet<String> ATTRIBUTES = ImmutableSet.of("owner");
+
+  private static final UserPrincipal DEFAULT_OWNER = createUserPrincipal("user");
+
+  @Override
+  public String name() {
+    return "owner";
+  }
+
+  @Override
+  public ImmutableSet<String> fixedAttributes() {
+    return ATTRIBUTES;
+  }
+
+  @Override
+  public ImmutableMap<String, ?> defaultValues(Map<String, ?> userProvidedDefaults) {
+    Object userProvidedOwner = userProvidedDefaults.get("owner:owner");
+
+    UserPrincipal owner = DEFAULT_OWNER;
+    if (userProvidedOwner != null) {
+      if (userProvidedOwner instanceof String) {
+        owner = createUserPrincipal((String) userProvidedOwner);
+      } else {
+        throw invalidType("owner", "owner", userProvidedOwner, String.class, UserPrincipal.class);
+      }
+    }
+
+    return ImmutableMap.of("owner:owner", owner);
+  }
+
+  @NullableDecl
+  @Override
+  public Object get(File file, String attribute) {
+    if (attribute.equals("owner")) {
+      return file.getAttribute("owner", "owner");
+    }
+    return null;
+  }
+
+  @Override
+  public void set(File file, String view, String attribute, Object value, boolean create) {
+    if (attribute.equals("owner")) {
+      checkNotCreate(view, attribute, create);
+      UserPrincipal user = checkType(view, attribute, value, UserPrincipal.class);
+      // TODO(cgdecker): Do we really need to do this? Any reason not to allow any UserPrincipal?
+      if (!(user instanceof UserLookupService.JimfsUserPrincipal)) {
+        user = createUserPrincipal(user.getName());
+      }
+      file.setAttribute("owner", "owner", user);
+    }
+  }
+
+  @Override
+  public Class<FileOwnerAttributeView> viewType() {
+    return FileOwnerAttributeView.class;
+  }
+
+  @Override
+  public FileOwnerAttributeView view(
+      FileLookup lookup, ImmutableMap<String, FileAttributeView> inheritedViews) {
+    return new View(lookup);
+  }
+
+  /** Implementation of {@link FileOwnerAttributeView}. */
+  private static final class View extends AbstractAttributeView implements FileOwnerAttributeView {
+
+    public View(FileLookup lookup) {
+      super(lookup);
+    }
+
+    @Override
+    public String name() {
+      return "owner";
+    }
+
+    @Override
+    public UserPrincipal getOwner() throws IOException {
+      return (UserPrincipal) lookupFile().getAttribute("owner", "owner");
+    }
+
+    @Override
+    public void setOwner(UserPrincipal owner) throws IOException {
+      lookupFile().setAttribute("owner", "owner", checkNotNull(owner));
+    }
+  }
+}