/
run.py
97 lines (81 loc) · 3.76 KB
/
run.py
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
import spider
import sys
import os
import json
import traceback
import argparse
'''
requires spider.py be in the same directory as this module
spider.py can be found at http://github.com/shariq/notion-on-firebase
'''
def get_firebase_json_path(firebase_path):
return os.path.abspath(os.path.join(firebase_path, 'firebase.json'))
def add_to_firebase_json(firebase_path, new_rewrites):
firebase_json_path = get_firebase_json_path(firebase_path)
with open(firebase_json_path) as handle:
firebase_json = json.loads(handle.read())
if 'rewrites' not in firebase_json['hosting']:
firebase_json['hosting']['rewrites'] = []
existing_rewrites = firebase_json['hosting']['rewrites']
for new_rewrite in new_rewrites:
for existing_rewrite in existing_rewrites[:]:
if existing_rewrite['destination'] == new_rewrite['destination']:
print 'warning: removing', existing_rewrite
existing_rewrites.remove(existing_rewrite)
elif existing_rewrite['source'] == new_rewrite['source']:
print 'warning: removing', existing_rewrite
existing_rewrites.remove(existing_rewrite)
existing_rewrites.append(new_rewrite)
firebase_json['hosting']['rewrites'] = existing_rewrites
dumped = json.dumps(firebase_json, indent=4)
with open(firebase_json_path, 'w') as handle:
handle.write(dumped)
def get_firebase_public_path(firebase_path):
firebase_json_path = get_firebase_json_path(firebase_path)
with open(firebase_json_path) as handle:
contents = handle.read()
relative_public = json.loads(contents)['hosting']['public']
return os.path.join(firebase_path, relative_public)
def main(root_page, firebase_path, meta_json={}):
print 'root_page:', root_page
print 'firebase_path:', firebase_path
firebase_public_path = get_firebase_public_path(firebase_path)
print 'firebase_public_path:', firebase_public_path
print 'beginning spider...'
rewrites = spider.run(root_page, results_path=firebase_public_path, meta_json=meta_json)
print 'completed spider'
print 'rewrites:', rewrites
add_to_firebase_json(firebase_path, rewrites)
original_path = os.getcwd()
os.chdir(firebase_path)
print 'deploying...'
os.system('firebase deploy')
os.chdir(original_path)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--notion-root', help='The root page ID of your notion document. e.g, d065149ff38a4e7a9b908aeb262b0f4f', required=True)
parser.add_argument('--firebase-path', help='The path to your Firebase project. e.g, ../firebase', required=True)
parser.add_argument('--meta-json-path', help='The path to your meta.json file. e.g, ../meta.json', required=True)
args = parser.parse_args()
notion_root = args.notion_root
firebase_path = args.firebase_path
meta_json_path = args.meta_json_path
if not os.path.exists(firebase_path):
print 'error: that firebase_path could not be found. '
print '(path evaluated to {})'.format(os.path.abspath(firebase_path))
sys.exit(-1)
firebase_public_path = get_firebase_public_path(firebase_path)
if not os.path.exists(os.path.join(firebase_public_path, 'ga.js')):
print 'warning: ga.js was not found in your firebase public path'
print 'hit enter after placing it there or if you don\'t want ga.js'
print '(hint: this is a JS file from Google Analytics)'
raw_input()
meta_json = None
try:
with open(meta_json_path) as f:
meta_json = json.loads(f.read())
except Exception:
print 'error: failed to read or parse json at meta_json_path'
traceback.print_exc()
sys.exit(-1)
main(notion_root, firebase_path, meta_json)