mirror of
https://github.com/haiwen/seafile-server.git
synced 2025-09-01 23:46:53 +00:00
162
ci/serverctl.py
Executable file
162
ci/serverctl.py
Executable file
@@ -0,0 +1,162 @@
|
||||
#!/usr/bin/env python
|
||||
#coding: UTF-8
|
||||
|
||||
import argparse
|
||||
import glob
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
from collections import namedtuple
|
||||
from contextlib import contextmanager
|
||||
from os.path import abspath, basename, dirname, exists, join
|
||||
|
||||
import requests
|
||||
from tenacity import TryAgain, retry, stop_after_attempt, wait_fixed
|
||||
|
||||
from utils import (
|
||||
cd, chdir, debug, green, info, mkdirs, red, setup_logging, shell, warning
|
||||
)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
MYSQL_ROOT_PASSWD = 's123'
|
||||
|
||||
|
||||
class ServerCtl(object):
|
||||
def __init__(self, datadir, db='sqlite3'):
|
||||
self.db = db
|
||||
self.datadir = datadir
|
||||
self.central_conf_dir = join(datadir, 'conf')
|
||||
self.seafile_conf_dir = join(datadir, 'seafile-data')
|
||||
self.ccnet_conf_dir = join(datadir, 'ccnet')
|
||||
|
||||
self.log_dir = join(datadir, 'logs')
|
||||
mkdirs(self.log_dir)
|
||||
self.ccnet_log = join(self.log_dir, 'ccnet.log')
|
||||
self.seafile_log = join(self.log_dir, 'seafile.log')
|
||||
|
||||
self.ccnet_proc = None
|
||||
self.seafile_proc = None
|
||||
|
||||
def setup(self):
|
||||
if self.db == 'mysql':
|
||||
create_mysql_dbs()
|
||||
|
||||
self.init_ccnet()
|
||||
self.init_seafile()
|
||||
|
||||
def init_ccnet(self):
|
||||
cmd = [
|
||||
'ccnet-init',
|
||||
'-F',
|
||||
self.central_conf_dir,
|
||||
'-c',
|
||||
self.ccnet_conf_dir,
|
||||
'--name',
|
||||
'test',
|
||||
'--host',
|
||||
'test.seafile.com',
|
||||
]
|
||||
shell(cmd)
|
||||
|
||||
def init_seafile(self):
|
||||
cmd = [
|
||||
'seaf-server-init',
|
||||
'--central-config-dir',
|
||||
self.central_conf_dir,
|
||||
'--seafile-dir',
|
||||
self.seafile_conf_dir,
|
||||
'--fileserver-port',
|
||||
'8082',
|
||||
]
|
||||
|
||||
shell(cmd)
|
||||
|
||||
@contextmanager
|
||||
def run(self):
|
||||
try:
|
||||
self.start()
|
||||
yield self
|
||||
except:
|
||||
self.print_logs()
|
||||
raise
|
||||
finally:
|
||||
self.stop()
|
||||
|
||||
def print_logs(self):
|
||||
for logfile in self.ccnet_log, self.seafile_log:
|
||||
if exists(logfile):
|
||||
shell('cat {0}'.format(logfile))
|
||||
|
||||
@retry(wait=wait_fixed(1), stop=stop_after_attempt(10))
|
||||
def wait_ccnet_ready(self):
|
||||
if not exists(join(self.ccnet_conf_dir, 'ccnet.sock')):
|
||||
raise TryAgain
|
||||
|
||||
def start(self):
|
||||
logger.info('Starting ccnet server')
|
||||
self.start_ccnet()
|
||||
self.wait_ccnet_ready()
|
||||
logger.info('Starting seafile server')
|
||||
self.start_seafile()
|
||||
|
||||
def start_ccnet(self):
|
||||
cmd = [
|
||||
"ccnet-server",
|
||||
"-F",
|
||||
self.central_conf_dir,
|
||||
"-c",
|
||||
self.ccnet_conf_dir,
|
||||
"-f",
|
||||
self.ccnet_log,
|
||||
]
|
||||
self.ccnet_proc = shell(cmd, wait=False)
|
||||
|
||||
def start_seafile(self):
|
||||
cmd = [
|
||||
"seaf-server",
|
||||
"-F",
|
||||
self.central_conf_dir,
|
||||
"-c",
|
||||
self.ccnet_conf_dir,
|
||||
"-d",
|
||||
self.seafile_conf_dir,
|
||||
"-l",
|
||||
self.seafile_log,
|
||||
]
|
||||
self.seafile_proc = shell(cmd, wait=False)
|
||||
|
||||
def stop(self):
|
||||
if self.ccnet_proc:
|
||||
logger.info('Stopping ccnet server')
|
||||
self.ccnet_proc.terminate()
|
||||
if self.seafile_proc:
|
||||
logger.info('Stopping seafile server')
|
||||
self.seafile_proc.terminate()
|
||||
|
||||
def get_seaserv_envs(self):
|
||||
envs = dict(os.environ)
|
||||
envs.update({
|
||||
'SEAFILE_CENTRAL_CONF_DIR': self.central_conf_dir,
|
||||
'CCNET_CONF_DIR': self.ccnet_conf_dir,
|
||||
'SEAFILE_CONF_DIR': self.seafile_conf_dir,
|
||||
})
|
||||
return envs
|
||||
|
||||
|
||||
def create_mysql_dbs():
|
||||
shell('mysqladmin -u root password %s' % MYSQL_ROOT_PASSWD)
|
||||
sql = '''\
|
||||
create database `ccnet-existing` character set = 'utf8';
|
||||
create database `seafile-existing` character set = 'utf8';
|
||||
create database `seahub-existing` character set = 'utf8';
|
||||
|
||||
create user 'seafile'@'localhost' identified by 'seafile';
|
||||
|
||||
GRANT ALL PRIVILEGES ON `ccnet-existing`.* to `seafile`@localhost;
|
||||
GRANT ALL PRIVILEGES ON `seafile-existing`.* to `seafile`@localhost;
|
||||
GRANT ALL PRIVILEGES ON `seahub-existing`.* to `seafile`@localhost;
|
||||
'''
|
||||
|
||||
shell('mysql -u root -p%s' % MYSQL_ROOT_PASSWD, inputdata=sql)
|
Reference in New Issue
Block a user