roam.py

import datetime
import functools
import json
import subprocess
import re

import fire

ROAMRESEARCH_DBPATH = 'your_path.json'


def update():
  roam_dir = '/path/to/roam-backup'
  subprocess.call(['ssh-add', '/path/to/.ssh/id_rsa'])
  subprocess.call(['git', 'pull', 'origin', 'main'], cwd=roam_dir)
  clear_cache()

def clear_cache():
  get_pages.cache_clear()

@functools.lru_cache()
def get_pages():
  return json.load(open(ROAMRESEARCH_DBPATH, 'r'))

def get_children():
  children = []
  for page in get_pages():
    children.extend(get_children_of(page))
  return children

def get_child(uid):
  for child in get_children():
    if child.get('uid') == uid:
      return child

def get_children_of(page_or_child):
  children = []
  if isinstance(page_or_child, list):
    immediate_children = page_or_child
  else:
    immediate_children = page_or_child.get('children', [])
  for child in immediate_children:
    if 'uid' in page_or_child:  # child of child
      child['parent_uid'] = page_or_child['uid']
    elif 'title' in page_or_child:  # child of page
      child['parent_title'] = page_or_child.get('title')
    else:
      pass  # child of list
    children.append(child)
    children.extend(get_children_of(child))
  return children

def get_date(child): 
  for c in child.get('children', []):
    if c['string'].startswith('Date:'): 
      return c['string'][5:].lstrip(': ').strip('[]')  # e.g. 'November 5th, 2020'
  return child['create-time']  # e.g. 1606249842438

def get_property_of(page_or_child, name):
  for child in page_or_child.get('children', []):
    if child['string'].rstrip(': ').strip() == name.rstrip(': ').strip():
      return child.get('children', [])
    if child['string'].startswith(name):
      return child['string'][len(name):].lstrip(': ')

def get_last_edit_time(item):
  if isinstance(item, list):
    edit_time = 0
  else:
    edit_time = max(
        item.get('create-time', 0),
        item.get('edit-time', 0),
    )

  for child in get_children_of(item):
    child_edit_time = max(
        child.get('create-time', 0),
        child.get('edit-time', 0),
    )
    edit_time = max(edit_time, child_edit_time)
  return edit_time

def as_text(item, clean=False):
  """Handles strings, lists of children, children, and pages."""
  if isinstance(item, str):
    if clean:
      return clean_brackets(item)
    return item
  if isinstance(item, list):
    return '\n\n'.join(as_text(child, clean=clean) for child in item)
  text = item['string']
  if clean:
    text = clean_brackets(text)
  texts = [text]
  for child in get_children_of(item):
    text = child['string']
    if clean:
      text = clean_brackets(text)
    texts.append(text)
  return '\n\n'.join(texts)

def clean_brackets(text):
  return re.sub(r'\[\[(.*?)\]\]', r'\1', text)


def timestamp_as_date(timestamp):
  return datetime.datetime.fromtimestamp(timestamp / 1000)