1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
//!
//! # Examples
//!
//! ```no_run
//! use ashpd::documents::{DocumentsProxy, Permission};
//! use zbus::{fdo::Result, Connection};
//!
//! fn main() -> Result<()> {
//!     let connection = Connection::new_session()?;
//!     let proxy = DocumentsProxy::new(&connection)?;
//!
//!     println!("{:#?}", proxy.get_mount_point()?);
//!
//!     for (doc_id, host_path) in proxy.list("org.mozilla.firefox")? {
//!         if doc_id == "f2ee988d" {
//!             let info = proxy.info(&doc_id)?;
//!             println!("{:#?}", info);
//!         }
//!     }
//!
//!     proxy.grant_permissions(
//!         "f2ee988d",
//!         "org.mozilla.firefox",
//!         &[Permission::GrantPermissions],
//!     )?;
//!     proxy.revoke_permissions("f2ee988d", "org.mozilla.firefox", &[Permission::Write])?;
//!
//!     proxy.delete("f2ee988d")?;
//!
//!     Ok(())
//! }
//! ```

use crate::NString;
use enumflags2::BitFlags;
use serde::{de::Deserializer, Deserialize, Serialize, Serializer};
use serde_repr::{Deserialize_repr, Serialize_repr};
use std::collections::HashMap;
use std::str::FromStr;
use strum_macros::{AsRefStr, EnumString, IntoStaticStr, ToString};
use zbus::{dbus_proxy, fdo::Result};
use zvariant::{Fd, Signature};
use zvariant_derive::Type;

#[derive(Serialize_repr, Deserialize_repr, PartialEq, Copy, Clone, BitFlags, Debug, Type)]
#[repr(u32)]
///
pub enum Flags {
    /// Reuse the existing document store entry for the file.
    ReuseExisting = 1,
    /// Persistent file.
    Persistent = 2,
    /// Depends on the application needs.
    AsNeededByApp = 4,
    /// Export a directory.
    ExportDirectory = 8,
}

/// A `HashMap` mapping application IDs to the permissions for that application
pub type Permissions = HashMap<String, Vec<Permission>>;

#[derive(Debug, Clone, AsRefStr, EnumString, IntoStaticStr, ToString, PartialEq, Eq)]
#[strum(serialize_all = "lowercase")]
/// The possible permissions to grant to a specific application for a specific document.
pub enum Permission {
    /// Read access.
    Read,
    /// Write access.
    Write,
    #[strum(serialize = "grant-permissions")]
    /// The possibility to grant new permissions to the file.
    GrantPermissions,
    /// Delete access.
    Delete,
}

impl zvariant::Type for Permission {
    fn signature() -> Signature<'static> {
        Signature::from_string_unchecked("s".to_string())
    }
}

impl Serialize for Permission {
    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        String::serialize(&self.to_string(), serializer)
    }
}

impl<'de> Deserialize<'de> for Permission {
    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        Ok(Permission::from_str(&String::deserialize(deserializer)?).expect("invalid permission"))
    }
}

#[dbus_proxy(
    interface = "org.freedesktop.portal.Documents",
    default_service = "org.freedesktop.portal.Documents",
    default_path = "/org/freedesktop/portal/documents"
)]
/// The interface lets sandboxed applications make files from the outside world available to sandboxed applications in a controlled way.
///
/// Exported files will be made accessible to the application via a fuse filesystem
/// that gets mounted at `/run/user/$UID/doc/`. The filesystem gets mounted both outside
/// and inside the sandbox, but the view inside the sandbox is restricted to just
/// those files that the application is allowed to access.
///
/// Individual files will appear at `/run/user/$UID/doc/$DOC_ID/filename`,
/// where `$DOC_ID` is the ID of the file in the document store.
/// It is returned by the `Add()` and `AddNamed()` calls.
///
/// The permissions that the application has for a document store entry (see `GrantPermissions()`)
/// are reflected in the POSIX mode bits in the fuse filesystem.
trait Documents {
    /// Adds a file to the document store.
    /// The file is passed in the form of an open file descriptor
    /// to prove that the caller has access to the file.
    ///
    /// Returns the ID of the file in the document store.
    ///
    /// # Arguments
    ///
    /// * `o_path_fd` - open file descriptor for the file to add
    /// * `reuse_existing` - whether to reuse an existing document store entry for the file
    /// * `persistent` - whether to add the file only for this session or permanently
    fn add(&self, o_path_fd: Fd, reuse_existing: bool, persistent: bool) -> Result<String>;

