sms-proj/app-ajax.py

#!/usr/bin/env python3

import time
import datetime
import pprint
import configparser
import json
import re
import appdb, appsms
#from flask import Flask, render_template, request
import flask
from authlib.client import OAuth2Session
import google.oauth2.credentials
import googleapiclient.discovery

import google_auth
import callback_sms
import app_settings

config = configparser.ConfigParser()
config.read('config.ini')
app_debug = config.get("app","debug")

app = flask.Flask(__name__)
app.secret_key = config.get("auth","FN_FLASK_SECRET_KEY")

app.register_blueprint(google_auth.app)
app.register_blueprint(callback_sms.app)
app.register_blueprint(app_settings.app)

loginMsg = "You are not logged in.</a>"

if app_debug == '1':
    app.debug = True
else:
    app.debug = False

@app.route('/')
def index():
    if not google_auth.is_logged_in():
        return flask.render_template('deny.html', denymsg = loginMsg, loggedin = False)
    user_info = google_auth.get_user_info()
    indbRes = appdb.isUserinDB(user_info['id'])
    if indbRes:
        if app_debug == '1':
            pprint.pprint(indbRes)
        refreshtoken = google_auth.getRefreshToken()
        if not refreshtoken:
            return "Error with your refresh token"
        
        userid = appdb.getUserIDfromGoogleID(user_info['id'])
        if not userid:
            return 'You are not currently logged in.'
        
        rows = appdb.getDIDsbyAccount(userid)
        return flask.render_template('index.html',
                                    name = user_info['name'],
                                    picture = user_info['picture'],
                                    dids = rows,
                                    loggedin = True)
    else:
        # Lets setup the user
        refreshtoken = google_auth.getRefreshToken()
        if user_info['verified_email'] == True:
            verifiedVar = True
            appdb.setNewUser(user_info['id'], refreshtoken, user_info['name'], user_info['email'], verifiedVar)
            return flask.redirect(uri, code=302)
        else:
            #This means they aren't verified.
            verifiedVar = False
            return flask.render_template('deny.html',denymsg = 'Your google account does not have a verified email. This is required to use this service.', loggedin = False)


@app.route('/single/<int:number>', methods=['GET'])
def manageSingleSMS(number):
    if not google_auth.is_logged_in():
        return flask.render_template('deny.html',denymsg = loginMsg, loggedin = False)
    
    refreshtoken = google_auth.getRefreshToken()
    googleid = google_auth.getGoogleId()
    userid = appdb.getUserIDfromGoogleID(googleid)
    result = appdb.authIdforDID(userid,number)
    
    prettynum = appsms.prettyPhone(number)
    if appdb.validateFrom(int(number)) and result:
        return flask.render_template('single.html',srcnumber = number, prettynum = prettynum,loggedin = True)
    else:
        return flask.render_template('notvalid.html', srcnumber = number, prettynum = prettynum, loggedin = True)

@app.route('/getNumber/<int:number>',methods=['GET'])
def getNumMessages(number):
    #This gets the mssages based on the provided from or two DID
    if not google_auth.is_logged_in():
        return json.dumps({'error': 'Unable to send SMS, you are not logged in'})
    
    refreshtoken = google_auth.getRefreshToken()
    googleid = google_auth.getGoogleId()
    userid = appdb.getUserIDfromGoogleID(googleid)
    result = appdb.authIdforDID(userid,number)
    smslog = appdb.getNumSMSLog(number,10)
    if not result:
        return json.dumps({'error': 'You are not allowed to use the requested DID'})
    
    i = 0
    msgjson = ""
    for line in smslog:
        prettyto = appsms.prettyPhone(line[7])
        prettyfrom = appsms.prettyPhone(line[6])
        if i >= 1:
            msgjson = msgjson + ',' + json.dumps({'to':prettyto,'from':prettyfrom,'body':line[9],'timestamp': line[4],'status': line[10]})
        else:
            msgjson =  json.dumps({'to':prettyto,'from':prettyfrom,'body':line[9],'timestamp': line[4], 'status': line[10]})
        i += 1
        
    
    msgArrayJson = '['+msgjson +']'
    return msgArrayJson

@app.route('/submitMessage', methods=['POST'])
def submitMessage():
    #This is to submit a message.
    
    if not google_auth.is_logged_in():
        return json.dumps({'error': 'Unable to send SMS'})
    
    message = flask.request.form['message']
    fromDid = flask.request.form['fromdid']
    targetDid = flask.request.form['targetdid']
    user_info = google_auth.get_user_info()
    refreshtoken = google_auth.getRefreshToken()
    googleid = google_auth.getGoogleId()
    userid = appdb.getUserIDfromGoogleID(googleid)
    result = appdb.authIdforDID(userid,fromDid)
    
    if userid != result:
        if app_debug == '1':
            pprint.pprint(userid)
            pprint.pprint(result)
        return json.dumps({'error': 'Unauthorized UserID of ' + str(userid) + " and DID id of " + str(result) + " and fromDID " + str(fromDid)})
     
    if appdb.validateFrom(fromDid) == False:
        return json.dumps({'error': 'Unauthorized source phone number.'})
    
    uglyphone = appsms.uglyPhone(targetDid)
    
    #pprint.pprint('Got ' + message + ',' + fromDid)
    msg_id = appsms.sendsms(uglyphone,fromDid,message)
    if msg_id == False: #This sends the sms!
        returndata = json.dumps({'error': 'Unable to send SMS'})
    else:
        msgTS = time.strftime("%Y-%m-%dT%H:%m:%SZ")
        appdb.logsms_db(msg_id, msgTS, 'outbound', uglyphone, fromDid, 0.0040, 'pending', message, result)
        returndata = json.dumps({"msg" : message, "fromdid" : fromDid, 'targetdid' : targetDid,})
    return returndata

@app.route('/testAjax')
def testAjax():
    google_auth.getGoogleContacts()
    return json.dumps({"msg" : 'Success!'})

@app.route('/pp')
def PrivacyPolicy():
    if app_debug == '1':
        pprint.pprint(flask.session)
    if google_auth.is_logged_in():
        loggedin = True
    else:
        loggedin = False
    return flask.render_template('pp.html',loggedin=loggedin)

@app.route('/tos')
def tos():
    if google_auth.is_logged_in():
        loggedin = True
    else:
        loggedin = False 
    return flask.render_template('tos.html',loggedin=loggedin)

@app.route('/about')
def about():
    if google_auth.is_logged_in():
        loggedin = True
    else:
        loggedin = False 
    return flask.render_template('about.html',loggedin=loggedin)

if __name__ == '__main__':
    app.run(
        host="127.0.0.1",
        port=int("8890")
    )