Hugging Face's logo

Spaces:
omnitool-ai
/
omnitool_on_hf
Running

App Files Community
omnitool_on_hf
File size: 2,932 Bytes 
b39afbe
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
/**
 * Copyright (c) 2023 MERCENARIES.AI PTE. LTD.
 * All rights reserved.
 */

//@ts-check

import { console_log } from './utils.js';

const OMNITOOL_DOCUMENT_TYPES_USERDOC = 'udoc';

// @ts-ignore
function get_effective_key(ctx, key) {
  return `${ctx.userId}:${key}`;
}

// @ts-ignore
function get_db(ctx) {
  const db = ctx.app.services.get('db');
  return db;
}

// @ts-ignore
async function user_db_delete(ctx, key, rev = undefined) {
  const db = get_db(ctx);
  const effectiveKey = get_effective_key(ctx, key);
  console_log(`DELETING key: ${effectiveKey}`);

  let effective_rev = rev;
  if (effective_rev === undefined) {
    try {
      const get_result = await db.getDocumentById(OMNITOOL_DOCUMENT_TYPES_USERDOC, effectiveKey);
      effective_rev = get_result._rev;

      console_log(`fixing rev SUCCEEDED - deleteted rev ${effective_rev}`);

      try {
        await db.deleteDocumentById(OMNITOOL_DOCUMENT_TYPES_USERDOC, effectiveKey, effective_rev);
      } catch (e) {
        console.warn(`deleting ${key} = ${effectiveKey} failed with error: ${e}`);
      }
      return true;
    } catch (e) {
      console_log('deleting: fixing rev failed');
    }
  }
}

// @ts-ignore
async function user_db_put(ctx, value, key, rev = undefined) {
  const db = get_db(ctx);
  const effectiveKey = get_effective_key(ctx, key);

  console_log(`put: ${key} = ${effectiveKey} with rev ${rev}`);

  let effective_rev = rev;
  if (effective_rev === undefined) {
    try {
      const get_result = await db.getDocumentById(OMNITOOL_DOCUMENT_TYPES_USERDOC, effectiveKey);
      effective_rev = get_result._rev;

      console_log(`fixing rev SUCCEEDED - deleteted rev ${effective_rev}`);
    } catch (e) {
      console_log('fixing rev failed');
    }
  }

  try {
    const json = await db.putDocumentById(OMNITOOL_DOCUMENT_TYPES_USERDOC, effectiveKey, { value }, effective_rev);
    if (json == null) {
      console_log(`put: ${key} = ${effectiveKey} failed`);
      return false;
    } else {
      console_log(`put: ${key} = ${effectiveKey} succeeded`);
    }
  } catch (e) {
    throw new Error(`put: ${key} = ${effectiveKey} failed with error: ${e}`);
  }

  return true;
}

// @ts-ignore
async function user_db_get(ctx, key) {
  const effectiveKey = get_effective_key(ctx, key);
  const db = get_db(ctx);

  let json = null;
  try {
    json = await db.getDocumentById(OMNITOOL_DOCUMENT_TYPES_USERDOC, effectiveKey);
  } catch (e) {
    console_log(`usr_db_get: ${key} = ${effectiveKey} failed with error: ${e}`);
  }

  if (json == null) return null;

  const json_value = json.value;
  if (json_value == null) {
    console_log(`usr_db_get NULL VALUE. DELETING IT: ${key} = ${effectiveKey} json = ${JSON.stringify(json)}`);
    await db.deleteDocumentById(OMNITOOL_DOCUMENT_TYPES_USERDOC, effectiveKey, json._rev);
    return null;
  }

  return json_value;
}

export { get_db, user_db_delete, user_db_get, user_db_put };