Mozilla/mozilla/toolkit/mozapps/extensions/service/VersionCheck.java

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (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.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is the Extension Manager.
 *
 * The Initial Developer of the Original Code is Ben Goodger.
 * Portions created by the Initial Developer are Copyright (C) 2004
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *  Ben Goodger <ben@bengoodger.com>
 *
 * Alternatively, the contents of this file may be used under the terms of
 * either the GNU General Public License Version 2 or later (the "GPL"), or
 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 * in which case the provisions of the GPL or the LGPL are applicable instead
 * of those above. If you wish to allow use of your version of this file only
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 *
 * ***** END LICENSE BLOCK ***** */

package org.mozilla.update.extensions;

import java.sql.*;
import java.util.*;

public class VersionCheck
{
  public VersionCheck()
  {
  }

  /*
   public static void main(String[] args)
   {
   VersionCheck impl = new VersionCheck();
   // int id = impl.getNewestExtension("{bb8ee064-ccb9-47fc-94ae-ec335af3fe2d}", "3.0", "{ec8030f7-c20a-464f-9b0e-13a3a9e97384}", "0.8.0+");
   int id = impl.getNewestExtension("{93c4cb22-bf10-40a2-adff-c4c64a38df0c}", "1.5", "{ec8030f7-c20a-464f-9b0e-13a3a9e97384}", "0.8.0+");
   System.out.println("result row = " + id + ", xpiUrl = " + impl.getProperty(id, "xpiurl"));
   }
   */

  protected UpdateItem getNewestExtension(UpdateItem aInstalledItem, 
                                          String aTargetApp, 
                                          String aTargetAppVersion)
  {
    UpdateItem remoteItem = new UpdateItem();

    String installedVersion = aInstalledItem.getVersion();
    int extensionVersionParts = getPartCount(installedVersion);
    int targetAppVersionParts = getPartCount(aTargetAppVersion);

    int extensionVersion = parseVersion(installedVersion, extensionVersionParts);
    int targetAppVersion = parseVersion(aTargetAppVersion, targetAppVersionParts);

    Connection c;
    Statement sMain, sVersion, sApps;
    ResultSet rsMain, rsVersion, rsApps;
    try 
    {
      c = getConnection();
      sMain = c.createStatement();
  
      // We need to find all rows matching aInstalledItem.id, and filter like so:
      // 1) version > extensionVersion
      // 2) targetapp == aTargetApp
      // 3) mintargetappversion <= targetAppVersion <= maxtargetappversion
      String sqlMain = "SELECT * FROM t_main WHERE GUID = '" + aInstalledItem.getId() + "'";
      boolean temp = sMain.execute(sqlMain);
      rsMain = sMain.getResultSet();
      
      int newestExtensionVersion = extensionVersion;
      while (rsMain.next()) 
      {
        String sqlVersion = "SELECT * FROM t_version WHERE ID = '" + rsMain.getInt("ID") + "'";
        
        sVersion = c.createStatement();
        temp = sVersion.execute(sqlVersion);
        rsVersion = sVersion.getResultSet();
        
        while (rsVersion.next()) 
        {
          String sqlApps = "SELECT * FROM t_applications WHERE AppID = '" + rsVersion.getInt("AppID") + "' AND GUID = '" + aTargetApp + "'";
          
          sApps = c.createStatement();
          temp = sApps.execute(sqlApps);
          rsApps = sApps.getResultSet();
          
          if (rsApps.next())
          {
            int minTargetAppVersion = parseVersion(rsVersion.getString("MinAppVer"), targetAppVersionParts);
            int maxTargetAppVersion = parseVersion(rsVersion.getString("MaxAppVer"), targetAppVersionParts);
        
            int version = parseVersion(rsVersion.getString("Version"), extensionVersionParts);
            if (version > extensionVersion && 
                version > newestExtensionVersion &&
                minTargetAppVersion <= targetAppVersion && 
                targetAppVersion < maxTargetAppVersion) 
            {
              newestExtensionVersion = version;
              remoteItem.setRow(rsMain.getInt("ID"));
              remoteItem.setId(rsMain.getString("GUID"));
              remoteItem.setName(rsMain.getString("Name"));
              remoteItem.setVersion(rsVersion.getString("Version"));
              remoteItem.setMinAppVersion(rsVersion.getString("MinAppVer"));
              remoteItem.setMaxAppVersion(rsVersion.getString("MaxAppVer"));
              remoteItem.setUpdateURL(rsMain.getString("URI"));
              remoteItem.setIconURL("");
            }
          }
          rsApps.close();
          sApps.close();
        }
        rsVersion.close();
        sVersion.close();
      }
      rsMain.close();
      sMain.close();
      c.close();
    }
    catch (Exception e)
    {
    }

    return remoteItem;
  }

  protected int parseVersion(String aVersionString, int aPower)
  {
    int version = 0;
    StringTokenizer tokenizer = new StringTokenizer(aVersionString, ".");

    if (aPower == 0)
      aPower = tokenizer.countTokens();
    
    for (int i = 0; tokenizer.hasMoreTokens(); ++i) 
    {
      String token = tokenizer.nextToken();
      if (token.endsWith("+")) 
      {
        token = token.substring(0, token.lastIndexOf("+"));
        version += 1;
        if (token.length() == 0)
          continue;
      }

      version += Integer.parseInt(token) * Math.pow(10, aPower - i);
    }
    return version;
  }

  protected int getPartCount(String aVersionString)
  {
    return (new StringTokenizer(aVersionString, ".")).countTokens();
  }

  protected Connection getConnection() throws Exception
  {
    Class.forName("com.mysql.jdbc.Driver");
    return DriverManager.getConnection("jdbc:mysql://localhost/umo_extensions", "root", "");
  }
}