Source code for binaryninja.collaboration.folder

from . import _collaboration as core
from . import remote
from . import project
from . import util
from typing import List, Tuple, Optional, Dict
import ctypes


class Folder:
	"""
	Class representing a remote folder in a project.
	"""
	def __init__(self, handle):
		self._handle = core.BNCollaborationFolderCopy(handle)

	def __del__(self):
		core.BNCollaborationFolderFree(self._handle)

	def __eq__(self, other):
		if not isinstance(other, Folder):
			return False
		return self.id == other.id

	def __repr__(self):
		path = self.name
		parent = self.parent
		while parent is not None:
			path = parent.name + '/' + path
			parent = parent.parent
		return f'<folder: {self.remote.name}/{self.project.name}/{path}>'

	def __str__(self):
		path = self.name
		parent = self.parent
		while parent is not None:
			path = parent.name + '/' + path
			parent = parent.parent
		return f'<folder: {self.remote.name}/{self.project.name}/{path}>'

	@property
	def project(self) -> 'project.Project':
		"""
		Owning Project.

		:return: Project object
		"""
		value = core.BNCollaborationFolderGetProject(self._handle)
		if value is None:
			raise RuntimeError(util._last_error())
		result = project.Project(handle=value)
		core.BNCollaborationProjectFree(value)
		return result

	@property
	def remote(self) -> 'remote.Remote':
		"""
		Owning Remote

		:return: Remote object
		"""
		value = core.BNCollaborationFolderGetRemote(self._handle)
		if value is None:
			raise RuntimeError(util._last_error())
		result = remote.Remote(handle=value)
		core.BNCollaborationRemoteFree(value)
		return result

	@property
	def parent(self) -> Optional['Folder']:
		"""
		Parent folder, if one exists. None if this is in the root of a project.

		:return: Parent folder object or None
		"""
		if not self.project.has_pulled_folders:
			self.project.pull_folders()
		parent = ctypes.POINTER(core.BNCollaborationFolder)()
		if not core.BNCollaborationFolderGetParent(self._handle, parent):
			raise RuntimeError(util._last_error())
		if not parent:
			return None
		result = Folder(handle=parent)
		core.BNCollaborationFolderFree(parent)
		return result

	@parent.setter
	def parent(self, parent: Optional['Folder']):
		"""
		Set the parent folder. You will need to push the folder to update the remote version.

		:param parent: New parent folder
		:raises RuntimeError: If there was an error
		"""
		if not core.BNCollaborationFolderSetParent(self._handle, parent._handle if parent is not None else None):
			raise RuntimeError(util._last_error())

	@property
	def url(self) -> str:
		"""
		Web api endpoint URL

		:return: URL string
		"""
		return core.BNCollaborationFolderGetUrl(self._handle)

	@property
	def id(self) -> str:
		"""
		Unique id

		:return: Id string
		"""
		return core.BNCollaborationFolderGetId(self._handle)

	@property
	def parent_id(self) -> Optional[str]:
		"""
		Unique id of parent folder, if there is a parent. None, otherwise

		:return: Id string or None
		"""
		parent_id = ctypes.c_char_p()
		if not core.BNCollaborationFolderGetParentId(self._handle, parent_id):
			return None
		return core.pyNativeStr(parent_id.value)

	@property
	def name(self) -> str:
		"""
		Displayed name of folder

		:return: Name string
		"""
		return core.BNCollaborationFolderGetName(self._handle)

	@name.setter
	def name(self, name: str):
		"""
		Set the display name of the folder. You will need to push the folder to update the remote version.

		:param name: New name
		"""
		core.BNCollaborationFolderSetName(self._handle, name)

	@property
	def description(self) -> str:
		"""
		Description of the folder

		:return: Description string
		"""
		return core.BNCollaborationFolderGetDescription(self._handle)

	@description.setter
	def description(self, description: str):
		"""
		Set the description of the folder. You will need to push the folder to update the remote version.

		:param description: New description
		"""
		core.BNCollaborationFolderSetDescription(self._handle, description)