# cython:language_level=3 from django.http import JsonResponse import json import time import datetime # import hashlib import math from django.contrib import auth # from django.forms.models import model_to_dict from api.models import RustDeskToken, UserProfile, RustDeskTag, RustDeskPeer, RustDesDevice, ConnLog, FileLog from django.db.models import Q import copy from .views_front import * from django.utils.translation import gettext as _ def login(request): result = {} if request.method == 'GET': result['error'] = _('请求方式错误！请使用POST方式。') return JsonResponse(result) data = json.loads(request.body.decode()) username = data.get('username', '') password = data.get('password', '') rid = data.get('id', '') uuid = data.get('uuid', '') autoLogin = data.get('autoLogin', True) rtype = data.get('type', '') deviceInfo = data.get('deviceInfo', '') user = auth.authenticate(username=username, password=password) if not user: result['error'] = _('帐号或密码错误！请重试，多次重试后将被锁定IP！') return JsonResponse(result) user.rid = rid user.uuid = uuid user.autoLogin = autoLogin user.rtype = rtype user.deviceInfo = json.dumps(deviceInfo) user.save() token = RustDeskToken.objects.filter(Q(uid=user.id) & Q(username=user.username) & Q(rid=user.rid)).first() # 检查是否过期 if token: now_t = datetime.datetime.now() nums = (now_t - token.create_time).seconds if now_t > token.create_time else 0 if nums >= EFFECTIVE_SECONDS: token.delete() token = None if not token: # 获取并保存token token = RustDeskToken( username=user.username, uid=user.id, uuid=user.uuid, rid=user.rid, access_token=getStrMd5(str(time.time()) + salt) ) token.save() result['access_token'] = token.access_token result['type'] = 'access_token' result['user'] = {'name': user.username} return JsonResponse(result) def logout(request): if request.method == 'GET': result = {'error': _('请求方式错误！')} return JsonResponse(result) data = json.loads(request.body.decode()) rid = data.get('id', '') uuid = data.get('uuid', '') user = UserProfile.objects.filter(Q(rid=rid) & Q(uuid=uuid)).first() if not user: result = {'error': _('异常请求！')} return JsonResponse(result) token = RustDeskToken.objects.filter(Q(uid=user.id) & Q(rid=user.rid)).first() if token: token.delete() result = {'code': 1} return JsonResponse(result) def currentUser(request): result = {} if request.method == 'GET': result['error'] = _('错误的提交方式！') return JsonResponse(result) # postdata = json.loads(request.body) # rid = postdata.get('id', '') # uuid = postdata.get('uuid', '') access_token = request.META.get('HTTP_AUTHORIZATION', '') access_token = access_token.split('Bearer ')[-1] token = RustDeskToken.objects.filter(Q(access_token=access_token)).first() user = None if token: user = UserProfile.objects.filter(Q(id=token.uid)).first() if user: if token: result['access_token'] = token.access_token result['type'] = 'access_token' result['name'] = user.username return JsonResponse(result) def ab(request): ''' ''' access_token = request.META.get('HTTP_AUTHORIZATION', '') access_token = access_token.split('Bearer ')[-1] token = RustDeskToken.objects.filter(Q(access_token=access_token)).first() if not token: result = {'error': _('拉取列表错误！')} return JsonResponse(result) if request.method == 'GET': result = {} uid = token.uid tags = RustDeskTag.objects.filter(Q(uid=uid)) tag_names = [] tag_colors = {} if tags: tag_names = [str(x.tag_name) for x in tags] tag_colors = {str(x.tag_name): int(x.tag_color) for x in tags if x.tag_color != ''} peers_result = [] peers = RustDeskPeer.objects.filter(Q(uid=uid)) if peers: for peer in peers: tmp = { 'id': peer.rid, 'username': peer.username, 'hostname': peer.hostname, 'alias': peer.alias, 'platform': peer.platform, 'tags': peer.tags.split(','), 'hash': peer.rhash, } peers_result.append(tmp) result['updated_at'] = datetime.datetime.now() result['data'] = { 'tags': tag_names, 'peers': peers_result, 'tag_colors': json.dumps(tag_colors) } result['data'] = json.dumps(result['data']) return JsonResponse(result) else: postdata = json.loads(request.body.decode()) data = postdata.get('data', '') data = {} if data == '' else json.loads(data) tagnames = data.get('tags', []) tag_colors = data.get('tag_colors', '') tag_colors = {} if tag_colors == '' else json.loads(tag_colors) peers = data.get('peers', []) if tagnames: # 删除旧的tag RustDeskTag.objects.filter(uid=token.uid).delete() # 增加新的 newlist = [] for name in tagnames: tag = RustDeskTag( uid=token.uid, tag_name=name, tag_color=tag_colors.get(name, '') ) newlist.append(tag) RustDeskTag.objects.bulk_create(newlist) if peers: RustDeskPeer.objects.filter(uid=token.uid).delete() newlist = [] for one in peers: peer = RustDeskPeer( uid=token.uid, rid=one['id'], username=one['username'], hostname=one['hostname'], alias=one['alias'], platform=one['platform'], tags=','.join(one['tags']), rhash=one['hash'], ) newlist.append(peer) RustDeskPeer.objects.bulk_create(newlist) result = { 'code': 102, 'data': _('更新地址簿有误') } return JsonResponse(result) def ab_get(request): # 兼容 x86-sciter 版客户端，此版客户端通过访问 "POST /api/ab/get" 来获取地址簿 request.method = 'GET' return ab(request) def sysinfo(request): # 客户端注册服务后，才会发送设备信息 result = {} if request.method == 'GET': result['error'] = _('错误的提交方式！') return JsonResponse(result) postdata = json.loads(request.body) device = RustDesDevice.objects.filter(Q(rid=postdata['id']) & Q(uuid=postdata['uuid'])).first() if not device: device = RustDesDevice( rid=postdata['id'], cpu=postdata['cpu'], hostname=postdata['hostname'], memory=postdata['memory'], os=postdata['os'], username=postdata.get('username', '-'), uuid=postdata['uuid'], version=postdata['version'], ) device.save() else: postdata2 = copy.copy(postdata) postdata2['rid'] = postdata2['id'] postdata2.pop('id') RustDesDevice.objects.filter(Q(rid=postdata['id']) & Q(uuid=postdata['uuid'])).update(**postdata2) result['data'] = 'ok' return JsonResponse(result) def heartbeat(request): postdata = json.loads(request.body) device = RustDesDevice.objects.filter(Q(rid=postdata['id']) & Q(uuid=postdata['uuid'])).first() if device: device.save() # token保活 create_time = datetime.datetime.now() + datetime.timedelta(seconds=EFFECTIVE_SECONDS) RustDeskToken.objects.filter(Q(rid=postdata['id']) & Q(uuid=postdata['uuid'])).update(create_time=create_time) result = {} result['data'] = _('在线') return JsonResponse(result) def audit(request): postdata = json.loads(request.body) # print(postdata) audit_type = postdata['action'] if 'action' in postdata else '' if audit_type == 'new': new_conn_log = ConnLog( action=postdata['action'] if 'action' in postdata else '', conn_id=postdata['conn_id'] if 'conn_id' in postdata else 0, from_ip=postdata['ip'] if 'ip' in postdata else '', from_id='', rid=postdata['id'] if 'id' in postdata else '', conn_start=datetime.datetime.now(), session_id=postdata['session_id'] if 'session_id' in postdata else 0, uuid=postdata['uuid'] if 'uuid' in postdata else '', ) new_conn_log.save() elif audit_type == "close": ConnLog.objects.filter(Q(conn_id=postdata['conn_id'])).update(conn_end=datetime.datetime.now()) elif 'is_file' in postdata: print(postdata) files = json.loads(postdata['info'])['files'] filesize = convert_filesize(int(files[0][1])) new_file_log = FileLog( file=postdata['path'], user_id=postdata['peer_id'], user_ip=json.loads(postdata['info'])['ip'], remote_id=postdata['id'], filesize=filesize, direction=postdata['type'], logged_at=datetime.datetime.now(), ) new_file_log.save() else: try: peer = postdata['peer'] ConnLog.objects.filter(Q(conn_id=postdata['conn_id'])).update(session_id=postdata['session_id']) ConnLog.objects.filter(Q(conn_id=postdata['conn_id'])).update(from_id=peer[0]) except Exception as e: print(postdata, e) result = { 'code': 1, 'data': 'ok' } return JsonResponse(result) def convert_filesize(size_bytes): if size_bytes == 0: return "0B" size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB") i = int(math.floor(math.log(size_bytes, 1024))) p = math.pow(1024, i) s = round(size_bytes / p, 2) return "%s %s" % (s, size_name[i]) def users(request): result = { 'code': 1, 'data': _('好的') } return JsonResponse(result) def peers(request): result = { 'code': 1, 'data': 'ok' } return JsonResponse(result)