[incubator-warble-server] 01/07: Initial checkout: a setup script!!

[hu...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

humbedooh pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-warble-server.git

commit 5f0c3f6d3efba041d21a8244e5b6f1d11a2dceb6
Author: Daniel Gruno <hu...@apache.org>
AuthorDate: Sun Jun 24 15:10:43 2018 -0500

    Initial checkout: a setup script!!
    
    More to come later
---
 setup/dbs.yaml           |  26 ++++++++++
 setup/setup.py           | 122 +++++++++++++++++++++++++++++++++++++++++++++++
 setup/warble.yaml.sample |   9 ++++
 3 files changed, 157 insertions(+)

diff --git a/setup/dbs.yaml b/setup/dbs.yaml
new file mode 100644
index 0000000..7d0dd64
--- /dev/null
+++ b/setup/dbs.yaml
@@ -0,0 +1,26 @@
+# Master UI account setup
+accounts:
+  driver:   sqlite
+  path:     accounts.db
+  layout:
+    cookie:     text              # Web cookie
+    userid:     text primary key  # user ID
+    password:   text              # password digest
+    superuser:  boolean           # admin or not?
+
+# Node registry database setup
+registry:
+  driver:   sqlite
+  path:     nodes.db
+  layout:
+    id:           integer primary key   # ID of node
+    hostname:     text      # hostname of node
+    apikey:       text      # API key for requests
+    pubkey:       text      # public key for encryption/certification
+    verified:     boolean   # Whether we have verified (accepted) this node via UI
+    enabled:      boolean   # enabled/disabled
+    description:  text      # Optional description of node
+    location:     text      # Physical location of node (addr or DC)
+    ip:           text      # Known public IP of node
+    lastping:     integer   # Last time node was alive
+
diff --git a/setup/setup.py b/setup/setup.py
new file mode 100644
index 0000000..b8ed32a
--- /dev/null
+++ b/setup/setup.py
@@ -0,0 +1,122 @@
+#!/usr/bin/env python3
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.
+# The ASF licenses this file to You under the Apache License, Version 2.0
+# (the "License"); you may not use this file except in compliance with
+# the License.  You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+WARBLE_VERSION = '0.1.0' # ABI/API compat demarcation.
+WARBLE_DB_VERSION = 1 # First DB version evah!
+
+import sys
+
+if sys.version_info <= (3, 3):
+    print("This script requires Python 3.4 or higher")
+    sys.exit(-1)
+
+import os
+import sys
+import yaml
+import bcrypt
+import sqlite3
+
+# Figure out absolute setup and api paths
+setup_path = os.path.dirname(os.path.realpath(__file__))
+api_path = os.path.realpath("%s/../api" % setup_path)
+db_path = os.path.realpath("%s/../db" % setup_path)
+
+# Open the DB setup instructions and sample warble server config
+dbsetup = yaml.load(open("%s/dbs.yaml" % setup_path))
+myyaml = yaml.load(open("%s/warble.yaml.sample" % setup_path))
+
+# Set up database path
+datapath = input("Where would you like to put the Warble databases?: [%s]" % db_path).strip()
+if not datapath:
+    datapath = db_path
+print("Creating databases...")
+if not os.path.isdir(datapath):
+    print("Creating directory %s" % datapath)
+    try:
+        os.mkdir(datapath)
+    except Exception as err:
+        print("Could not create database directory: %s" % err)
+        sys.exit(-1)
+
+# Set up individual databases (sqlite only for now!!)
+for name, settings in dbsetup.items():
+    if settings['driver'] == 'sqlite':
+        db_filepath = os.path.join(datapath, settings['path'])
+        if os.path.exists(db_filepath):
+            print("Database %s already exists, skipping!" % db_filepath)
+            continue
+        else:
+            print("Creating database file %s" % db_filepath)
+            conn = sqlite3.connect(db_filepath)
+            cursor = conn.cursor()
+            # create a CREATE statement from the yaml layout
+            cstate = "CREATE TABLE %s (%s)" % (
+                name,
+                ", ".join( ["%s %s" % (k, v) for k, v in settings['layout'].items()] )
+                )
+            cursor.execute(cstate)
+            conn.commit()
+            conn.close()
+
+# Generate a super user
+supername = input("Please enter the username of the primary super user to create [admin]: ").strip()
+if not supername:
+    supername = 'admin'
+superpass = input("Please enter the password for the super user account: ").strip()
+
+# Digest pass with bcrypt
+salt = bcrypt.gensalt()
+pwd = bcrypt.hashpw(superpass.encode('utf-8'), salt).decode('ascii')
+
+# Save user to new DB
+db_filepath = os.path.join(datapath, 'accounts.db')
+conn = sqlite3.connect(db_filepath)
+c = conn.cursor()
+try:
+    c.execute("INSERT INTO `accounts` (`cookie`, `userid`, `password`, `superuser`) VALUES ('zzz', ?, ?, 1)", (supername, pwd))
+    conn.commit()
+    print("Saved new super user account %s in %s" % (supername, db_filepath))
+except sqlite3.IntegrityError:
+    print("WARNING: Could not add super user - is the user already in the DB?!")
+conn.close()
+
+
+# Write new warble yaml config
+warble_yaml_path = '%s/yaml/warble.yaml' % api_path
+print("Writing API server config to %s" % warble_yaml_path)
+if os.path.exists(warble_yaml_path):
+    print("WARNING: Warble YAML already exists on disk, skipping this step!!")
+else:
+    myconfig = {
+        'database': {
+            'version': WARBLE_DB_VERSION,
+            'driver': 'sqlite',
+            'path': datapath
+        },
+        'mail': {
+            'host': 'localhost',
+            'port': 25,
+            'sender': 'Apache Warble<wa...@demo.warble.xyz>'
+        }
+    }
+    with open(warble_yaml_path, "w") as f:
+        f.write(yaml.dump(myconfig, default_flow_style = False))
+        f.close()
+
+    
+print("All done, Warble should...work now :)")
+print("If needed, you can fine tune %s to suit your needs." % warble_yaml_path)
+
diff --git a/setup/warble.yaml.sample b/setup/warble.yaml.sample
new file mode 100644
index 0000000..4533136
--- /dev/null
+++ b/setup/warble.yaml.sample
@@ -0,0 +1,9 @@
+database:
+    driver: sqlite
+    path: /foo/bar/baz
+    
+mail:
+    mailhost:   localhost
+    mailport:   25
+    sender:     Apache Warble <wa...@example.org>
+


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@warble.apache.org
For additional commands, e-mail: commits-help@warble.apache.org