    /// Adds multiple files to the document store.
    /// The files are passed in the form of an open file descriptor
    /// to prove that the caller has access to the file.
    ///
    /// Returns the IDs of the files in the document store along with other extra info.
    ///
    /// # Arguments
    ///
    /// * `o_path_fds` - open file descriptors for the files to export
    /// * `flags` - a `Flags` enum.
    /// * `app_id` - an application ID, or empty string
    /// * `permissions` - the permissions to grant
    fn add_full(
        &self,
        o_path_fds: &[Fd],
        flags: BitFlags<Flags>,
        app_id: &str,
        permissions: &[Permission],
    ) -> Result<(Vec<String>, HashMap<String, zvariant::OwnedValue>)>;

    /// Creates an entry in the document store for writing a new file.
    ///
    /// Returns the ID of the file in the document store.
    ///
    /// # Arguments
    ///
    /// * `o_path_parent_fd` - open file descriptor for the parent directory
    /// * `filename` - the basename for the file
    /// * `reuse_existing` - whether to reuse an existing document store entry for the file
    /// * `persistent` - whether to add the file only for this session or permanently
    fn add_named(
        &self,
        o_path_parent_fd: Fd,
        filename: &NString,
        reuse_existing: bool,
        persistent: bool,
    ) -> Result<String>;

    /// Adds multiple files to the document store.
    /// The files are passed in the form of an open file descriptor
    /// to prove that the caller has access to the file.
    ///
    /// Returns the ID of the file in the document store along with other extra info.
    ///
    /// # Arguments
    ///
    /// * `o_path_fd` - open file descriptor for the parent directory
    /// * `filename` - the basename for the file
    /// * `flags` - a `Flags`
    /// * `app_id` - an application ID, or empty string
    /// * `permissions` - the permissions to grant.
    fn add_named_full(
        &self,
        o_path_fd: Fd,
        filename: &NString,
        flags: BitFlags<Flags>,
        app_id: &str,
        permissions: &[Permission],
    ) -> Result<(String, HashMap<String, zvariant::OwnedValue>)>;

    /// Removes an entry from the document store. The file itself is not deleted.
    /// This call is available inside the sandbox if the application
    /// has the 'delete' permission for the document.
    ///
    /// # Arguments
    ///
    /// * `doc_id` - The ID of the file in the document store
    fn delete(&self, doc_id: &str) -> Result<()>;

    /// Returns the path at which the document store fuse filesystem is mounted.
    /// This will typically be /run/user/$UID/doc/.
    fn get_mount_point(&self) -> Result<NString>;

    /// Grants access permissions for a file in the document store to an application.
    /// This call is available inside the sandbox if the application has the 'grant-permissions' permission for the document.
    ///
    /// # Arguments
    ///
    /// * `doc_id` - the ID of the file in the document store.
    /// * `app_id` - the ID of the application to which permissions are granted.
    /// * `permissions` - the permissions to grant.
    fn grant_permissions(
        &self,
        doc_id: &str,
        app_id: &str,
        permissions: &[Permission],
    ) -> Result<()>;

    /// Gets the filesystem path and application permissions for a document store entry.
    ///
    /// Returns the path of the file in the host filesystem along with the [`Permissions`]
    ///
    /// # Arguments
    ///
    /// * `doc_id` - The ID of the file in the document store
    fn info(&self, doc_id: &str) -> Result<(NString, Permissions)>;

    /// Lists documents in the document store for an application (or for all applications).
    ///
    /// Returns a `HashMap` mapping document IDs to their filesystem path on the host system
    ///
    /// # Arguments
    ///
    /// * `app-id` - The application ID, or '' to list all documents
    fn list(&self, app_id: &str) -> Result<HashMap<String, NString>>;

    /// Looks up the document ID for a file.
    /// This call is not available inside the sandbox.
    ///
    /// Returns the ID of the file in the document store,
    /// or '' if the file is not in the document store
    ///
    /// # Arguments
    ///
    /// - `filename` - A path in the host filesystem
    fn lookup(&self, filename: NString) -> Result<String>;

    /// Revokes access permissions for a file in the document store from an application.
    /// This call is available inside the sandbox if the application
    /// has the 'grant-permissions' permission for the document.
    ///
    /// # Arguments
    ///
    /// * `doc_id` - The ID of the file in the document store
    /// * `app_id` - The ID of the application from which permissions are revoked
    /// * `permissions` - The permissions to revoke.
    fn revoke_permissions(
        &self,
        doc_id: &str,
        app_id: &str,
        permissions: &[Permission],
    ) -> Result<()>;

    /// version property
    #[dbus_proxy(property, name = "version")]
    fn version(&self) -> Result<u32>;
}

/// Interact with `org.freedesktop.portal.FileTransfer` interface.
pub mod file_transfer;